Browse Source

fix(mge/utils): disable memory forwarding for subgraph

GitOrigin-RevId: 3be6d5fe6b
tags/v1.9.0
Megvii Engine Team 3 years ago
parent
commit
7de1bb11ab
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      imperative/src/impl/ops/utility.cpp

+ 1
- 1
imperative/src/impl/ops/utility.cpp View File

@@ -411,7 +411,7 @@ struct ComputingGraphHolder {
executable->wait(); executable->wait();
size_t nr_inputs = inputs.size(); size_t nr_inputs = inputs.size();
for (size_t i = 0; i < nr_inputs; ++i) { for (size_t i = 0; i < nr_inputs; ++i) {
auto input_dev_tensor = input_tensors[i]->dev_tensor(false);
auto input_dev_tensor = input_tensors[i]->dev_tensor(true);
inputs[i].device_value->reset( inputs[i].device_value->reset(
input_dev_tensor.storage(), input_dev_tensor.layout()); input_dev_tensor.storage(), input_dev_tensor.layout());
if (inputs[i].host_value) { if (inputs[i].host_value) {


Loading…
Cancel
Save