Test on CPU machine

iree-org · Sep 13, 2023 · 1c5ee4b · 1c5ee4b
1 parent 3116512
commit 1c5ee4b
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/.github/workflows/update_model_artifacts.yml b/.github/workflows/update_model_artifacts.yml
@@ -50,7 +50,7 @@ jobs:
       - self-hosted  # must come first
       - runner-group=${{ needs.setup.outputs.runner-group }}
       - environment=prod
-      - machine-type=a2-highgpu-1g
+      - machine-type=c2-standard-16
     steps:
       - name: "Checking out PR repository"
         uses: actions/checkout@e2f20e631ae6d7dd3b768f56a5d2af784dd54791  # v2.5.0

diff --git a/...nchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh b/...nchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh
@@ -64,9 +64,9 @@ else
   args+=(
     --filter
     "RESNET50_FP32_JAX_.+"
-    "BERT_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48|64|512)"
-    "T5_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48)"
-    "T5_4CG_LARGE_FP32_JAX_.+"
+    # "BERT_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48|64|512)"
+    # "T5_LARGE_FP32_JAX_.+_BATCH(1|16|24|32|48)"
+    # "T5_4CG_LARGE_FP32_JAX_.+"
     "GPT2LMHEAD_FP32_JAX_.+"
   )
 fi