From 83d827682349c34f53a5dbb3dd689b44bedff68b Mon Sep 17 00:00:00 2001 From: zhouwei25 Date: Fri, 12 Jul 2019 15:35:06 +0000 Subject: [PATCH] test=develop --- paddle/fluid/platform/gpu_info.cc | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/platform/gpu_info.cc b/paddle/fluid/platform/gpu_info.cc index 69204f45c67..7074cf2b1fb 100644 --- a/paddle/fluid/platform/gpu_info.cc +++ b/paddle/fluid/platform/gpu_info.cc @@ -110,7 +110,8 @@ int GetCUDAComputeCapability(int id) { cudaDeviceProp device_prop; auto e = cudaGetDeviceProperties(&device_prop, id); int a = 10; - PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), e); + char c = '1'; + PADDLE_ENFORCE(cudaGetDeviceProperties(&device_prop, id), a, c); return device_prop.major * 10 + device_prop.minor; } -- GitLab