test=develop

83d82768 · zhouwei25 · a53ac06a · 83d82768
隐藏空白更改
内联并排

Showing with 2 addition and 1 deletion

paddle/fluid/platform/gpu_info.cc paddle/fluid/platform/gpu_info.cc +2 -1

未找到文件。
--- 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;
 }