From 08c987d7c086e4176a27f2685712bbb9226e635e Mon Sep 17 00:00:00 2001
From: qijun <qijun1994@hotmail.com>
Date: Mon, 21 Aug 2017 17:23:15 +0800
Subject: [PATCH] use dynload curand

---
 paddle/operators/gaussian_random_op.h  | 4 ++--
 paddle/operators/math/math_function.cu | 8 ++++----
 paddle/operators/uniform_random_op.h   | 4 ++--
 3 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/paddle/operators/gaussian_random_op.h b/paddle/operators/gaussian_random_op.h
index 041390e954..c90b665fe0 100644
--- a/paddle/operators/gaussian_random_op.h
+++ b/paddle/operators/gaussian_random_op.h
@@ -34,5 +34,5 @@ class GaussianRandomKernel : public framework::OpKernel {
     math::RandGaussian<Place, T>(n, mean, std, data, device_context);
   }
 };
-}
-}
+}  // namespace operators
+}  // namespace paddle
diff --git a/paddle/operators/math/math_function.cu b/paddle/operators/math/math_function.cu
index 3ff622f308..908efe9e0f 100644
--- a/paddle/operators/math/math_function.cu
+++ b/paddle/operators/math/math_function.cu
@@ -149,8 +149,8 @@ void RandUniform<platform::GPUPlace, float>(const int n, const float min,
                                             const float max, float* output,
                                             platform::DeviceContext* context) {
   auto* cuda_context = reinterpret_cast<platform::CUDADeviceContext*>(context);
-  PADDLE_ENFORCE(
-      curandGenerateUniform(cuda_context->curand_generator(), output, n));
+  PADDLE_ENFORCE(platform::dynload::curandGenerateUniform(
+      cuda_context->curand_generator(), output, n));
   int block = 512;
   int grid = (n + block - 1) / block;
   UniformShift<float><<<grid, block, 0, cuda_context->stream()>>>(n, min, max,
@@ -179,8 +179,8 @@ void RandGaussian<platform::GPUPlace, float>(const int n, const float mean,
 
   const int even_n =
       HandleOddLengthRandGaussian<float>(n, mean, std, output, cuda_context);
-  PADDLE_ENFORCE(curandGenerateNormal(cuda_context->curand_generator(), output,
-                                      even_n, mean, std));
+  PADDLE_ENFORCE(platform::dynload::curandGenerateNormal(
+      cuda_context->curand_generator(), output, even_n, mean, std));
 }
 
 }  // namespace math
diff --git a/paddle/operators/uniform_random_op.h b/paddle/operators/uniform_random_op.h
index ec009b025e..dffa640f84 100644
--- a/paddle/operators/uniform_random_op.h
+++ b/paddle/operators/uniform_random_op.h
@@ -34,5 +34,5 @@ class UniformRandomKernel : public framework::OpKernel {
     math::RandUniform<Place, T>(n, min, max, data, device_context);
   }
 };
-}
-}
+}  // namespace operators
+}  // namespace paddle
-- 
GitLab