pytorch
diff --git a/‎kernels/optimized/CMakeLists.txt
Lines changed: 1 addition & 0 deletions b/‎kernels/optimized/CMakeLists.txt
Lines changed: 1 addition & 0 deletions
@@ -42,6 +42,7 @@ endif()
 # Build cpublas.
 list(TRANSFORM _optimized_cpublas__srcs PREPEND "${EXECUTORCH_ROOT}/")
 add_library(cpublas STATIC ${_optimized_cpublas__srcs})
+target_include_directories(cpublas PRIVATE ${TORCH_INCLUDE_DIRS})
 target_link_libraries(
   cpublas PUBLIC executorch_core eigen_blas extension_threadpool
 )
Original file line number	Diff line number	Diff line change
`@@ -42,6 +42,7 @@ endif()`
`42`	`42`	`# Build cpublas.`
`43`	`43`	`list(TRANSFORM _optimized_cpublas__srcs PREPEND "${EXECUTORCH_ROOT}/")`
`44`	`44`	`add_library(cpublas STATIC ${_optimized_cpublas__srcs})`
	`45`	`+target_include_directories(cpublas PRIVATE ${TORCH_INCLUDE_DIRS})`
`45`	`46`	`target_link_libraries(`
`46`	`47`	`cpublas PUBLIC executorch_core eigen_blas extension_threadpool`
`47`	`48`	`)`