Skip to content

Commit 73ccca7

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 4a3726e + d7e5c4a commit 73ccca7

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

extension/llm/custom_ops/CMakeLists.txt

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,11 @@ else()
5454
endif()
5555

5656
add_library(custom_ops ${_custom_ops__srcs})
57-
57+
find_package_torch_headers()
5858
target_include_directories(custom_ops PUBLIC "${_common_include_directories}")
5959
target_include_directories(
6060
custom_ops PRIVATE "${CMAKE_CURRENT_BINARY_DIR}/../../../include"
61+
${TORCH_INCLUDE_DIRS}
6162
)
6263
target_link_libraries(custom_ops PUBLIC ${custom_ops_libs} executorch_core)
6364

@@ -68,7 +69,6 @@ target_compile_options(
6869
install(TARGETS custom_ops DESTINATION lib)
6970

7071
if(EXECUTORCH_BUILD_KERNELS_CUSTOM_AOT)
71-
find_package_torch_headers()
7272
add_library(
7373
custom_ops_aot_lib SHARED
7474
${_custom_ops__srcs}

0 commit comments

Comments
 (0)