File tree Expand file tree Collapse file tree 1 file changed +3
-2
lines changed
Expand file tree Collapse file tree 1 file changed +3
-2
lines changed Original file line number Diff line number Diff line change @@ -68,8 +68,8 @@ target_compile_options(
6868install (TARGETS custom_ops DESTINATION lib)
6969
7070if (EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
71- # Use NO_CMAKE_FIND_ROOT_PATH to make sure this works even if
72- # cross-compiling. See note in kernels/optimized/CMakeLists.txt.
71+ # Use NO_CMAKE_FIND_ROOT_PATH to make sure this works even if cross-compiling.
72+ # See note in kernels/optimized/CMakeLists.txt.
7373 find_package (Torch CONFIG REQUIRED NO_CMAKE_FIND_ROOT_PATH )
7474 add_library (
7575 custom_ops_aot_lib SHARED
@@ -84,6 +84,7 @@ if(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
8484 )
8585 target_include_directories (
8686 custom_ops_aot_lib PRIVATE "${CMAKE_CURRENT_BINARY_DIR} /../../../include"
87+ ${TORCH_INCLUDE_DIRS}
8788 )
8889 # TODO: This only works if we install portable_lib.so to
8990 # <site-packages>/executorch/extension/pybindings/.
You can’t perform that action at this time.
0 commit comments