Skip to content

Commit feaf168

Browse files
author
Abhinav Arora
committed
Fiux compile error
1 parent b6ec45d commit feaf168

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

paddle/fluid/framework/tensor_util_test.cu

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ TEST(TensorContainsNAN, GPU) {
5757
}
5858
{
5959
Tensor tensor;
60-
float16* buf = tensor.mutable_data<float16>({3}, gpu);
60+
float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
6161
FillNAN<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
6262
cuda_ctx->Wait();
6363
ASSERT_TRUE(TensorContainsNAN(tensor));
@@ -77,7 +77,7 @@ TEST(TensorContainsInf, GPU) {
7777
}
7878
{
7979
Tensor tensor;
80-
float16* buf = tensor.mutable_data<float16>({3}, gpu);
80+
float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
8181
FillInf<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
8282
cuda_ctx->Wait();
8383
ASSERT_TRUE(TensorContainsInf(tensor));

0 commit comments

Comments
 (0)