Fix android-perf job

pytorch · Aug 21, 2024 · 8dac826 · 8dac826
1 parent 1fc0cfc
commit 8dac826
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/android-perf.yml b/.github/workflows/android-perf.yml
@@ -156,7 +156,7 @@ jobs:
         BUILD_MODE="cmake"
         DTYPE="fp32"
 
-        if [[ ${{ matrix.model }} == "stories*"" ]]; then
+        if [[ ${{ matrix.model }} =~ ^stories* ]]; then
             # Install requirements for export_llama
             PYTHON_EXECUTABLE=python bash examples/models/llama2/install_requirements.sh
             # Test llama2