1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
|
#pragma once
#include <torch/csrc/distributed/c10d/ProcessGroup.hpp>
namespace c10d {
C10_EXPORT void set_thread_isolation_mode(bool enable);
bool get_thread_isolation_mode();
C10_EXPORT void register_process_group(
const std::string& group_name,
const c10::intrusive_ptr<c10d::ProcessGroup>& group);
C10_EXPORT c10::intrusive_ptr<c10d::ProcessGroup> resolve_process_group(
const std::string& group_name);
C10_EXPORT void unregister_process_group(const std::string& group_name);
C10_EXPORT void unregister_all_process_groups();
} // namespace c10d
|
