@@ -78,7 +78,7 @@ void testVol2col() {
7878 if (paddle::platform::is_cpu_place (*place)) {
7979 input = input_tmp;
8080 } else {
81- input.CopyFrom <float >(input_tmp, *place);
81+ input.CopyFrom <float >(input_tmp, *place, *context );
8282 }
8383 output.mutable_data <float >({1 , filter_size, filter_size, filter_size,
8484 output_depth, output_height, output_width},
@@ -93,7 +93,7 @@ void testVol2col() {
9393 if (paddle::platform::is_cpu_place (*place)) {
9494 out_cfo_ptr = output.data <float >();
9595 } else {
96- output_tmp.CopyFrom <float >(output, paddle::platform::CPUPlace ());
96+ output_tmp.CopyFrom <float >(output, paddle::platform::CPUPlace (), *context );
9797 out_cfo_ptr = output_tmp.data <float >();
9898 }
9999
@@ -107,7 +107,7 @@ void testVol2col() {
107107 if (paddle::platform::is_cpu_place (*place)) {
108108 input = input_tmp;
109109 } else {
110- input.CopyFrom <float >(input_tmp, *place);
110+ input.CopyFrom <float >(input_tmp, *place, *context );
111111 }
112112
113113 paddle::operators::math::Col2VolFunctor<Place, float > col2vol;
@@ -118,7 +118,7 @@ void testVol2col() {
118118 if (paddle::platform::is_cpu_place (*place)) {
119119 in_ptr = input.data <float >();
120120 } else {
121- input_tmp.CopyFrom <float >(input, paddle::platform::CPUPlace ());
121+ input_tmp.CopyFrom <float >(input, paddle::platform::CPUPlace (), *context );
122122 in_ptr = input_tmp.data <float >();
123123 }
124124
0 commit comments