Skip to content

Commit 270d055

Browse files
zingofreddan80
authored andcommitted
Arm backend: Make run.sh handle portable ops using aten::<OP>.<modifier>out
This is needed for squeeze e.g. --portable_kernels="aten::squeeze_copy.dims_out" Signed-off-by: Zingo Andersen <[email protected]> Change-Id: I996573732b6d9e30948be1c0287b43d62875b171
1 parent 08ed54e commit 270d055

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

examples/arm/run.sh

+1-1
Original file line numberDiff line numberDiff line change
@@ -83,7 +83,7 @@ fi
8383
toolchain_cmake=${script_dir}/ethos-u-setup/arm-none-eabi-gcc.cmake
8484
_setup_msg="please refer to ${script_dir}/ethos-u-setup/setup.sh to properly install necessary tools."
8585

86-
if ! [[ $portable_kernels =~ ^((^|,)aten::[a-zA-Z0-9_]+\.out)*$ ]]; then
86+
if ! [[ $portable_kernels =~ ^((^|,)aten::[a-zA-Z0-9_]+\.[a-zA-Z0-9_]*out)*$ ]]; then
8787
echo " ERROR: specified argument --portable_kernels=${portable_kernels}"
8888
echo " is in the wrong format please use \"aten::<OP1>.out,aten::<OP2>.out,...\""
8989
echo " e.g. \"aten::_softmax.out,aten::add.out\""

0 commit comments

Comments
 (0)