Skip to content

P
Paddle

Crayon鑫 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 bf1a7018 编写于 作者: Z zhouwei25
浏览文件
操作

test=develop

上级 0d986b8f
显示空白变更内容
内联 并排
Showing with 9 addition and 9 deletion
paddle/fluid/platform/gpu_info.cc
浏览文件 @ bf1a7018
...@@ -109,15 +109,15 @@ int GetCUDAComputeCapability(int id) { ...@@ -109,15 +109,15 @@ 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);
char ErrorLog[50]; std::string ErrorLog;
snprintf(ErrorLog, sizeof(ErrorLog), std::ostringstream ostr;
"cudaGetDeviceProperties failed in" ostr << "cudaGetDeviceProperties failed in"
"paddle::platform::GetCUDAComputeCapability!" "paddle::platform::GetCUDAComputeCapability!"
"Error Type ID = %d , Detail:" "Error Type ID = " << e << " Please see detail in:"
"https://docs.nvidia.com/cuda/cuda-runtime-api/" "https://docs.nvidia.com/cuda/cuda-runtime-api/"
"group__CUDART__TYPES.html#group__CUDART__TYPES_" "group__CUDART__TYPES.html#group__CUDART__TYPES_"
"1g3f51e3575c2178246db0a94a430e0038", "1g3f51e3575c2178246db0a94a430e0038");
e); ErrorLog = ostr.str();
PADDLE_ENFORCE(e, ErrorLog); PADDLE_ENFORCE(e, ErrorLog);
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.
先完成此消息的编辑！
想要评论请 注册 或