Searched refs:ar_output_addr_ (Results 1 – 4 of 4) sorted by relevance
72 …ar_output_addr_ = static_cast<uint8_t *>(GPUMemoryAllocator::GetInstance().AllocTensorMem(total_si… in AllocateAllReduceAddr()90 if (ar_output_addr_ != nullptr) { in FreeAllDeviceMem()91 FreeDeviceMem(ar_output_addr_); in FreeAllDeviceMem()92 ar_output_addr_ = nullptr; in FreeAllDeviceMem()150 …(*all_reduce_funcptr)(ar_input_addr_, ar_output_addr_, total_size_ / type_size, nccl_data_type_ite… in LaunchAllReduce()
61 ar_output_addr_ = runtime_instance->MallocCommunicationMemFromMemPool(total_size); in AllocateAllReduceAddr()83 if (ar_output_addr_ != nullptr) { in FreeAllDeviceMem()84 uint8_t *origin_dev_addr = ar_output_addr_ - kMemAlignSize; in FreeAllDeviceMem()86 ar_output_addr_ = nullptr; in FreeAllDeviceMem()151 …auto hccl_result = hccl::HcclAdapter::GetInstance().HcclAllReduce(ar_input_addr_, ar_output_addr_,… in LaunchAllReduce()
45 ar_output_addr_(nullptr) {} in Bucket()69 uint8_t *ar_output_addr_; variable
91 UpdateTensorOutputAddr(ar_output_addr_); in CalculateMean()99 launch_mul_->SetInputAddr(ar_output_addr_); in CalculateMean()