Fix compile error

Abhinav Arora · Abhinav Arora · commit c08752ca43fe · 2018-04-20T18:35:50.000-07:00
diff --git a/paddle/fluid/framework/tensor_util_test.cu b/paddle/fluid/framework/tensor_util_test.cu
@@ -57,7 +57,8 @@ TEST(TensorContainsNAN, GPU) {
   }
   {
     Tensor tensor;
-    float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
+    paddle::platform::float16* buf =
+        tensor.mutable_data<paddle::platform::float16>({3}, gpu);
     FillNAN<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
     cuda_ctx->Wait();
     ASSERT_TRUE(TensorContainsNAN(tensor));
@@ -77,7 +78,8 @@ TEST(TensorContainsInf, GPU) {
   }
   {
     Tensor tensor;
-    float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
+    paddle::platform::float16* buf =
+        tensor.mutable_data<paddle::platform::float16>({3}, gpu);
     FillInf<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
     cuda_ctx->Wait();
     ASSERT_TRUE(TensorContainsInf(tensor));

Original file line number	Diff line number	Diff line change
`@@ -57,7 +57,8 @@ TEST(TensorContainsNAN, GPU) {`
`57`	`57`	`}`
`58`	`58`	`{`
`59`	`59`	`Tensor tensor;`
`60`		`- float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);`
	`60`	`+ paddle::platform::float16* buf =`
	`61`	`+ tensor.mutable_data<paddle::platform::float16>({3}, gpu);`
`61`	`62`	`FillNAN<<<1, 1, 0, cuda_ctx->stream()>>>(buf);`
`62`	`63`	`cuda_ctx->Wait();`
`63`	`64`	`ASSERT_TRUE(TensorContainsNAN(tensor));`
`@@ -77,7 +78,8 @@ TEST(TensorContainsInf, GPU) {`
`77`	`78`	`}`
`78`	`79`	`{`
`79`	`80`	`Tensor tensor;`
`80`		`- float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);`
	`81`	`+ paddle::platform::float16* buf =`
	`82`	`+ tensor.mutable_data<paddle::platform::float16>({3}, gpu);`
`81`	`83`	`FillInf<<<1, 1, 0, cuda_ctx->stream()>>>(buf);`
`82`	`84`	`cuda_ctx->Wait();`
`83`	`85`	`ASSERT_TRUE(TensorContainsInf(tensor));`