diff --git a/mindspore/lite/examples/train_lenet/src/net_runner.cc b/mindspore/lite/examples/train_lenet/src/net_runner.cc index 5e4dea836fc..50006373550 100644 --- a/mindspore/lite/examples/train_lenet/src/net_runner.cc +++ b/mindspore/lite/examples/train_lenet/src/net_runner.cc @@ -154,7 +154,6 @@ int NetRunner::InitDB() { TypeCast typecast(mindspore::DataType::kNumberTypeInt32); train_ds_ = train_ds_->Map({&typecast}, {"label"}); - // train_ds_ = train_ds_->Shuffle(2); train_ds_ = train_ds_->Batch(batch_size_, true); if (verbose_) { diff --git a/mindspore/lite/src/scheduler.cc b/mindspore/lite/src/scheduler.cc index 41a9570e3a1..b7138052674 100644 --- a/mindspore/lite/src/scheduler.cc +++ b/mindspore/lite/src/scheduler.cc @@ -722,9 +722,6 @@ int Scheduler::ConstructSubGraphs(std::vector src_kernel, if (head_kernel->subgraph_type() != kernel::kNotSubGraph) { (*is_kernel_finish)[head_kernel] = true; dst_kernel->push_back(head_kernel); - - /* npu support split */ - /* ConstructSubGraphs(head_kernel->nodes(), dst_kernel, is_kernel_finish); */ continue; } if (head_kernel->desc().arch == mindspore::kernel::kAPU) { diff --git a/mindspore/lite/src/sub_graph_split.cc b/mindspore/lite/src/sub_graph_split.cc index ee2fb84d95f..5ca7ab31336 100644 --- a/mindspore/lite/src/sub_graph_split.cc +++ b/mindspore/lite/src/sub_graph_split.cc @@ -45,7 +45,6 @@ void SearchSubGraph::ConvertSubGraphToModel() { if (subgraph.nodes_.empty()) { continue; } - // DeviceType device = subgraph.device_; int new_sub_index = model_->sub_graphs_.size(); int partial_index = model_->all_nodes_.size(); @@ -71,7 +70,6 @@ void SearchSubGraph::ConvertSubGraphToModel() { new_sub_graph->node_indices_.push_back(node_index); VectorErase(&main_graphs->node_indices_, node_index); VectorErase(&subgraph.nodes_, node_index); - // model_->all_nodes_[node_index]->device_type_ = device; } for (uint32_t head_index : subgraph.heads_) { @@ -227,13 +225,7 @@ void SearchSubGraph::InitSubgraphDevice() { } } -void SearchSubGraph::InitMainGraphDevice() { - // DeviceType main_device = DT_GPU; - // Model::SubGraph *main_graph = model_->sub_graphs_.front(); - // for (uint32_t node_index : main_graph->node_indices_) { - // Model::Node *node = model_->all_nodes_[node_index]; - // node->device_type_ = main_device; -} +void SearchSubGraph::InitMainGraphDevice() { return; } void SearchSubGraph::SubgraphFusion() { while (sub_graphs_.size() > 2) {