@@ -442,7 +442,7 @@ jobs:
442442
443443 # Compile only as weight sharing is not applicable on x86
444444 test-static-llama-size-qnn-linux :
445- name : test-static-llama-runner -qnn-linux
445+ name : test-static-llama-size -qnn-linux
446446 uses : pytorch/test-infra/.github/workflows/linux_job_v2.yml@main
447447 permissions :
448448 id-token : write
@@ -460,6 +460,8 @@ jobs:
460460 CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
461461 conda activate "${CONDA_ENV}"
462462
463+ BUILD_TOOL="cmake"
464+
463465 PYTHON_EXECUTABLE=python bash .ci/scripts/setup-qnn-deps.sh
464466 PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
465467
@@ -475,7 +477,7 @@ jobs:
475477
476478 # Checks accuracy with weight sharing disabled since x86 does not support weight sharing.
477479 test-static-llama-accuracy-qnn-linux :
478- name : test-static-llama-runner -qnn-linux
480+ name : test-static-llama-accuracy -qnn-linux
479481 uses : pytorch/test-infra/.github/workflows/linux_job.yml@main
480482 strategy :
481483 fail-fast : false
@@ -490,6 +492,8 @@ jobs:
490492 CONDA_ENV=$(conda env list --json | jq -r ".envs | .[-1]")
491493 conda activate "${CONDA_ENV}"
492494
495+ BUILD_TOOL="cmake"
496+
493497 PYTHON_EXECUTABLE=python bash .ci/scripts/setup-qnn-deps.sh
494498 PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
495499
0 commit comments