Update Pixel 8 Pro CPU feature flags

iree-org · Jan 4, 2024 · fc9d78b · fc9d78b
1 parent ca9244f
commit fc9d78b
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/model_definitions.py b/common_benchmark_suite/openxla/benchmark/comparative_suite/jax/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/jax/jax_models_0.4.20_1699872537"
+PARENT_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1702848353"
 ARTIFACTS_DIR_URL_TEMPLATE = string.Template(PARENT_GCS_DIR + "/${name}")
 
 T5_JAX_IMPL = def_types.ModelImplementation(
@@ -572,4 +572,5 @@
     )) + [
         GPT2LMHEAD_PIPELINE_JAX_1X4XI32,
         T5_SMALL_FP32_JAX_1X128XI32,
+        VIT_CLASSIFICATION_JAX_3X224X224XF32,
     ]
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
@@ -10,7 +10,8 @@
 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_1701944859"
+PARENT_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/tflite/tflite_models_1703027804"
+
 ARTIFACTS_DIR_URL_TEMPLATE = string.Template(PARENT_GCS_DIR + "/${name}")
 
 TFLITE_MODEL_IMPL = def_types.ModelImplementation(
@@ -20,7 +21,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.21_1701943542"
+JAX_MODELS_GCS_DIR = "https://storage.googleapis.com/iree-model-artifacts/jax/jax_models_0.4.23_1702848353"
 
 BERT_BASE_FP32_TFLITE_I32_INPUT_SEQUENCE_TEMPLATE = utils.ModelTemplate(
     name=utils.SEQ_LEN_NAME("BERT_BASE_FP32_TFLITE_I32"),

diff --git a/experimental/iree/compile_workloads.py b/experimental/iree/compile_workloads.py
@@ -66,7 +66,7 @@
     ],
     "pixel-8-pro": [
         "--iree-llvmcpu-target-triple=aarch64-none-linux-android34",
-        "--iree-llvmcpu-target-cpu-features=+fp-armv8,+neon,+crc,+lse,+rdm,+ras,+rcpc,+dotprod,+v9a,+sb,+ssbs,+fullfp16,+fp16fml,+i8mm,+bf16,+flagm"
+        "--iree-llvmcpu-target-cpu-features=+v9a,+fullfp16,fp-armv8,+neon,+aes,+sha2,+crc,+lse,+rdm,+complxnum,+rcpc,+sha3,+sm4,+dotprod,+fp16fml,+dit,+flagm,+ssbs,+sb,+sve2-aes,+sve2-bitperm,+sve2-sha3,+sve2-sm4,+altnzcv,+fptoint,+bf16,+i8mm,+bti,+mte,+pauth,+perfmon,+predres,+spe,+ras"
     ],
 }
 

diff --git a/experimental/iree/compile_workloads.sh b/experimental/iree/compile_workloads.sh
@@ -55,7 +55,8 @@ declare -a ANDROID_BENCHMARKS=(
   "models/VIT_CLASSIFICATION_DYN_QUANT_TFLITE_3X224X224XF32/.+"
   "models/VIT_CLASSIFICATION_INT8_TFLITE_3X224X224XINT8/.+"
   "models/BERT_BASE_FP32_JAX_I32_SEQLEN.+/.+"
-  "models/BERT_BASE_FP16_JAX_I32_SEQLEN.+/.+"
+  # Disable sequence lengths >8 due to https://github.com/openxla/iree/issues/16042.
+  "models/BERT_BASE_FP16_JAX_I32_SEQLEN8/.+"
   "models/T5_4CG_SMALL_FP32_JAX_1X128XI32_GEN.+/.+"
   "models/SD_PIPELINE_FP32_JAX_64XI32_BATCH1/.+"
   "models/SD_PIPELINE_FP16_JAX_64XI32_BATCH1/.+"