Browse Source

fix bug of transdata transpose split

tags/v0.5.0-beta
WilliamLian 5 years ago
parent
commit
9d364f509f
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.cc

+ 1
- 1
mindspore/ccsrc/pre_activate/ascend/ir_fission/transdata_split.cc View File

@@ -90,7 +90,7 @@ bool TransDataSplit::DoSplit(const FuncGraphPtr &func_graph, const AnfNodePtr &n
new_transdata_node =
NewTransOpNode(func_graph, new_transpose_node, kernel_select_, false, prim::KPrimTransData->name());
RefreshKernelBuildInfo(kOpFormat_HWCN, output_format, AnfAlgo::GetOutputDeviceDataType(new_transdata_node, 0),
new_transpose_node);
new_transdata_node);
new_replace_node = new_transdata_node;
}
FuncGraphManagerPtr manager = func_graph->manager();


Loading…
Cancel
Save