update

benchmarks/dynamo/timm_models.py

@@ -230,6 +230,10 @@ class TimmRunner(BenchmarkRunner):
     def _skip(self):
         return self._config["skip"]
 
+    @property
+    def skip_models_for_cpu(self):
+        return self._skip["device"]["cpu"]
+
     @property
     def skip_models(self):
         return self._skip["all"]

benchmarks/dynamo/timm_models.yaml

@@ -2,3 +2,11 @@
 skip:
   all:
     - ~
+  device:
+    cpu:
+      # timeout on aarch64
+      - dm_nfnet_f0
+      - nfnet_l0
+      - resnest101e
+      - swsl_resnext101_32x16d
+      - visformer_small

benchmarks/dynamo/torchbench.yaml

@@ -212,6 +212,9 @@ skip:
       - stable_diffusion_text_encoder
       - llava
       - moco
+      # timeout on aarch64
+      - timm_regnet
+      - timm_nfnet
 
     cuda: []