提交 03c25559 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!247 pynative-fix-bug-type-str-is-invalied-uint16

Merge pull request !247 from JoyLvliang/fix-bug-type-str-is-invalied-uint16
......@@ -94,7 +94,7 @@ AnfNodePtr GetTransInputNodePtr(const FuncGraphPtr &func_graph, const CNodePtr &
MS_EXCEPTION_IF_NULL(node);
bool padding_flag = false;
auto input_node = AnfAlgo::GetInputNode(node, index);
if (!AnfAlgo::IsFeatureMapInput(node, index)) {
if (input_node->isa<ValueNode>() || input_node->isa<Parameter>()) {
input_node = InsertTransOpForOutput(func_graph, input_node, kernel_select);
MS_EXCEPTION_IF_NULL(input_node);
AnfAlgo::SetNodeInput(node, input_node, index);
......
......@@ -186,7 +186,7 @@ void AscendSession::RunGraph(const GraphId &graph_id, const std::vector<tensor::
void AscendSession::RunOpHardwareOptimize(const std::shared_ptr<session::KernelGraph> &kernel_graph) const {
MS_LOG(INFO) << "Start";
// data layout optimization
opt::AscendDataLayout(kernel_graph);
opt::RunOpAscendDataLayout(kernel_graph);
// mixed precision optimization
opt::AscendMixPrecision(kernel_graph);
MS_LOG(INFO) << "Finish";
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册