Enables CLIP_train by removing the compilation_xfail mark (#804)

Enables several timm models by removing the `compilation_xfail` mark
tenstorrent · Mar 3, 2025 · 0c062c4 · 0c062c4
1 parent 8c78b36
commit 0c062c4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 11 deletions.
diff --git a/tests/models/clip/test_clip.py b/tests/models/clip/test_clip.py
@@ -47,10 +47,7 @@ def get_results_train(self, model, inputs, outputs):
 @pytest.mark.parametrize(
     "mode",
     [
-        pytest.param(
-            "train",
-            marks=pytest.mark.compilation_xfail,
-        ),
+        "train",
         "eval",
     ],
 )

diff --git a/tests/models/timm/test_timm_image_classification.py b/tests/models/timm/test_timm_image_classification.py
@@ -38,13 +38,13 @@ def _load_inputs(self):
 
 
 model_and_mode_list = [
-    pytest.param(["tf_efficientnet_lite0.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["tf_efficientnet_lite1.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["tf_efficientnet_lite2.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["tf_efficientnet_lite3.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["tf_efficientnet_lite4.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["ghostnet_100.in1k", "train"], marks=pytest.mark.compilation_xfail),
-    pytest.param(["ghostnetv2_100.in1k", "train"], marks=pytest.mark.compilation_xfail),
+    ["tf_efficientnet_lite0.in1k", "train"],
+    ["tf_efficientnet_lite1.in1k", "train"],
+    ["tf_efficientnet_lite2.in1k", "train"],
+    ["tf_efficientnet_lite3.in1k", "train"],
+    ["tf_efficientnet_lite4.in1k", "train"],
+    ["ghostnet_100.in1k", "train"],
+    ["ghostnetv2_100.in1k", "train"],
     ["inception_v4.tf_in1k", "train"],
     ["mixer_b16_224.goog_in21k", "train"],
     ["mobilenetv1_100.ra4_e3600_r224_in1k", "train"],