Skip to content

Commit 7719d31

Browse files
authored
Mostly sync BlasKernel.cpp with ATen ReducedPrecisionGemvFastPathKernel (#10941)
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, lintrunner, and a couple of EXECUTORCH NOTEs. ghstack-source-id: 284287912 @exported-using-ghexport Differential Revision: [D74702689](https://our.internmc.facebook.com/intern/diff/D74702689/)
1 parent 9cce48d commit 7719d31

File tree

4 files changed

+274
-141
lines changed

4 files changed

+274
-141
lines changed

kernels/optimized/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -39,6 +39,7 @@ include(${EXECUTORCH_ROOT}/tools/cmake/Codegen.cmake)
3939
# Build cpublas.
4040
list(TRANSFORM _optimized_cpublas__srcs PREPEND "${EXECUTORCH_ROOT}/")
4141
add_library(cpublas STATIC ${_optimized_cpublas__srcs})
42+
target_include_directories(cpublas PRIVATE ${TORCH_INCLUDE_DIRS})
4243
target_link_libraries(
4344
cpublas PUBLIC executorch_core eigen_blas extension_threadpool
4445
)

0 commit comments

Comments
 (0)