Function RangesEnabled

tensorflow/core/profiler/nvtx_utils.h:35–44

A helper function to decide whether to enable CUDA NVTX profiling ranges.

// A helper function to decide whether to enable CUDA NVTX profiling ranges.
inline bool RangesEnabled() {
  static bool is_enabled = [] {
    bool _is_enabled = false;
    TF_CHECK_OK(tensorflow::ReadBoolFromEnvVar("TF_ENABLE_NVTX_RANGES",
                                               /*default_val=*/false,
                                               &_is_enabled));
    return _is_enabled;
  }();
  return is_enabled;
}

// Note: The memory backing msg must persist until the result of this function
// has been consumed by an NVTX API.