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