Browse Source

fix the device address set of same tensor

tags/v1.3.0
limingqi107 5 years ago
parent
commit
9fe2aa1647
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/ccsrc/backend/session/gpu_session.cc

+ 1
- 1
mindspore/ccsrc/backend/session/gpu_session.cc View File

@@ -475,7 +475,7 @@ void GPUSession::UpdateOutputTensors(const VectorRef *outputs,
const auto &address = AnfAlgo::GetMutableOutputAddr(node, output_index);
// The outputs may have the same tensor, so need skip when the tensor has been set to device address.
if ((address == nullptr) || (address->GetPtr() == nullptr)) {
return;
continue;
}
tensor->set_device_address(address);



Loading…
Cancel
Save