Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc168 …nit(const std::vector<void *> &compute_streams, const std::vector<void *> &communication_streams) { in Init() argument
175 if (!communication_streams.empty()) { in Init()
176 stream_ = communication_streams.front(); in Init()
Dgpu_bucket.h30 …t std::vector<void *> &compute_streams, const std::vector<void *> &communication_streams) override;
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_bucket.cc188 …nit(const std::vector<void *> &compute_streams, const std::vector<void *> &communication_streams) { in Init() argument
195 if (!communication_streams.empty()) { in Init()
196 stream_ = communication_streams.front(); in Init()
Dascend_bucket.h30 …t std::vector<void *> &compute_streams, const std::vector<void *> &communication_streams) override;
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dbucket.h53 …(const std::vector<void *> &compute_streams, const std::vector<void *> &communication_streams) = 0;