From 6b12c697ff3e2a86e555fafa53ab5b1017e982ce Mon Sep 17 00:00:00 2001
From: qijun <qijun1994@hotmail.com>
Date: Mon, 7 Aug 2017 14:35:50 +0800
Subject: [PATCH] handle mkl

---
 paddle/operators/math/CMakeLists.txt | 12 ++++++++++--
 1 file changed, 10 insertions(+), 2 deletions(-)

diff --git a/paddle/operators/math/CMakeLists.txt b/paddle/operators/math/CMakeLists.txt
index 586347668ef..d34bc92594d 100644
--- 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()    
-- 
GitLab