1 #pragma once 2 3 #include <torch/csrc/distributed/c10d/Work.hpp> 4 5 namespace c10d { 6 7 C10_EXPORT void register_work( 8 const at::Tensor& tensor, 9 const c10::intrusive_ptr<c10d::Work>& work); 10 11 } // namespace c10d 12