Fix compile error

c08752ca · Abhinav Arora · feaf168d · c08752ca
显示空白变更内容
内联并排

Showing with 4 addition and 2 deletion

paddle/fluid/framework/tensor_util_test.cu paddle/fluid/framework/tensor_util_test.cu +4 -2

未找到文件。
--- 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));