Browse Source

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

Merge pull request !247 from JoyLvliang/fix-bug-type-str-is-invalied-uint16
tags/v0.2.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
03c2555977
2 changed files with 2 additions and 2 deletions
  1. +1
    -1
      mindspore/ccsrc/pre_activate/ascend/ascend_helper.cc
  2. +1
    -1
      mindspore/ccsrc/session/ascend_session.cc

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

@@ -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);


+ 1
- 1
mindspore/ccsrc/session/ascend_session.cc View File

@@ -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";


Loading…
Cancel
Save