File tree Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Expand file tree Collapse file tree 2 files changed +4
-3
lines changed Original file line number Diff line number Diff line change 4444
4545 # Test ExecuTorch CUDA build - ExecuTorch will automatically detect CUDA version
4646 # and install the appropriate PyTorch wheel
47- CMAKE_ARGS="-DEXECUTORCH_BUILD_QNN=OFF" source .ci/scripts/test-cuda-build.sh "${{ matrix.cuda-version }}"
47+ source .ci/scripts/test-cuda-build.sh "${{ matrix.cuda-version }}"
4848
4949 # This job will fail if any of the CUDA versions fail
5050 check-all-cuda-builds :
8383 script : |
8484 set -eux
8585
86- CMAKE_ARGS="-DEXECUTORCH_BUILD_QNN=OFF" PYTHON_EXECUTABLE=python ./install_executorch.sh
86+ PYTHON_EXECUTABLE=python ./install_executorch.sh
8787 export LD_LIBRARY_PATH=/opt/conda/lib:$LD_LIBRARY_PATH
8888 PYTHON_EXECUTABLE=python source .ci/scripts/test_model.sh "${{ matrix.model }}" cmake cuda
8989
@@ -132,7 +132,7 @@ jobs:
132132 set -eux
133133
134134 echo "::group::Setup ExecuTorch"
135- CMAKE_ARGS="-DEXECUTORCH_BUILD_QNN=OFF" ./install_executorch.sh
135+ ./install_executorch.sh
136136 echo "::endgroup::"
137137
138138 echo "::group::Setup Huggingface"
Original file line number Diff line number Diff line change @@ -23,6 +23,7 @@ get_filename_component(
2323 _common_include_directories "${EXECUTORCH_SOURCE_DIR} /.." ABSOLUTE
2424)
2525
26+ # We only download QNN SDK when we build pip wheel for ExecuTorch.
2627if (EXECUTORCH_BUILD_PYBIND)
2728 set (_qnn_default_sdk_dir "${CMAKE_CURRENT_BINARY_DIR} /sdk/qnn" )
2829
You can’t perform that action at this time.
0 commit comments