fix std::min type in nan_inf, test=develop (#21725)

paddle_tiny_install
WangXi 6 years ago committed by gongweibao
parent 549f24b5f1
commit 8754cbd1f2

@ -167,7 +167,9 @@ void TensorCheckerVisitor<platform::CUDADeviceContext>::apply(
}
const size_t threads = 1024;
size_t blocks = std::min(128ul, (tensor_.numel() + threads - 1) / threads);
size_t blocks =
std::min(static_cast<size_t>(128),
static_cast<size_t>((tensor_.numel() + threads - 1) / threads));
CheckNanInfKernel<<<blocks, threads, 0, dev_ctx->stream()>>>(
tensor_.data<T>(), tensor_.numel(), print_num, gpu_str_ptr);
}

Loading…
Cancel
Save