diff --git a/graphengine b/graphengine
index 9248a2fd15f..45ca7863ac6 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 168e1f4844b..256ab707107 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;