Fix android-perf job

Differential Revision: D61632951 Pull Request resolved: #4825
pytorch · Aug 22, 2024 · c7aff77 · c7aff77
1 parent 288e758
commit c7aff77
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