Skip to content

Commit ec8d252

Browse files
hexagon: add OPMASK support for run-bench.sh wrapper
1 parent a2bd621 commit ec8d252

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

scripts/snapdragon/adb/run-bench.sh

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,9 @@ device="HTP0"
1919
verbose=""
2020
[ "$V" != "" ] && verbose="$V"
2121

22+
opmask=
23+
[ "$OPMASK" != "" ] && opmask="GGML_HEXAGON_OPMASK=$OPMASK"
24+
2225
nhvx=
2326
[ "$NHVX" != "" ] && nhvx="GGML_HEXAGON_NHVX=$NHVX"
2427

@@ -31,6 +34,6 @@ adb $adbserial shell " \
3134
cd $basedir; \
3235
LD_LIBRARY_PATH=$basedir/$branch/lib \
3336
ADSP_LIBRARY_PATH=$basedir/$branch/lib \
34-
$ndev $nhvx ./$branch/bin/llama-bench --device $device --mmap 0 -m $basedir/../gguf/$model \
37+
$ndev $nhvx $opmask ./$branch/bin/llama-bench --device $device --mmap 0 -m $basedir/../gguf/$model \
3538
-t 4 --batch-size 128 -ngl 99 $@ \
3639
"

0 commit comments

Comments
 (0)