mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

fix input_node is data condaition

This commit is contained in:
ttudu 2022-03-11 17:38:54 +08:00
parent 4bf49bf6b9
commit 9e986fcedf
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
mindspore/ccsrc/runtime/graph_scheduler/graph_scheduler.cc
View File

@ -1607,8 +1607,11 @@ void GraphScheduler::LinkControlArrowForCustomActor(ActorSet *const actor_set,
FetchKernelTransformType(from_kernel_with_output_idx.first, graph, graph_compiler_info.origin_parameters_order_,
graph_compiler_info.strategy_);
auto from_actor = FetchActor(kernel_type, graph_compiler_info.name_, from_kernel_with_output_idx.first, graph);
AddDataArrow(from_actor, custom_actor.get(), from_kernel_with_output_idx.first,
from_kernel_with_output_idx.second, *iter);
// The input_node maybe a data(Tensor) and the from_actor is nullptr
if (from_actor != nullptr) {
AddDataArrow(from_actor, custom_actor.get(), from_kernel_with_output_idx.first,
from_kernel_with_output_idx.second, *iter);
}
}
AddControlArrow(actor_set->data_prepare_actor_.get(), custom_actor.get());