feat: fix win compat

Signed-off-by: Zihua Wu <13583761+lucifer1004@users.noreply.github.com>
This commit is contained in:
Zihua Wu
2025-04-22 23:01:23 -07:00
parent a3210ac850
commit 40c09fb883
2 changed files with 18 additions and 16 deletions

View File

@@ -10,7 +10,9 @@ from .utils import run_gemm
def get_symbol(file_path: str, pattern: str) -> Optional[str]:
command = [f'{CUDA_HOME}/bin/cuobjdump', '-symbols', file_path]
if CUDA_HOME is None:
raise Exception("CUDA_HOME is not set")
command = [os.path.join(CUDA_HOME, 'bin', 'cuobjdump'), '-symbols', file_path]
result = subprocess.run(command, stdout=subprocess.PIPE,
stderr=subprocess.PIPE, text=True)
assert result.returncode == 0