forked from mindspore-Ecosystem/mindspore
!47323 close the wingman queue only when it is created
Merge pull request !47323 from wYann/data_queue_master
This commit is contained in:
commit
215053cdd9
|
@ -657,6 +657,9 @@ void WingmanQueue::Close() {
|
|||
std::shared_ptr<BlockingQueue> GetTdtWingManQueue(const std::shared_ptr<AnfNode> &node) {
|
||||
if (common::AnfAlgo::GetCNodeName(node) != kGetNextOpName) return nullptr;
|
||||
auto queue_name = common::AnfAlgo::GetNodeAttr<std::string>(node, "shared_name");
|
||||
if (!DataQueueMgr::GetInstance().IsCreated(queue_name)) {
|
||||
return nullptr;
|
||||
}
|
||||
return DataQueueMgr::GetInstance().GetDataQueue(queue_name);
|
||||
}
|
||||
|
||||
|
|
Loading…
Reference in New Issue