saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix missing mkl-dnn include paths (#117492)

Browse source 
Fixes #91968 and #100960
This commit fixes missing  include paths by linking `caffe2_pybind11_state_gpu` against `caffe2::mkldnn`
Pull Request resolved: https://github.com/pytorch/pytorch/pull/117492
Approved by: https://github.com/ezyang
This commit is contained in:
nidefawl 2024-01-16 23:28:14 +00:00 committed by PyTorch MergeBot
parent 46a8408fa1
commit b1c3f9f1b9
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
caffe2/CMakeLists.txt
View file

@ -1971,6 +1971,9 @@ if(BUILD_PYTHON)
target_include_directories(caffe2_pybind11_state_gpu PRIVATE ${Caffe2_CPU_INCLUDE})
target_link_libraries(caffe2_pybind11_state_gpu PRIVATE
torch_library python::python pybind::pybind11)
if(USE_MKLDNN)
target_link_libraries(caffe2_pybind11_state_gpu PRIVATE caffe2::mkldnn)
endif()
if(WIN32)
target_link_libraries(caffe2_pybind11_state_gpu PRIVATE onnx_proto)
endif(WIN32)