Home
last modified time | relevance | path

Searched refs:GPUBucket (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc43 GPUBucket::GPUBucket(uint32_t id, uint32_t bucket_size) : Bucket(id, bucket_size), collective_handl… in GPUBucket() function in mindspore::device::gpu::GPUBucket
47 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()
Dgpu_bucket.h25 class GPUBucket : public Bucket {
27 GPUBucket(uint32_t id, uint32_t bucket_size);
28 ~GPUBucket() override = default;
/third_party/mindspore/mindspore/ccsrc/runtime/hardware/gpu/
Dgpu_device_context.cc510 auto bucket = std::make_shared<GPUBucket>(bucket_id, bucket_size); in CreateBucket()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dgpu_session.cc738 auto bucket = std::make_shared<device::gpu::GPUBucket>(bucket_id, bucket_size); in CreateBucket()