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

!46933 data sync before get tensor value 

Merge pull request !46933 from looop5/data_sync_master
This commit is contained in:
i-robot 2022-12-19 06:32:27 +00:00 committed by Gitee
parent 74be123e43 267db0bbb7
commit 5cc85b0467
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/ccsrc/pipeline/pynative/grad/bprop_expander/grad_ops/common_utils.cc
View File

@ -211,6 +211,9 @@ int64_t GetIntValue(const NodePtr &node) { return AnfUtils::GetIntValue(node->ge
std::vector<int64_t> GetIntList(const ValuePtr &value) {
MS_EXCEPTION_IF_NULL(value);
if (value->isa<tensor::Tensor>()) {
auto tensor = value->cast<tensor::TensorPtr>();
MS_EXCEPTION_IF_NULL(tensor);
tensor->data_sync();
return CheckAndConvertUtils::CheckTensorIntValue("tensor", value, "bprop");
} else {
return CheckAndConvertUtils::CheckIntOrTupleInt("value", value, "bprop");