[sweeps] fix tt-smi util bug

tenstorrent · Feb 25, 2025 · fb1375d · fb1375d
1 parent 4e82e7d
commit fb1375d
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/sweep_framework/framework/tt_smi_util.py b/tests/sweep_framework/framework/tt_smi_util.py
@@ -57,6 +57,8 @@ def __init__(self, arch: str):
                 if smi_process.returncode == 0:
                     self.command = executable
                     self.args = args
+                    break
+
         if self.command is None:
             raise Exception(f"SWEEPS: Unable to location tt-smi executable")
         print(f"SWEEPS: tt-smi util initialized with command: {self.command}, args: {self.args}")