提交 7de1bb11 编写于 作者: M Megvii Engine Team

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

GitOrigin-RevId: 3be6d5fe6bfa933eb6d4b4dd38acdb2629b31755
上级 b7c9361f
......@@ -411,7 +411,7 @@ struct ComputingGraphHolder {
executable->wait();
size_t nr_inputs = inputs.size();
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(
input_dev_tensor.storage(), input_dev_tensor.layout());
if (inputs[i].host_value) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册