!575 fix pynative and staging mode data sync

Merge pull request !575 from dinghao/master
This commit is contained in:
mindspore-ci-bot 2020-04-23 10:01:35 +08:00 committed by Gitee
commit adc42818cc
1 changed files with 4 additions and 1 deletions

View File

@ -164,8 +164,11 @@ Tensor::Tensor(const py::float_ &input, const TypePtr &data_type) { init(py::arr
Tensor::Tensor(const py::int_ &input, const TypePtr &data_type) { init(py::array(input), data_type); }
Tensor::Tensor(const Tensor &tensor, const TypePtr &data_type)
: MetaTensor(tensor), device_address_(tensor.device_address()) {
: MetaTensor(tensor), dirty_(tensor.dirty_), device_address_(tensor.device_address_) {
init(tensor.data_, data_type);
if (device_address_ != nullptr) {
(void)data_sync();
}
}
Tensor &Tensor::operator=(const Tensor &tensor) {