handle mkl

6b12c697 · qijun · 08159359 · 6b12c697
显示空白变更内容
内联并排

Showing with 10 addition and 2 deletion

paddle/operators/math/CMakeLists.txt paddle/operators/math/CMakeLists.txt +10 -2

未找到文件。
--- a/paddle/operators/math/CMakeLists.txt
+++ b/paddle/operators/math/CMakeLists.txt
 if (WITH_GPU)
+    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()
+    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()