!1971 Enable new label switch task info

Merge pull request !1971 from zhoufeng/enable-label-task
5 years ago · 867cea77fd
parent eff90369bd e74125b295
commit 867cea77fd
2 changed files with 2 additions and 4 deletions
--- a/2
+++ b/2
@ -1 +1 @@
-Subproject commit 9248a2fd15ffc64d9d04b40c6b2836d1c94ca0b4
+Subproject commit 45ca7863ac6410c8e2f83168481ddc6b43bcea33
--- 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;