diff options
Diffstat (limited to 'tensorflow/core/common_runtime/gpu/gpu_device.cc')
| -rw-r--r-- | tensorflow/core/common_runtime/gpu/gpu_device.cc | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/gpu/gpu_device.cc b/tensorflow/core/common_runtime/gpu/gpu_device.cc index 1390810c28..f7b733aa00 100644 --- a/tensorflow/core/common_runtime/gpu/gpu_device.cc +++ b/tensorflow/core/common_runtime/gpu/gpu_device.cc @@ -762,6 +762,10 @@ int64 MinSystemMemory(int64 available_memory) { // is necessary. min_system_memory *= 2; #endif +#if defined(NVIDIA_TEGRA) + // 1GB system mem for NVIDIA Tegra devices since they use the same mem for RAM and Video RAM + min_system_memory = 1<<30; +#endif return min_system_memory; } |