Fix math fidelity HiFi4 for duplicated tests

tenstorrent · Jan 20, 2025 · 478e1e3 · 478e1e3
1 parent 42c6122
commit 478e1e3
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/forge/test/operators/utils/plan.py b/forge/test/operators/utils/plan.py
@@ -629,11 +629,8 @@ def test_id_to_test_vector(cls, test_id: str) -> TestVector:
         math_fidelity_part = parts[math_fidelity_index]
         if math_fidelity_part == "None":
             math_fidelity_part = None
-        # TODO remove hardcoded values here
-        if math_fidelity_part in (
-            "HiFi40",
-            "HiFi41",
-        ):
+        # 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"
         math_fidelity = eval(f"forge._C.{math_fidelity_part}") if math_fidelity_part is not None else None