Skip to content

P
PaddleDetection

PaddlePaddle / PaddleDetection
大约 1 年 前同步成功

通知 694
Star 11112
Fork 2696
P
PaddleDetection

体验新版 GitCode，发现更多精彩内容 >>

提交 c08752ca 编写于 作者: A Abhinav Arora
浏览文件
操作

Fix compile error

上级 feaf168d
隐藏空白更改
内联 并排
Showing with 4 addition and 2 deletion
paddle/fluid/framework/tensor_util_test.cu
浏览文件 @ c08752ca
......@@ -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));
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或