From 97ae086fd6d01e24f7e75df996e899e50ee205a5 Mon Sep 17 00:00:00 2001 From: mariecwhite Date: Wed, 31 Jan 2024 22:40:40 +0000 Subject: [PATCH] Update JAX and TFLite models --- .github/workflows/run_comparative_benchmark.yml | 2 +- .../benchmark/comparative_suite/tflite/model_definitions.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/.github/workflows/run_comparative_benchmark.yml b/.github/workflows/run_comparative_benchmark.yml index d7ad8915..08722fb7 100644 --- a/.github/workflows/run_comparative_benchmark.yml +++ b/.github/workflows/run_comparative_benchmark.yml @@ -249,7 +249,7 @@ jobs: run: | RESULTS_PATH="${RESULTS_DIR}/${JAX_IREE_RESULTS_JSON}" docker run --mount="type=bind,src="${PWD}",target=/work" --workdir="/work" \ - "gcr.io/iree-oss/openxla-benchmark/base-python3.10@sha256:245a074284cfed5de60cf06a153e3bcd9a9c42702b6bb66a39bb47ef23b61669" \ + "gcr.io/iree-oss/openxla-benchmark/base-python3.11@sha256:b9b98da7bcc5e431800ff798a6dcc394b1838a9ed3d695f5cd0dac3510fc8c8d" \ ./comparative_benchmark/jax/benchmark_iree.sh \ "${TARGET_DEVICE}"\ "${RESULTS_PATH}" diff --git a/common_benchmark_suite/openxla/benchmark/comparative_suite/tflite/model_definitions.py b/common_benchmark_suite/openxla/benchmark/comparative_suite/tflite/model_definitions.py index e218198a..50d34d3c 100644 --- a/common_benchmark_suite/openxla/benchmark/comparative_suite/tflite/model_definitions.py +++ b/common_benchmark_suite/openxla/benchmark/comparative_suite/tflite/model_definitions.py @@ -10,7 +10,7 @@ from openxla.benchmark import def_types from openxla.benchmark.comparative_suite import utils -PARENT_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/tflite/tflite_models_1706050932" +PARENT_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/tflite/tflite_models_1706739936" ARTIFACTS_DIR_URL_TEMPLATE = string.Template(PARENT_GCS_DIR + "/${name}") TFLITE_MODEL_IMPL = def_types.ModelImplementation( @@ -20,7 +20,7 @@ module_path=f"{utils.MODELS_MODULE_PATH}.tflite.tflite_model", ) -JAX_MODELS_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085" +JAX_MODELS_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1706594181" BERT_BASE_FP32_TFLITE_I32_INPUT_SEQUENCE_TEMPLATE = utils.ModelTemplate( name=utils.SEQ_LEN_NAME("BERT_BASE_FP32_TFLITE_I32"),