Searched refs:GPUBucket (Results 1 – 4 of 4) sorted by relevance
43 GPUBucket::GPUBucket(uint32_t id, uint32_t bucket_size) : Bucket(id, bucket_size), collective_handl… in GPUBucket() function in mindspore::device::gpu::GPUBucket47 void GPUBucket::AllocateAllReduceAddr() { in AllocateAllReduceAddr()82 void GPUBucket::FreeDeviceMem(void *dev_ptr) { GPUMemoryAllocator::GetInstance().FreeTensorMem(dev_… in FreeDeviceMem()84 void GPUBucket::FreeAllDeviceMem() { in FreeAllDeviceMem()101 void GPUBucket::CopyTensorToContiguousMemory() { in CopyTensorToContiguousMemory()121 void GPUBucket::LaunchAllReduce() { in LaunchAllReduce()159 std::shared_ptr<LaunchKernel> GPUBucket::CreateLaunchMul() { in CreateLaunchMul()168 void GPUBucket::Init(const std::vector<void *> &compute_streams, const std::vector<void *> &communi… in Init()
25 class GPUBucket : public Bucket {27 GPUBucket(uint32_t id, uint32_t bucket_size);28 ~GPUBucket() override = default;
510 auto bucket = std::make_shared<GPUBucket>(bucket_id, bucket_size); in CreateBucket()
738 auto bucket = std::make_shared<device::gpu::GPUBucket>(bucket_id, bucket_size); in CreateBucket()