Skip to content

Commit ccafe18

Browse files
committed
Update on "Remove ExecuTorch copy of Vectorized"
All uses are outside ExecuTorch core, so we can just use ATen Vectorized. Differential Revision: [D66396016](https://our.internmc.facebook.com/intern/diff/D66396016/) [ghstack-poisoned]
2 parents 086820e + fd0f790 commit ccafe18

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

extension/llm/custom_ops/CMakeLists.txt

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,8 +68,8 @@ target_compile_options(
6868
install(TARGETS custom_ops DESTINATION lib)
6969

7070
if(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/.

0 commit comments

Comments
 (0)