fix bug in dynload

paddle/operators/math/CMakeLists.txt

@@ -13,4 +13,4 @@ else()
 endif()    
 
 
-nv_test(math_function_test SRCS math_function_test.cc DEPS math_function)
+nv_test(math_function_test SRCS math_function_test.cc DEPS math_function tensor)

paddle/operators/math/math_function_test.cc

@@ -12,16 +12,19 @@ TEST(math_function, GPU) {
   auto* cpu_place = new paddle::platform::CPUPlace();
   float* input1_ptr = input1.mutable_data<float>({2, 2}, *cpu_place);
   float arr[4] = {0, 1, 2, 3};
+  memcpy(input1_ptr, arr, 4 * sizeof(int));
 
   auto* gpu_place = new paddle::platform::GPUPlace(0);
-  paddle::platform::DeviceContext* context = new CUDADeviceContext(gpu_place);
+  paddle::platform::DeviceContext* context =
+    new paddle::platform::CUDADeviceContext(*gpu_place);
 
   input1_gpu.CopyFrom<float>(input1, *gpu_place);
   input2_gpu.CopyFrom<float>(input1, *gpu_place);
   out_gpu.CopyFrom<float>(input1, *gpu_place);
 
-  matmul<paddle::platform::GPUPlace, float>(input1_gpu, false, input2_gpu,
-                                            false, 1, &out_gpu, 0, context);
+  paddle::operators::math::matmul<paddle::platform::GPUPlace, float>(
+            input1_gpu, false, input2_gpu,
+            false, 1, &out_gpu, 0, context);
 
   out.CopyFrom<float>(out_gpu, *cpu_place);
 
@@ -31,4 +34,4 @@ TEST(math_function, GPU) {
   EXPECT_EQ(out_ptr[2], 6);
   EXPECT_EQ(out_ptr[3], 11);
 }
-#endif
+#endif

paddle/platform/dynload/cublas.h

@@ -62,12 +62,12 @@ extern void *cublas_dso_handle;
   DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(__name)
 
 #define CUBLAS_BLAS_ROUTINE_EACH(__macro) \
-  __macro(cublasSgemv);                   \
-  __macro(cublasDgemv);                   \
-  __macro(cublasSgemm);                   \
-  __macro(cublasDgemm);                   \
-  __macro(cublasSgeam);                   \
-  __macro(cublasDgeam);                   \
+  __macro(cublasSgemv_v2);                   \
+  __macro(cublasDgemv_v2);                   \
+  __macro(cublasSgemm_v2);                   \
+  __macro(cublasDgemm_v2);                   \
+  __macro(cublasSgeam_v2);                   \
+  __macro(cublasDgeam_v2);                   \
   __macro(cublasCreate_v2);               \
   __macro(cublasDestroy_v2);              \
   __macro(cublasSetStream_v2);            \