|
|
@ -23,8 +23,7 @@ TEST(math_function, GPU) {
|
|
|
|
out_gpu.CopyFrom<float>(input1, *gpu_place);
|
|
|
|
out_gpu.CopyFrom<float>(input1, *gpu_place);
|
|
|
|
|
|
|
|
|
|
|
|
paddle::operators::math::matmul<paddle::platform::GPUPlace, float>(
|
|
|
|
paddle::operators::math::matmul<paddle::platform::GPUPlace, float>(
|
|
|
|
input1_gpu, false, input2_gpu,
|
|
|
|
input1_gpu, false, input2_gpu, false, 1, &out_gpu, 0, context);
|
|
|
|
false, 1, &out_gpu, 0, context);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
out.CopyFrom<float>(out_gpu, *cpu_place);
|
|
|
|
out.CopyFrom<float>(out_gpu, *cpu_place);
|
|
|
|
|
|
|
|
|
|
|
|