File tree Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Expand file tree Collapse file tree 2 files changed +6
-1
lines changed Original file line number Diff line number Diff line change @@ -48,6 +48,9 @@ MODE=${MODE:-"xnnpack+custom"}
48
48
# Default UPLOAD_DIR to empty string if not set
49
49
UPLOAD_DIR=" ${UPLOAD_DIR:- } "
50
50
51
+ # Default PT2E_QUANTIZE to empty string if not set
52
+ PT2E_QUANTIZE=" ${PT2E_QUANTIZE:- } "
53
+
51
54
if [[ $# -lt 4 ]]; then # Assuming 4 mandatory args
52
55
echo " Expecting atleast 4 positional arguments"
53
56
echo " Usage: [...]"
Original file line number Diff line number Diff line change @@ -368,6 +368,7 @@ jobs:
368
368
strategy :
369
369
matrix :
370
370
dtype : [fp32]
371
+ pt2e_quantize : [qnn_16a16w, qnn_8a8w]
371
372
mode : [qnn]
372
373
fail-fast : false
373
374
with :
@@ -384,6 +385,7 @@ jobs:
384
385
DTYPE=${{ matrix.dtype }}
385
386
BUILD_TOOL="cmake"
386
387
MODE=${{ matrix.mode }}
388
+ PT2E_QUANTIZE=${{ matrix.pt2e_quantize }}
387
389
388
390
PYTHON_EXECUTABLE=python bash .ci/scripts/setup-qnn-deps.sh
389
391
PYTHON_EXECUTABLE=python bash .ci/scripts/build-qnn-sdk.sh
@@ -393,7 +395,7 @@ jobs:
393
395
# Install requirements for export_llama
394
396
PYTHON_EXECUTABLE=python bash examples/models/llama/install_requirements.sh
395
397
# Test llama2
396
- PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh -model stories110M -build_tool "${BUILD_TOOL}" -dtype "${DTYPE}" -mode "${MODE }"
398
+ PYTHON_EXECUTABLE=python bash .ci/scripts/test_llama.sh -model stories110M -build_tool "${BUILD_TOOL}" -mode "${MODE}" - dtype "${DTYPE}" -pt2e_quantize "${PT2E_QUANTIZE }"
397
399
398
400
test-phi-3-mini-runner-linux :
399
401
name : test-phi-3-mini-runner-linux
You can’t perform that action at this time.
0 commit comments