Searched refs:origin_size_list (Results 1 – 1 of 1) sorted by relevance
40 std::vector<size_t> origin_size_list; in AllocateAllReduceAddr() local48 origin_size_list.emplace_back(origin_size); in AllocateAllReduceAddr()65 if (origin_size_list.size() < bucket_size_ || align_size_list_.size() < bucket_size_) { in AllocateAllReduceAddr()66 … << "Invalid bucket_size_:" << bucket_size_ << " origin_size_list.size:" << origin_size_list.size() in AllocateAllReduceAddr()70 …_output_addrs_.emplace_back(std::make_shared<kernel::Address>(memcpy_output, origin_size_list[i])); in AllocateAllReduceAddr()