m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

handle mkl

Browse Source
revert-3824-remove_grad_op_type
qijun 8 years ago
parent 0815935916
commit 6b12c697ff
1 changed files with 10 additions and 2 deletions
Show Stats Download Patch File Download Diff File

12
paddle/operators/math/CMakeLists.txt
Unescape View File

@ -1,5 +1,13 @@
if (WITH_GPU) if (WITH_GPU)
nv_library(math_function SRCS math_function.cc math_function.cu DEPS cblas device_context) if (WITH_MKLML)
nv_library(math_function SRCS math_function.cc math_function.cu DEPS mklml device_context)
else()
nv_library(math_function SRCS math_function.cc math_function.cu DEPS cblas device_context)
endif()
else() else()
cc_library(math_function SRCS math_function.cc DEPS cblas device_context) if (WITH_MKLML)
cc_library(math_function SRCS math_function.cc DEPS mklml device_context)
else()
cc_library(math_function SRCS math_function.cc DEPS cblas device_context)
endif()
endif() endif()

Write Preview
Loading…
Cancel
Save