handle mkl

8 years ago · 6b12c697ff
parent 0815935916
commit 6b12c697ff
1 changed files with 10 additions and 2 deletions
--- a/paddle/operators/math/CMakeLists.txt
+++ b/paddle/operators/math/CMakeLists.txt
@ -1,5 +1,13 @@
 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()
-    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()