提交 e74125b2 编写于 作者: Z zhoufeng

enable new label_switch task info

Signed-off-by: Nzhoufeng <zhoufeng54@huawei.com>
上级 444d9484
graphengine @ 45ca7863
Subproject commit 9248a2fd15ffc64d9d04b40c6b2836d1c94ca0b4
Subproject commit 45ca7863ac6410c8e2f83168481ddc6b43bcea33
......@@ -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;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册