From 5f1b5e58d6c1f137e9a8fd43eecb2b877edf31bf Mon Sep 17 00:00:00 2001 From: Ean Garvey <garveyej@gmail.com> Date: Sun, 2 Jun 2024 12:56:44 -0500 Subject: [PATCH] igpu dont error on device parse fail --- apps/shark_studio/api/utils.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/apps/shark_studio/api/utils.py b/apps/shark_studio/api/utils.py index aabecc32d6..0f599c4b2d 100644 --- 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):