Skip to content

Commit

Permalink
follow comments
Browse files Browse the repository at this point in the history
  • Loading branch information
gangliao committed Aug 18, 2017
1 parent 83d0016 commit b3ab15a
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion paddle/platform/gpu_info.h
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,8 @@ namespace paddle {
namespace platform {

//! Environment variable: fraction of GPU memory to use on each device.
const std::string kEnvFractionGpuMemoryToUse = "FRACTION_GPU_MEMORY_TO_USE";
const std::string kEnvFractionGpuMemoryToUse =
"PADDLE_FRACTION_GPU_MEMORY_TO_USE";

//! Get the total number of GPU devices in system.
int GetDeviceCount();
Expand Down

0 comments on commit b3ab15a

Please sign in to comment.