Fix parsing math fidelity

tenstorrent · Feb 7, 2025 · bfa250c · bfa250c
1 parent 74704d2
commit bfa250c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/forge/test/operators/utils/plan.py b/forge/test/operators/utils/plan.py
@@ -691,6 +691,8 @@ def test_id_to_test_vector(cls, test_id: str) -> TestVector:
         # As last parameter in test id is math fidelity, in case of duplicated tests numeric suffix should be ignored
         if math_fidelity_part is not None and math_fidelity_part.startswith("HiFi4"):
             math_fidelity_part = "HiFi4"
+        if math_fidelity_part is not None and math_fidelity_part.startswith("None"):
+            math_fidelity_part = None
         math_fidelity = eval(f"forge._C.{math_fidelity_part}") if math_fidelity_part is not None else None
 
         return TestVector(