mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
9beddde1d7
Automatically enable `NCCL_DESYNC_DEBUG` when `TORCH_DISTRIBUTED_DEBUG` is set to `DETAIL`. Saving user from setting two env variables. Pull Request resolved: https://github.com/pytorch/pytorch/pull/83881 Approved by: https://github.com/malfet, https://github.com/rohan-varma, https://github.com/H-Huang
24 lines
604 B
C++
24 lines
604 B
C++
// Copyright (c) Meta Platforms, Inc. and its affiliates.
|
|
// All rights reserved.
|
|
//
|
|
// This source code is licensed under the BSD-style license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
#pragma once
|
|
|
|
#include <c10/macros/Macros.h>
|
|
|
|
namespace c10d {
|
|
|
|
enum class DebugLevel { Off = 0, Info = 1, Detail = 2 };
|
|
|
|
TORCH_API void setDebugLevel(DebugLevel level);
|
|
|
|
// Sets the debug level based on the value of the `TORCH_DISTRIBUTED_DEBUG`
|
|
// environment variable.
|
|
TORCH_API void setDebugLevelFromEnvironment();
|
|
|
|
TORCH_API DebugLevel debug_level() noexcept;
|
|
|
|
} // namespace c10d
|