Skip to content

Commit aa01ff8

Browse files
committed
Mostly sync BlasKernel.cpp with ATen ReducedPrecisionGemvFastPathKernel
Pull Request resolved: #10868 The two files were similar, but diverged due to recent changes. Since we have sharing of PyTorch headers, we can keep them mostly the same; differences are some of the namespace stuff and a couple of EXECUTORCH NOTEs. ghstack-source-id: 284080528 @exported-using-ghexport Differential Revision: [D74702689](https://our.internmc.facebook.com/intern/diff/D74702689/)
1 parent 9ded0a2 commit aa01ff8

File tree

3 files changed

+260
-150
lines changed

3 files changed

+260
-150
lines changed

kernels/optimized/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ endif()
4242
# Build cpublas.
4343
list(TRANSFORM _optimized_cpublas__srcs PREPEND "${EXECUTORCH_ROOT}/")
4444
add_library(cpublas STATIC ${_optimized_cpublas__srcs})
45+
target_include_directories(cpublas PRIVATE ${TORCH_INCLUDE_DIRS})
4546
target_link_libraries(
4647
cpublas PUBLIC executorch_core eigen_blas extension_threadpool
4748
)

0 commit comments

Comments
 (0)