From ac4ac12488eb28d4a6ee9e66337c5a3b4214de96 Mon Sep 17 00:00:00 2001
From: Zingo Andersen <zingo.andersen@arm.com>
Date: Fri, 8 Nov 2024 12:11:28 +0100
Subject: [PATCH] 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 <zingo.andersen@arm.com>
Change-Id: I996573732b6d9e30948be1c0287b43d62875b171
---
 examples/arm/run.sh | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/examples/arm/run.sh b/examples/arm/run.sh
index e4941519449..c2c04cd2fd1 100755
--- a/examples/arm/run.sh
+++ b/examples/arm/run.sh
@@ -83,7 +83,7 @@ fi
 toolchain_cmake=${script_dir}/ethos-u-setup/arm-none-eabi-gcc.cmake
 _setup_msg="please refer to ${script_dir}/ethos-u-setup/setup.sh to properly install necessary tools."
 
-if ! [[ $portable_kernels =~ ^((^|,)aten::[a-zA-Z0-9_]+\.out)*$ ]]; then
+if ! [[ $portable_kernels =~ ^((^|,)aten::[a-zA-Z0-9_]+\.[a-zA-Z0-9_]*out)*$ ]]; then
     echo " ERROR: specified argument --portable_kernels=${portable_kernels}"
     echo "        is in the wrong format please use \"aten::<OP1>.out,aten::<OP2>.out,...\""
     echo "        e.g. \"aten::_softmax.out,aten::add.out\""