m53297601
/
Paddle
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Fiux compile error

Browse Source
wangkuiyi-patch-2
Abhinav Arora 7 years ago
parent b6ec45d8da
commit feaf168d95
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
paddle/fluid/framework/tensor_util_test.cu
Unescape View File

@ -57,7 +57,7 @@ TEST(TensorContainsNAN, GPU) {
} }
{ {
Tensor tensor; Tensor tensor;
float16* buf = tensor.mutable_data<float16>({3}, gpu); float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
FillNAN<<<1, 1, 0, cuda_ctx->stream()>>>(buf); FillNAN<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
cuda_ctx->Wait(); cuda_ctx->Wait();
ASSERT_TRUE(TensorContainsNAN(tensor)); ASSERT_TRUE(TensorContainsNAN(tensor));
@ -77,7 +77,7 @@ TEST(TensorContainsInf, GPU) {
} }
{ {
Tensor tensor; Tensor tensor;
float16* buf = tensor.mutable_data<float16>({3}, gpu); float16* buf = tensor.mutable_data<paddle::platform::float16>({3}, gpu);
FillInf<<<1, 1, 0, cuda_ctx->stream()>>>(buf); FillInf<<<1, 1, 0, cuda_ctx->stream()>>>(buf);
cuda_ctx->Wait(); cuda_ctx->Wait();
ASSERT_TRUE(TensorContainsInf(tensor)); ASSERT_TRUE(TensorContainsInf(tensor));

Write Preview
Loading…
Cancel
Save