Nigel
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

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

Merge pull request !247 from JoyLvliang/fix-bug-type-str-is-invalied-uint16
This commit is contained in:
mindspore-ci-bot 2020-04-13 20:58:55 +08:00 committed by Gitee
parent 57f953ca38 79f74745ec
commit 03c2555977
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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