Skip to content

Commit dbfb7f3

Browse files
committed
Build DNNL as a static library and IPEX statically links it.
1 parent 25e4690 commit dbfb7f3

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

cmake/CPU.cmake

+4-2
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ set(CMAKE_EXPORT_COMPILE_COMMANDS ON)
88
SET(DNNL_BUILD_TESTS FALSE CACHE BOOL "" FORCE)
99
SET(DNNL_BUILD_EXAMPLES FALSE CACHE BOOL "" FORCE)
1010
SET(DNNL_ENABLE_PRIMITIVE_CACHE TRUE CACHE BOOL "" FORCE)
11-
# SET(DNNL_LIBRARY_TYPE STATIC CACHE STRING "" FORCE)
11+
SET(DNNL_LIBRARY_TYPE STATIC CACHE STRING "" FORCE)
1212

1313
set(DPCPP_CPU_ROOT "${PROJECT_SOURCE_DIR}/torch_ipex/csrc/cpu")
1414
add_subdirectory(${DPCPP_THIRD_PARTY_ROOT}/mkl-dnn)
@@ -175,9 +175,11 @@ else()
175175
message(FATAL_ERROR "Unknown ATen parallel backend: ${ATEN_THREADING}")
176176
endif()
177177

178-
target_link_libraries(${PLUGIN_NAME} PUBLIC dnnl)
179178
add_dependencies(${PLUGIN_NAME} pybind11)
179+
180180
add_dependencies(${PLUGIN_NAME} dnnl)
181+
target_link_libraries(${PLUGIN_NAME} PUBLIC ${PROJECT_SOURCE_DIR}/build/third_party/mkl-dnn/src/libdnnl.a)
182+
181183
add_dependencies(${PLUGIN_NAME} xsmm)
182184

183185
link_directories(${PYTORCH_INSTALL_DIR}/lib)

0 commit comments

Comments
 (0)