Searched refs:get_buffer_ptrs (Results 1 – 4 of 4) sorted by relevance
43 virtual std::vector<void*> get_buffer_ptrs() = 0;
32 std::vector<void*> get_buffer_ptrs() override;
538 symmetricMemory_->get_buffer_ptrs()[rank_], in oneShotAllReduce()598 symmetricMemory_->get_buffer_ptrs()[rank_], in twoShotAllReduce()648 symmetricMemory_->get_buffer_ptrs()[rank_], in hybridCubeMeshAllReduce()
326 std::vector<void*> CUDASymmetricMemory::get_buffer_ptrs() { in get_buffer_ptrs() function in c10d::symmetric_memory::CUDASymmetricMemory