mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
ce631939f0
Fixes #ISSUE_NUMBER Pull Request resolved: https://github.com/pytorch/pytorch/pull/138692 Approved by: https://github.com/ezyang
23 lines
569 B
C++
23 lines
569 B
C++
#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
|