experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

- fixing comments to reflect reality

Browse Source 
Change: 155256914
This commit is contained in:
A. Unique TensorFlower 2017-05-05 15:59:43 -08:00 committed by TensorFlower Gardener
parent 0632564172
commit fdb4eba5b1
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/common_runtime/gpu/gpu_device.cc
View File

@ -567,15 +567,14 @@ int64 MinSystemMemory(int64 available_memory) {
// We use the following heuristic for now: // We use the following heuristic for now:
// //
// If the available_memory is < 2GiB, we allocate 200MiB to system memory. // If the available_memory is < 2GiB, we allocate 200MiB to system memory.
// Otherwise, allocate 300MiB to system memory. // Otherwise, allocate max(300MiB, 0.05 * available_memory) to system memory.
// //
// In the future we could be more sophisticated by using a table of // In the future we could be more sophisticated by using a table of devices.
// devices.
if (available_memory < (1LL << 31)) { if (available_memory < (1LL << 31)) {
// 200MiB // 200MiB
return 209715200LL; return 209715200LL;
} else { } else {
// max(300 MiB, 0.95 * available_memory) // max(300 MiB, 0.05 * available_memory)
return std::max(314572800LL, static_cast<int64>(available_memory * 0.05)); return std::max(314572800LL, static_cast<int64>(available_memory * 0.05));
} }
} }