Update JAX and TFLite MLIR artifacts for benchmarking
diff --git a/build_tools/python/e2e_test_framework/models/jax_models.py b/build_tools/python/e2e_test_framework/models/jax_models.py
index 225246f..1a8ce4b 100644
--- a/build_tools/python/e2e_test_framework/models/jax_models.py
+++ b/build_tools/python/e2e_test_framework/models/jax_models.py
@@ -11,7 +11,7 @@
from e2e_test_framework.definitions import common_definitions
import e2e_test_framework.models.utils as model_utils
-GCS_ARTIFACT_ROOT_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085"
+GCS_ARTIFACT_ROOT_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220"
ID_FORMAT = string.Template("${model_id}-batch${batch_size}")
NAME_FORMAT = string.Template("${name}_BATCH${batch_size}")
diff --git a/tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake b/tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake
index e26b098..fba064a 100644
--- a/tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake
+++ b/tests/e2e/test_artifacts/generated_e2e_test_fetch_models.cmake
@@ -147,21 +147,21 @@
iree_fetch_artifact(
NAME "model-BERT_LARGE_JAX_384XI32_BATCH1"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/BERT_LARGE_FP32_JAX_384XI32_BATCH1/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/BERT_LARGE_FP32_JAX_384XI32_BATCH1/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_BERT_LARGE_JAX_384XI32_BATCH1.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-BERT_LARGE_JAX_384XI32_BATCH32"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/BERT_LARGE_FP32_JAX_384XI32_BATCH32/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/BERT_LARGE_FP32_JAX_384XI32_BATCH32/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_BERT_LARGE_JAX_384XI32_BATCH32.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-BERT_LARGE_JAX_384XI32_BATCH64"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/BERT_LARGE_FP32_JAX_384XI32_BATCH64/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/BERT_LARGE_FP32_JAX_384XI32_BATCH64/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_BERT_LARGE_JAX_384XI32_BATCH64.mlirbc"
UNPACK
)
@@ -182,42 +182,42 @@
iree_fetch_artifact(
NAME "model-RESNET50_FP32_JAX_3X224X224XF32_BATCH1"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/RESNET50_FP32_JAX_3X224X224XF32_BATCH1/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/RESNET50_FP32_JAX_3X224X224XF32_BATCH1/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_RESNET50_FP32_JAX_3X224X224XF32_BATCH1.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-RESNET50_FP32_JAX_3X224X224XF32_BATCH64"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/RESNET50_FP32_JAX_3X224X224XF32_BATCH64/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/RESNET50_FP32_JAX_3X224X224XF32_BATCH64/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_RESNET50_FP32_JAX_3X224X224XF32_BATCH64.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-RESNET50_FP32_JAX_3X224X224XF32_BATCH128"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/RESNET50_FP32_JAX_3X224X224XF32_BATCH128/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/RESNET50_FP32_JAX_3X224X224XF32_BATCH128/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_RESNET50_FP32_JAX_3X224X224XF32_BATCH128.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-T5_LARGE_FP32_JAX_512XI32_BATCH1"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/T5_LARGE_FP32_JAX_512XI32_BATCH1/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/T5_LARGE_FP32_JAX_512XI32_BATCH1/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_T5_LARGE_FP32_JAX_512XI32_BATCH1.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-T5_LARGE_FP32_JAX_512XI32_BATCH16"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/T5_LARGE_FP32_JAX_512XI32_BATCH16/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/T5_LARGE_FP32_JAX_512XI32_BATCH16/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_T5_LARGE_FP32_JAX_512XI32_BATCH16.mlirbc"
UNPACK
)
iree_fetch_artifact(
NAME "model-T5_LARGE_FP32_JAX_512XI32_BATCH32"
- SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1705868085/T5_LARGE_FP32_JAX_512XI32_BATCH32/stablehlo.mlirbc"
+ SOURCE_URL "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.25_1709787220/T5_LARGE_FP32_JAX_512XI32_BATCH32/stablehlo.mlirbc"
OUTPUT "${ROOT_ARTIFACTS_DIR}/model_T5_LARGE_FP32_JAX_512XI32_BATCH32.mlirbc"
UNPACK
)
