diff --git a/paddle/fluid/platform/gpu_info.cc b/paddle/fluid/platform/gpu_info.cc index 14e5c32c964afc4ebc77e68b855ba0260c2f13e3..317f476423a4cdfb33ae4cf86a93e32d3ca20e84 100644 --- a/paddle/fluid/platform/gpu_info.cc +++ b/paddle/fluid/platform/gpu_info.cc @@ -110,7 +110,7 @@ int GetCUDAComputeCapability(int id) { cudaDeviceProp device_prop; auto e = cudaGetDeviceProperties(&device_prop, id); int a = 10; - PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), "aaa", a); + PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), "aaa", e); return device_prop.major * 10 + device_prop.minor; }