igpu dont error on device parse fail

nod-ai · Jun 2, 2024 · 5f1b5e5 · 5f1b5e5
1 parent 6adae49
commit 5f1b5e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/apps/shark_studio/api/utils.py b/apps/shark_studio/api/utils.py
@@ -196,9 +196,10 @@ def get_rocm_target_chip(device_str):
     for key in rocm_chip_map:
         if key in device_str:
             return rocm_chip_map[key]
-    raise AssertionError(
-        f"Device {device_str} not recognized. Please file an issue at https://github.com/nod-ai/SHARK/issues."
-    )
+    return None
+    # raise AssertionError(
+    #     f"Device {device_str} not recognized. Please file an issue at https://github.com/nod-ai/SHARK/issues."
+    # )
 
 
 def get_all_devices(driver_name):