Searched refs:true_stream_id (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/task/ |
| D | stream_switch_task.cc | 44 …MS_LOG(INFO) << "Stream " << task_info_->stream_id() << " active " << task_info_->true_stream_id(); in Distribute()
47 if (static_cast<uint64_t>(task_info_->true_stream_id()) >= stream_list_.size()) { in Distribute()
48 …MS_LOG(EXCEPTION) << "true_stream_id " << task_info_->true_stream_id() << " must be less than stre… in Distribute()
55 rtStream_t true_stream = stream_list_[LongToSize(task_info_->true_stream_id())]; in Distribute()
59 …<< ", trueStreamID: " << task_info_->true_stream_id() << ", datatype: " << task_info_->data_type(); in Distribute()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ge_runtime/ |
| D | task_info.h | 328 …StreamSwitchTaskInfo(const std::string &op_name, uint32_t stream_id, int64_t true_stream_id, void … in StreamSwitchTaskInfo() argument
331 true_stream_id_(true_stream_id), in StreamSwitchTaskInfo()
338 int64_t true_stream_id() const { return true_stream_id_; } in true_stream_id() function
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 958 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrTrueBranchStream); in ActiveRootGraphHcom() local
960 << "; true branch stream id:" << true_stream_id; in ActiveRootGraphHcom()
962 AnfAlgo::SetStreamId(true_stream_id, active_ptr.get()); in ActiveRootGraphHcom()
1136 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(cur_cnode_ptr, kAttrTrueBranchStream); in GetProcessedStream() local
1137 processed_streams_.emplace(true_stream_id); in GetProcessedStream()
2168 auto true_stream_id = AnfAlgo::GetNodeAttr<uint32_t>(node_ptr, kAttrTrueBranchStream); in GetStreamSwitchStreamRelation() local
2169 if (true_stream_id <= cur_stream_id) { in GetStreamSwitchStreamRelation()
2171 << " is greater than true branch stream id:" << true_stream_id; in GetStreamSwitchStreamRelation()
2175 stream_relations_[cur_stream_id] = {true_stream_id}; in GetStreamSwitchStreamRelation()
2178 …(stream_relations_[cur_stream_id].begin(), stream_relations_[cur_stream_id].end(), true_stream_id); in GetStreamSwitchStreamRelation()
[all …]
|