Non-GenAI models coverage XNNPACK

pytorch · Jul 30, 2024 · 81a6c94 · 81a6c94
1 parent b7c8378
commit 81a6c94
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 4 deletions.
diff --git a/.ci/scripts/gather_test_models.py b/.ci/scripts/gather_test_models.py
@@ -39,6 +39,12 @@
     "macos": {
         "mobilebert": 90,
     },
+    "linux": {
+        "emformer_predict": 360,
+    },
+    "macos": {
+        "emformer_predict": 360,
+    },
 }
 
 
@@ -84,7 +90,11 @@ def model_should_run_on_event(model: str, event: str) -> bool:
     """
     if event == "pull_request":
         return model in ["mv3", "vit"]
-    return True
+    elif event == "push":
+        # 'emformer_predict' is running super slow. Only run it periodically
+        return model not in ["emformer_predict"]
+    else:
+        return True
 
 
 def model_should_run_on_target_os(model: str, target_os: str) -> bool:

diff --git a/examples/xnnpack/__init__.py b/examples/xnnpack/__init__.py
@@ -24,11 +24,14 @@ class XNNPACKOptions(object):
     "mv3": XNNPACKOptions(True, True),
     "resnet18": XNNPACKOptions(True, True),
     "resnet50": XNNPACKOptions(True, True),
-    "vit": XNNPACKOptions(False, True),
-    "w2l": XNNPACKOptions(False, True),
+    "vit": XNNPACKOptions(False, True),  # T161242362
+    "w2l": XNNPACKOptions(True, True),
     "edsr": XNNPACKOptions(True, True),
-    "mobilebert": XNNPACKOptions(False, True),  # T170286473
+    "mobilebert": XNNPACKOptions(False, True),  # T197452682
     "llama2": XNNPACKOptions(False, True),
+    "emformer_join": XNNPACKOptions(True, True),
+    "emformer_predict": XNNPACKOptions(True, True),
+    "emformer_transcribe": XNNPACKOptions(False, True),  # T197449765
 }