Merge pull request #25 from ROCm/arch_detect_fix

cherry pick arch detect fix
ROCm · May 9, 2024 · aa088e6 · aa088e6
2 parents 5bada9b + ee04a7e
commit aa088e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bitsandbytes/cuda_specs.py b/bitsandbytes/cuda_specs.py
@@ -52,7 +52,7 @@ def get_rocm_gpu_arch() -> str:
     try:
         if torch.version.hip:
             result = subprocess.run(["rocminfo"], capture_output=True, text=True)
-            match = re.search(r"Name:\s+gfx(\d+)", result.stdout)
+            match = re.search(r"Name:\s+gfx([a-zA-Z\d]+)", result.stdout)
             if match:
                 return "gfx" + match.group(1)
             else: