File tree Expand file tree Collapse file tree 4 files changed +6
-4
lines changed
examples/portable/custom_ops Expand file tree Collapse file tree 4 files changed +6
-4
lines changed Original file line number Diff line number Diff line change @@ -106,7 +106,7 @@ jobs:
106
106
uses : pytorch/test-infra/.github/workflows/mobile_job.yml@main
107
107
with :
108
108
device-type : android
109
- runner : ubuntu-latest
109
+ runner : linux.2xlarge
110
110
test-infra-ref : ' '
111
111
# This is the ARN of ExecuTorch project on AWS
112
112
project-arn : arn:aws:devicefarm:us-west-2:308535385114:project:02a2cf0f-6d9b-45ee-ba1a-a086587469e6
Original file line number Diff line number Diff line change @@ -553,7 +553,7 @@ endif()
553
553
# Install `executorch` library as well as `executorch-config.cmake` under
554
554
# ${CMAKE_INSTALL_PREFIX}/
555
555
install (
556
- TARGETS executorch executorch_no_prim_ops
556
+ TARGETS executorch executorch_no_prim_ops executorch_no_prim_ops_shared
557
557
DESTINATION lib
558
558
INCLUDES
559
559
DESTINATION ${_common_include_directories}
Original file line number Diff line number Diff line change 13
13
cmake_minimum_required (VERSION 3.19 )
14
14
15
15
set (_root "${CMAKE_CURRENT_LIST_DIR} /../.." )
16
- set (required_lib_list executorch executorch_no_prim_ops portable_kernels )
16
+ set (required_lib_list executorch executorch_no_prim_ops
17
+ executorch_no_prim_ops_shared portable_kernels
18
+ )
17
19
foreach (lib ${required_lib_list} )
18
20
set (lib_var "LIB_${lib} " )
19
21
add_library (${lib} STATIC IMPORTED )
Original file line number Diff line number Diff line change @@ -103,7 +103,7 @@ test_cmake_custom_op_2() {
103
103
104
104
EXT=$( get_shared_lib_ext)
105
105
echo " Exporting ${model_name} .pte"
106
- ${PYTHON_EXECUTABLE} -m " examples.portable.custom_ops.${model_name} " --so_library=" cmake-out/examples/portable/custom_ops/libcustom_ops_aot_lib$EXT "
106
+ DYLD_LIBRARY_PATH= " cmake-out/lib " ${PYTHON_EXECUTABLE} -m " examples.portable.custom_ops.${model_name} " --so_library=" cmake-out/examples/portable/custom_ops/libcustom_ops_aot_lib$EXT "
107
107
# should save file custom_ops_2.pte
108
108
109
109
echo ' Running custom_ops_executor_runner'
You can’t perform that action at this time.
0 commit comments