From ae675fd7951999b3b21b41a6116f5ad3c8b6d905 Mon Sep 17 00:00:00 2001 From: Jerry Wu Date: Wed, 13 Sep 2023 18:45:01 +0000 Subject: [PATCH] Enable others --- .../jax/scripts/generate_model_artifacts.sh | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh b/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh index 601dd61c..a586e151 100755 --- a/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/scripts/generate_model_artifacts.sh +++ b/common_benchmark_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