From e74125b29538a7af9d281a2565751bab5fec80ff Mon Sep 17 00:00:00 2001
From: zhoufeng <zhoufeng54@huawei.com>
Date: Wed, 10 Jun 2020 16:13:14 +0800
Subject: [PATCH] enable new label_switch task info

Signed-off-by: zhoufeng <zhoufeng54@huawei.com>
---
 graphengine                                | 2 +-
 mindspore/ccsrc/kernel/rts/label_switch.cc | 4 +---
 2 files changed, 2 insertions(+), 4 deletions(-)

diff --git a/graphengine b/graphengine
index 9248a2fd1..45ca7863a 160000
--- a/graphengine
+++ b/graphengine
@@ -1 +1 @@
-Subproject commit 9248a2fd15ffc64d9d04b40c6b2836d1c94ca0b4
+Subproject commit 45ca7863ac6410c8e2f83168481ddc6b43bcea33
diff --git a/mindspore/ccsrc/kernel/rts/label_switch.cc b/mindspore/ccsrc/kernel/rts/label_switch.cc
index 168e1f484..256ab7071 100644
--- a/mindspore/ccsrc/kernel/rts/label_switch.cc
+++ b/mindspore/ccsrc/kernel/rts/label_switch.cc
@@ -67,9 +67,7 @@ std::vector<TaskInfoPtr> LabelSwitchKernel::GenTask(const std::vector<AddressPtr
   MS_LOG(INFO) << "LabelSwitchKernel GenTask label size:" << label_size_ << ", stream id:" << stream_id;
   std::vector<TaskInfoPtr> task_info_list;
   cond_ = inputs[0]->addr;
-  // todo: need update ge task info define
-  auto task_info_ptr = std::make_shared<LabelSwitchTaskInfo>(stream_id, 0);
-  // auto task_info_ptr = std::make_shared<LabelSwitchTaskInfo>(stream_id, label_size_, label_list_, cond_);
+  auto task_info_ptr = std::make_shared<LabelSwitchTaskInfo>(stream_id, label_size_, label_list_, cond_);
   MS_EXCEPTION_IF_NULL(task_info_ptr);
   task_info_list.emplace_back(task_info_ptr);
   return task_info_list;
-- 
GitLab