Skip to content

Commit 90fd962

Browse files
authored
Arm backend: Add dim_order_ops:: to the auto gen_oplist generations (#13455)
Improve ops checking and log message around the auto gen_oplist code Signed-off-by: Zingo Andersen <[email protected]>
1 parent 5ff0208 commit 90fd962

File tree

1 file changed

+22
-7
lines changed

1 file changed

+22
-7
lines changed

examples/arm/executor_runner/CMakeLists.txt

Lines changed: 22 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -173,23 +173,38 @@ execute_process(
173173
COMMAND
174174
python "${ET_DIR_PATH}/codegen/tools/gen_oplist.py"
175175
--model_file_path=${ET_PTE_FILE_PATH}
176-
--output_path=${CURRENT_BINARY_DIR}/temp.yaml
176+
--output_path=${CMAKE_CURRENT_BINARY_DIR}/temp.yaml
177177
OUTPUT_VARIABLE CMD_RESULT
178178
)
179-
if(NOT CMD_RESULT MATCHES "aten::")
180-
set(NO_OPS_IN_FILE "true")
179+
180+
if(CMD_RESULT MATCHES "aten::" OR CMD_RESULT MATCHES "dim_order_ops::")
181+
set(FOUND_OPS_IN_FILE "true")
181182
else()
182-
set(NO_OPS_IN_FILE "false")
183+
set(FOUND_OPS_IN_FILE "false")
183184
endif()
184185

185-
if(${SEMIHOSTING} OR ${ET_BUNDLE_IO})
186+
if(${SEMIHOSTING})
186187
set(EXECUTORCH_SELECT_OPS_MODEL "")
187-
elseif(${NO_OPS_IN_FILE})
188+
message(
189+
"gen_oplist: Building with semihosting, no model is used to auto generate ops from will use EXECUTORCH_SELECT_OPS_LIST=${EXECUTORCH_SELECT_OPS_LIST}"
190+
)
191+
elseif(${FOUND_OPS_IN_FILE})
188192
set(EXECUTORCH_SELECT_OPS_LIST "")
193+
set(EXECUTORCH_SELECT_OPS_MODEL "${ET_PTE_FILE_PATH}")
194+
message(
195+
"gen_oplist: EXECUTORCH_SELECT_OPS_MODEL=${ET_PTE_FILE_PATH} is used to auto generate ops from"
196+
)
197+
elseif(NOT ${FOUND_OPS_IN_FILE} AND ${ET_BUNDLE_IO})
189198
set(EXECUTORCH_SELECT_OPS_MODEL "")
199+
message(
200+
"gen_oplist: Building with ET_BUNDLE_IO and .bpte is not supported to auto generate ops from will use EXECUTORCH_SELECT_OPS_LIST=${EXECUTORCH_SELECT_OPS_LIST}"
201+
)
190202
else()
191203
set(EXECUTORCH_SELECT_OPS_LIST "")
192-
set(EXECUTORCH_SELECT_OPS_MODEL "${ET_PTE_FILE_PATH}")
204+
set(EXECUTORCH_SELECT_OPS_MODEL "")
205+
message(
206+
"gen_oplist: No non delagated ops was found in ${ET_PTE_FILE_PATH} no ops added to build"
207+
)
193208
endif()
194209

195210
# Ensure that either executorch_select_ops_list or executorch_select_ops_model

0 commit comments

Comments
 (0)