mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
86b86202b5
Fixes https://github.com/pytorch/pytorch/issues/74949, which reports that torch.config can't respect USE_MKLDNN flag. Pull Request resolved: https://github.com/pytorch/pytorch/pull/75001 Approved by: https://github.com/malfet
21 lines
697 B
CMake
21 lines
697 B
CMake
if(USE_MKLDNN)
|
|
message(STATUS "Including IDEEP operators")
|
|
|
|
# ---[ CPU files.
|
|
file(GLOB_RECURSE tmp *.cc)
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} ${tmp})
|
|
# exclude test files and gpu files
|
|
file(GLOB_RECURSE tmp *_test.cc)
|
|
exclude(Caffe2_CPU_SRCS "${Caffe2_CPU_SRCS}" ${tmp})
|
|
|
|
# ---[ CPU test files - currently none but just to be safe
|
|
file(GLOB_RECURSE tmp *_test.cc)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} ${tmp})
|
|
|
|
# ---[ Send the lists to the parent scope.
|
|
set(Caffe2_CPU_SRCS ${Caffe2_CPU_SRCS} PARENT_SCOPE)
|
|
set(Caffe2_CPU_TEST_SRCS ${Caffe2_CPU_TEST_SRCS} PARENT_SCOPE)
|
|
else()
|
|
message(STATUS "Excluding ideep operators as we are not using ideep")
|
|
endif()
|