Skip to content

Commit 510ba74

Browse files
Benjamin CheungBenjamin Cheung
authored andcommitted
fix: add more configurations
1 parent 26cb4b8 commit 510ba74

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

.ci/scripts/test_llama.sh

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -154,12 +154,20 @@ case "${PLATFORM}:${DTYPE}:${MODE}" in
154154
"x86:fp32:portable") ACT_EXPORT_TIME=72 ;;
155155
"x86:fp32:xnnpack+custom") ACT_EXPORT_TIME=276 ;;
156156
"x86:bf16:portable") ACT_EXPORT_TIME=75 ;;
157+
"x86:bf16:custom") ACT_EXPORT_TIME=65 ;;
158+
"x86:fp32:xnnpack+custom+qe") ACT_EXPORT_TIME=285 ;;
159+
"x86:fp32:xnnpack+custom+quantize_kv") ACT_EXPORT_TIME=295 ;;
160+
"x86:fp32:xnnpack+quantize_kv") ACT_EXPORT_TIME=356 ;;
161+
"x86:fp32:qnn") ACT_EXPORT_TIME=334 ;;
157162

158163
# Linux ARM64 configurations
159164
"arm64:fp32:portable") ACT_EXPORT_TIME=124 ;;
160165
"arm64:fp32:xnnpack+custom") ACT_EXPORT_TIME=483 ;;
161166
"arm64:bf16:portable") ACT_EXPORT_TIME=118 ;;
162167
"arm64:bf16:custom") ACT_EXPORT_TIME=102 ;;
168+
"arm64:fp32:xnnpack+custom+qe") ACT_EXPORT_TIME=486 ;;
169+
"arm64:fp32:xnnpack+custom+quantize_kv") ACT_EXPORT_TIME=521 ;;
170+
"arm64:fp32:xnnpack+quantize_kv") ACT_EXPORT_TIME=514 ;;
163171

164172
# macOS configurations
165173
"macos:fp32:mps") ACT_EXPORT_TIME=30 ;;

0 commit comments

Comments
 (0)