Searched refs:kStreamNeedActivedFirst (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ |
| D | kernel_adjust.h | 44 constexpr auto kStreamNeedActivedFirst = "stream_need_active_first"; variable
|
| D | kernel_adjust.cc | 147 AnfAlgo::SetNodeAttr(kStreamNeedActivedFirst, MakeValue<bool>(true), independent_switch_app); in InsertIndepentParallel()
167 AnfAlgo::SetNodeAttr(kStreamNeedActivedFirst, MakeValue<bool>(true), fpbp_switch_app); in InsertFpBpLoopStreamSwitch()
306 AnfAlgo::SetNodeAttr(kStreamNeedActivedFirst, MakeValue<bool>(true), getnext_switch_app); in InsertGetNextLoopStreamSwitch()
370 AnfAlgo::SetNodeAttr(kStreamNeedActivedFirst, MakeValue<bool>(true), eos_switch_app); in InsertEosStreamSwitch()
|
| /third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/ |
| D | ascend_stream_assign.cc | 1140 if (!AnfAlgo::HasNodeAttr(kStreamNeedActivedFirst, cur_cnode_ptr)) { in GetProcessedStream()
1143 auto need_active = AnfAlgo::GetNodeAttr<bool>(cur_cnode_ptr, kStreamNeedActivedFirst); in GetProcessedStream()
1155 if (!AnfAlgo::HasNodeAttr(kStreamNeedActivedFirst, switch_ptr)) { in CheckStreamSwitch()
1159 auto need_active = AnfAlgo::GetNodeAttr<bool>(switch_ptr, kStreamNeedActivedFirst); in CheckStreamSwitch()
1790 if (!AnfAlgo::HasNodeAttr(kStreamNeedActivedFirst, cur_cnode_ptr)) { in GetNeedActiveStreams()
1794 auto need_active = AnfAlgo::GetNodeAttr<bool>(cur_cnode_ptr, kStreamNeedActivedFirst); in GetNeedActiveStreams()
|