Test on CPU machine

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

Hide file tree

Showing 2 changed files with 17 additions and 7 deletions.
diff --git a/.github/workflows/update_model_artifacts.yml b/.github/workflows/update_model_artifacts.yml
@@ -50,19 +50,29 @@ 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
+      # - name: "Generate JAX model artifacts"
+      #   run: |
+      #     mkdir jax
+      #     docker run --gpus all --mount="type=bind,src="${PWD}",target=/work" --workdir="/work" \
+      #       --env "PYTHON=python3" \
+      #       --env "WITH_CUDA=1" \
+      #       --env "AUTO_UPLOAD=0" \
+      #       --env "OUTPUT_DIR=jax" \
+      #       "gcr.io/iree-oss/openxla-benchmark/cuda11.8-cudnn8.9@sha256:f43984cd6c16ad1faad4dfb6aac3f53e552dd728c9330c90752e78ae51e4276f" \
+      #       "common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh"
       - name: "Generate JAX model artifacts"
         run: |
           mkdir jax
-          docker run --gpus all --mount="type=bind,src="${PWD}",target=/work" --workdir="/work" \
+          docker run --mount="type=bind,src="${PWD}",target=/work" --workdir="/work" \
             --env "PYTHON=python3" \
-            --env "WITH_CUDA=1" \
+            --env "WITH_CUDA=0" \
             --env "AUTO_UPLOAD=0" \
             --env "OUTPUT_DIR=jax" \
-            "gcr.io/iree-oss/openxla-benchmark/cuda11.8-cudnn8.9@sha256:f43984cd6c16ad1faad4dfb6aac3f53e552dd728c9330c90752e78ae51e4276f" \
+            "gcr.io/iree-oss/openxla-benchmark/base@sha256:1bf3e319465ec8fb465baae3f6ba9a5b09cb84a5349a675c671a552fc77f2251" \
             "common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh"
       - name: "Upload JAX model artifacts"
         run: |

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