Skip to content

P
Paddle

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

通知 2299
Star 20931
Fork 5422
P
Paddle
提交 48d9e968 编写于 作者: Z zhouwei25
浏览文件
操作

test=develop

上级 fbe69293
隐藏空白更改
内联 并排
Showing with 3 addition and 7 deletion
paddle/fluid/platform/gpu_info.cc
浏览文件 @ 48d9e968
...@@ -109,13 +109,9 @@ int GetCUDAComputeCapability(int id) { ...@@ -109,13 +109,9 @@ int GetCUDAComputeCapability(int id) {
PADDLE_ENFORCE_LT(id, GetCUDADeviceCount(), "id must less than GPU count"); PADDLE_ENFORCE_LT(id, GetCUDADeviceCount(), "id must less than GPU count");
cudaDeviceProp device_prop; cudaDeviceProp device_prop;
auto e = cudaGetDeviceProperties(&device_prop, id); auto e = cudaGetDeviceProperties(&device_prop, id);
PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), PADDLE_ENFORCE(
"cudaGetDeviceProperties failed in " cudaGetDeviceProperties(&device_prop, id),
"paddle::platform::GetCUDAComputeCapability!" "cudaGetDeviceProperties failedp in cudaGetDeviceProperties failed in");
"Error Type ID ="
"https://docs.nvidia.com/cuda/cuda-runtime-api/"
"group__CUDART__TYPES.html#group__CUDART__TYPES_"
"1g3f51e3575c2178246db0a94a430e0038");
return device_prop.major * 10 + device_prop.minor; return device_prop.major * 10 + device_prop.minor;
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或