File tree Expand file tree Collapse file tree 3 files changed +7
-3
lines changed
examples/models/llama2/custom_ops Expand file tree Collapse file tree 3 files changed +7
-3
lines changed Original file line number Diff line number Diff line change @@ -275,7 +275,7 @@ filters = [
275
275
excludes = [
276
276
" ^codegen" ,
277
277
# Exclude blas, since it's built as a separate target.
278
- # "^kernels/optimized/blas",
278
+ " ^kernels/optimized/blas" ,
279
279
]
280
280
deps = [
281
281
" executorch" ,
Original file line number Diff line number Diff line change @@ -49,6 +49,8 @@ set(_common_include_directories ${EXECUTORCH_ROOT}/..)
49
49
set (custom_ops_libs extension_module )
50
50
list (APPEND custom_ops_libs pthreadpool )
51
51
list (APPEND custom_ops_libs cpuinfo )
52
+ list (APPEND custom_ops_libs cpublas )
53
+ list (APPEND custom_ops_libs eigen_blas )
52
54
53
55
# Generate C++ bindings to register kernels into both PyTorch (for AOT) and
54
56
# Executorch (for runtime). Here select all ops in optimized.yaml
@@ -65,7 +67,7 @@ add_library(custom_ops ${_custom_ops__srcs})
65
67
target_include_directories (custom_ops PUBLIC "${_common_include_directories} " )
66
68
target_include_directories (custom_ops PRIVATE
67
69
"${CMAKE_CURRENT_BINARY_DIR} /../../../../include" )
68
- target_link_libraries (custom_ops PRIVATE ${custom_ops_libs} )
70
+ target_link_libraries (custom_ops PUBLIC ${custom_ops_libs} )
69
71
70
72
target_compile_options (custom_ops PUBLIC ${_common_compile_options}
71
73
-DET_USE_THREADPOOL )
Original file line number Diff line number Diff line change @@ -46,7 +46,7 @@ if(NOT PYTHON_EXECUTABLE)
46
46
endif ()
47
47
# Build cpublas.
48
48
list (TRANSFORM _optimized_cpublas__srcs PREPEND "${EXECUTORCH_ROOT} /" )
49
- add_library (cpublas ${_optimized_cpublas__srcs} )
49
+ add_library (cpublas STATIC ${_optimized_cpublas__srcs} )
50
50
target_link_libraries (cpublas PRIVATE executorch eigen_blas )
51
51
target_compile_options (cpublas PUBLIC ${_common_compile_options} )
52
52
@@ -72,3 +72,5 @@ gen_operators_lib(
72
72
DEPS executorch )
73
73
74
74
install (TARGETS cpublas optimized_kernels optimized_ops_lib DESTINATION lib )
75
+
76
+ install (TARGETS cpublas DESTINATION lib )
You can’t perform that action at this time.
0 commit comments