Skip to content

Commit 2d71438

Browse files
authored
fix (#1710)
* fix * update lint
1 parent 94c5b98 commit 2d71438

1 file changed

Lines changed: 7 additions & 3 deletions

File tree

aiter/jit/core.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -377,11 +377,15 @@ def validate_and_update_archs():
377377
def hip_flag_checker(flag_hip: str) -> bool:
378378
import subprocess
379379

380-
cmd = ["hipcc", flag_hip, "-x", "hip", "-E", "-P", "/dev/null", "-o", "/dev/null"]
380+
cmd = (
381+
["hipcc"]
382+
+ flag_hip.split()
383+
+ ["-x", "hip", "-E", "-P", "/dev/null", "-o", "/dev/null"]
384+
)
381385
try:
382-
subprocess.check_output(cmd, stderr=subprocess.STDOUT)
386+
subprocess.check_output(cmd, stderr=subprocess.DEVNULL)
383387
except subprocess.CalledProcessError:
384-
logger.warning(f"Current hipcc not support: {flag_hip}")
388+
logger.warning(f"Current hipcc not support: {flag_hip}, skip it.")
385389
return False
386390
return True
387391

0 commit comments

Comments
 (0)