Searched refs:memcpy_input_addrs_ (Results 1 – 4 of 4) sorted by relevance
51 memcpy_input_addrs_.emplace_back(std::make_shared<kernel::Address>( in AllocateAllReduceAddr()101 if (memcpy_input_addrs_.size() < bucket_size_ || memcpy_output_addrs_.size() < bucket_size_) { in CopyTensorToContiguousMemory()103 << " memcpy_input_addr_.size:" << memcpy_input_addrs_.size() in CopyTensorToContiguousMemory()107 MS_EXCEPTION_IF_NULL(memcpy_input_addrs_[i]); in CopyTensorToContiguousMemory()110 << " src size:" << memcpy_input_addrs_[i]->size; in CopyTensorToContiguousMemory()111 if (memcpy_output_addrs_[i]->size < memcpy_input_addrs_[i]->size) { in CopyTensorToContiguousMemory()115 …pyAsync(memcpy_output_addrs_[i]->addr, memcpy_output_addrs_[i]->size, memcpy_input_addrs_[i]->addr, in CopyTensorToContiguousMemory()116 … memcpy_input_addrs_[i]->size, RT_MEMCPY_DEVICE_TO_DEVICE, compute_stream_); in CopyTensorToContiguousMemory()
66 memcpy_input_addrs_.emplace_back( in AllocateAllReduceAddr()111 MS_EXCEPTION_IF_NULL(memcpy_input_addrs_[i]); in CopyTensorToContiguousMemory()113 memcpy_input_addrs_[i]->addr, in CopyTensorToContiguousMemory()
74 std::vector<kernel::AddressPtr> memcpy_input_addrs_; variable
131 memcpy_input_addrs_.clear(); in Release()