Skip to content

Commit 8f46024

Browse files
author
Guang Yang
committed
Pin bump for optimum-executorch
1 parent 32d2dfa commit 8f46024

File tree

3 files changed

+10
-37
lines changed

3 files changed

+10
-37
lines changed

.github/workflows/android-perf.yml

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -344,7 +344,7 @@ jobs:
344344
git clone https://github.com/huggingface/optimum-executorch
345345
pushd optimum-executorch
346346
# There is no release yet, for CI stability, always test from the same commit on main
347-
git checkout 4c3b18f6cca68c5ccff809131d570062723d7188
347+
git checkout a3942627f5ac048e06b4b1d703b0a6a53bf6da5b
348348
python install_dev.py --skip_override_torch
349349
pip list
350350
@@ -353,21 +353,12 @@ jobs:
353353
"--task" "text-generation"
354354
"--recipe" "xnnpack"
355355
"--use_custom_sdpa"
356+
"--use_custom_kv_cache"
356357
"--qlinear"
357358
"--qembedding"
358359
"--output_dir" ".."
359360
)
360361
361-
# Add conditional arguments based on model
362-
case "${HF_MODEL_REPO}" in
363-
*"google/gemma-3-1b-it"*)
364-
echo "--use_custom_kv_cache can not be used for HybridCache"
365-
;;
366-
*)
367-
ARGS+=("--use_custom_kv_cache")
368-
;;
369-
esac
370-
371362
optimum-cli export executorch "${ARGS[@]}"
372363
popd
373364

.github/workflows/apple-perf.yml

Lines changed: 2 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -349,7 +349,7 @@ jobs:
349349
git clone https://github.com/huggingface/optimum-executorch
350350
pushd optimum-executorch
351351
# There is no release yet, for CI stability, always test from the same commit on main
352-
git checkout 4c3b18f6cca68c5ccff809131d570062723d7188
352+
git checkout a3942627f5ac048e06b4b1d703b0a6a53bf6da5b
353353
${CONDA_RUN} python install_dev.py --skip_override_torch
354354
pip list
355355
@@ -358,21 +358,12 @@ jobs:
358358
"--task" "text-generation"
359359
"--recipe" "xnnpack"
360360
"--use_custom_sdpa"
361+
"--use_custom_kv_cache"
361362
"--qlinear"
362363
"--qembedding"
363364
"--output_dir" ".."
364365
)
365366
366-
# Add conditional arguments based on model
367-
case "${HF_MODEL_REPO}" in
368-
*"google/gemma-3-1b-it"*)
369-
echo "--use_custom_kv_cache can not be used for HybridCache"
370-
;;
371-
*)
372-
ARGS+=("--use_custom_kv_cache")
373-
;;
374-
esac
375-
376367
${CONDA_RUN} optimum-cli export executorch "${ARGS[@]}"
377368
popd
378369

.github/workflows/trunk.yml

Lines changed: 6 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -597,7 +597,7 @@ jobs:
597597
git clone https://github.com/huggingface/optimum-executorch
598598
pushd optimum-executorch
599599
# There is no release yet, for CI stability, always test from the same commit on main
600-
git checkout 4c3b18f6cca68c5ccff809131d570062723d7188
600+
git checkout a3942627f5ac048e06b4b1d703b0a6a53bf6da5b
601601
python install_dev.py --skip_override_torch
602602
popd
603603
pip list
@@ -614,21 +614,12 @@ jobs:
614614
"--task" "text-generation"
615615
"--recipe" "xnnpack"
616616
"--use_custom_sdpa"
617+
"--use_custom_kv_cache"
617618
"--qlinear"
618619
"--qembedding"
619620
"--output_dir" "${OUTPUT_DIR}"
620621
)
621622
622-
# Add conditional arguments based on model
623-
case "${MODEL_ID}" in
624-
*"google/gemma-3-1b-it"*)
625-
echo "--use_custom_kv_cache can not be used for HybridCache"
626-
;;
627-
*)
628-
ARGS+=("--use_custom_kv_cache")
629-
;;
630-
esac
631-
632623
optimum-cli export executorch "${ARGS[@]}"
633624
634625
ls -FlAGhp ${OUTPUT_DIR}
@@ -732,18 +723,18 @@ jobs:
732723
timeout: 90
733724
script: |
734725
set -eux
735-
726+
736727
# The generic Linux job chooses to use base env, not the one setup by the image
737728
CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
738729
conda activate "${CONDA_ENV}"
739-
730+
740731
# Build and install Executorch
741732
PYTHON_EXECUTABLE=python \
742733
CMAKE_ARGS="-DEXECUTORCH_BUILD_NXP_NEUTRON=ON" \
743734
.ci/scripts/setup-linux.sh --build-tool "cmake"
744-
735+
745736
# Install test requirements
746737
pip install -r backends/nxp/requirements-tests.txt
747-
738+
748739
# Run pytest
749740
PYTHON_EXECUTABLE=python bash backends/nxp/run_unittests.sh

0 commit comments

Comments
 (0)