diff options
author | Xiaoqiang Zheng <zhengxq@google.com> | 2016-10-12 15:36:13 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2016-10-12 16:47:55 -0700 |
commit | 3ac64952aaf944cc6cc498474d3008cdb66bc8c7 (patch) | |
tree | 4496f6899bea3eb6b4234f9d81d5ef4fb5796042 /tensorflow/core/util/env_var.h | |
parent | 3e581482a97683c90a65969423dd75c41b2f8653 (diff) |
Adding an environmental variable to control the cuda host memory limit.
Change: 135978018
Diffstat (limited to 'tensorflow/core/util/env_var.h')
-rw-r--r-- | tensorflow/core/util/env_var.h | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/tensorflow/core/util/env_var.h b/tensorflow/core/util/env_var.h new file mode 100644 index 0000000000..ec661f1d81 --- /dev/null +++ b/tensorflow/core/util/env_var.h @@ -0,0 +1,40 @@ +/* Copyright 2016 The TensorFlow Authors. All Rights Reserved. + +Licensed under the Apache License, Version 2.0 (the "License"); +you may not use this file except in compliance with the License. +You may obtain a copy of the License at + + http://www.apache.org/licenses/LICENSE-2.0 + +Unless required by applicable law or agreed to in writing, software +distributed under the License is distributed on an "AS IS" BASIS, +WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +See the License for the specific language governing permissions and +limitations under the License. +==============================================================================*/ + +#ifndef TENSORFLOW_UTIL_ENV_VAR_H_ + +#include "tensorflow/core/lib/core/status.h" +#include "tensorflow/core/lib/core/stringpiece.h" +#include "tensorflow/core/platform/types.h" + +namespace tensorflow { + +// Return a boolean into "value" from the environmental variable "env_var_name". +// If it is unset, the default value is used. +// A string "0" or a case insensitive "false" is interpreted as false. +// A string "1" or a case insensitive "true" is interpreted as true. +// Otherwise, an error status is returned. +Status ReadBoolFromEnvVar(StringPiece env_var_name, bool default_val, + bool* value); + +// Return an int64 into "value" from the environmental variable "env_var_name". +// If it is unset, the default value is used. +// If the string cannot be parsed into int64, an error status is returned. +Status ReadInt64FromEnvVar(StringPiece env_var_name, int64 default_val, + int64* value); + +} // namespace tensorflow + +#endif // TENSORFLOW_UTIL_ENV_VAR_H_ |