Updated key areas identified from pr

This commit is contained in:
Alejandro Saucedo 2020-09-06 16:30:03 +01:00
parent eddba0fcf9
commit 56aa1ee96b
2 changed files with 0 additions and 7 deletions

View file

@ -50,14 +50,9 @@ OpTensorSyncDevice::init()
stagingTensor->init(
this->mPhysicalDevice, this->mDevice);
stagingTensor->mapDataIntoHostMemory();
this->mStagingTensors.push_back(stagingTensor);
} else {
tensor->mapDataIntoHostMemory();
// We push a nullptr when no staging tensor is needed to match
// index number in array to have one to one mapping with tensors
this->mStagingTensors.push_back(nullptr);