Home
last modified time | relevance | path

Searched refs:compute_streams (Results 1 – 5 of 5) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc168 void GPUBucket::Init(const std::vector<void *> &compute_streams, const std::vector<void *> &communi… in Init() argument
172 if (!compute_streams.empty()) { in Init()
173 compute_stream_ = compute_streams.front(); in Init()
Dgpu_bucket.h30 …void Init(const std::vector<void *> &compute_streams, const std::vector<void *> &communication_str…
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.cc188 void AscendBucket::Init(const std::vector<void *> &compute_streams, const std::vector<void *> &comm… in Init() argument
192 if (!compute_streams.empty()) { in Init()
193 compute_stream_ = compute_streams.front(); in Init()
Dascend_bucket.h30 …void Init(const std::vector<void *> &compute_streams, const std::vector<void *> &communication_str…
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dbucket.h53 …virtual void Init(const std::vector<void *> &compute_streams, const std::vector<void *> &communica…