Update collect_env.py

HabanaAI · Oct 25, 2024 · c665f48 · c665f48
1 parent 1e427c2
commit c665f48
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/collect_env.py b/collect_env.py
@@ -254,8 +254,8 @@ def get_nvidia_smi():
 
 
 def get_hpu_info():
-    command = ["hl-smi", "-q", "-d", "PRODUCT"]
     try:
+        command = ["hl-smi", "-q", "-d", "PRODUCT"]
         lines = subprocess.Popen(command, stdout=subprocess.PIPE, universal_newlines=True).stdout.readlines()
         lines = [l.strip('\t') for l in lines]
         hpu_count = None
@@ -283,7 +283,7 @@ def get_hpu_info():
         if hpu_model is None:
             return ('N/A', hpu_driver)
         return (f'{hpu_count}x {hpu_model}', hpu_driver)
-    except Exception as e:
+    except:
         return ('N/A', 'N/A')