diff --git a/paddle/operators/math/math_function_test.cc b/paddle/operators/math/math_function_test.cc index d0f0acab917570a0c30c05145742c07cc24e1b9b..a7a6881a5cecb0c35f2f5e9932b9aa5b9a61b860 100644 --- a/paddle/operators/math/math_function_test.cc +++ b/paddle/operators/math/math_function_test.cc @@ -16,15 +16,14 @@ TEST(math_function, GPU) { auto* gpu_place = new paddle::platform::GPUPlace(0); paddle::platform::DeviceContext* context = - new paddle::platform::CUDADeviceContext(*gpu_place); + new paddle::platform::CUDADeviceContext(*gpu_place); input1_gpu.CopyFrom(input1, *gpu_place); input2_gpu.CopyFrom(input1, *gpu_place); out_gpu.CopyFrom(input1, *gpu_place); paddle::operators::math::matmul( - input1_gpu, false, input2_gpu, - false, 1, &out_gpu, 0, context); + input1_gpu, false, input2_gpu, false, 1, &out_gpu, 0, context); out.CopyFrom(out_gpu, *cpu_place); diff --git a/paddle/platform/dynload/cublas.h b/paddle/platform/dynload/cublas.h index 617866d17c02138b4340ffaa871492fff5c96fbb..6b00b2aa486076f024a6cb7794cefd082d1a3194 100644 --- a/paddle/platform/dynload/cublas.h +++ b/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_v2); \ - __macro(cublasDgemv_v2); \ - __macro(cublasSgemm_v2); \ - __macro(cublasDgemm_v2); \ - __macro(cublasSgeam_v2); \ - __macro(cublasDgeam_v2); \ + __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); \