提交 5852ce14 编写于 作者: Z zhouwei25

test=develop

上级 48d9e968
......@@ -109,9 +109,8 @@ int GetCUDAComputeCapability(int id) {
PADDLE_ENFORCE_LT(id, GetCUDADeviceCount(), "id must less than GPU count");
cudaDeviceProp device_prop;
auto e = cudaGetDeviceProperties(&device_prop, id);
PADDLE_ENFORCE(
cudaGetDeviceProperties(&device_prop, id),
"cudaGetDeviceProperties failedp in cudaGetDeviceProperties failed in");
int a = 10;
PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), "aaa", a);
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.
先完成此消息的编辑!
想要评论请 注册