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

!28706 Fix Splitv Compile Bug 

Merge pull request !28706 from hwjiaorui/compile-bug
This commit is contained in:
i-robot 2022-01-08 09:05:58 +00:00 committed by Gitee
parent e1513a3726 06e6f35348
commit b87e6c542e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/backend/session/anf_runtime_algorithm.cc
View File

@ -869,7 +869,7 @@ std::vector<int64_t> AnfRuntimeAlgorithm::GetInputDeviceShapeForTbeBuild(const A
infer_shape = trans::PaddingShape(infer_shape, format, GetInputReshapeType(node, input_idx));
}
auto dtype = GetInputDeviceDataType(node, input_idx);
return trans::TransShapeToDevice(infer_shape, format, node, input_idx, dtype);
return trans::TransShapeToDevice(infer_shape, format, node, input_idx, dtype, false);
}
std::vector<size_t> AnfRuntimeAlgorithm::GetInputDeviceShape(const AnfNodePtr &node, size_t input_idx) {