mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-24 19:43:35 +00:00
6b73ab3e3e
### Description <!-- Describe your changes. --> Introduce reused_buffer_index_per_stream in allocation planner which will be reset after computing the reuse buffer for each stream. So if a NodeArg is an input of several Ops across different streams and reuses other NodeArg, the reused NodeArg won't be involved when computing the second stream's reuse plan. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> This is to fix https://github.com/microsoft/onnxruntime/issues/19480, which is a crash for the scenario mentioned above. --------- Co-authored-by: Lei Cao <leca@microsoft.com> |
||
|---|---|---|
| .. | ||
| 3_gpu_streams.json | ||
| conv_add_relu.onnx | ||
| conv_add_relu_single_stream.json | ||
| conv_add_relu_single_stream_mismatch_device.json | ||
| conv_add_relu_single_stream_missing_node.json | ||
| cpu_if.onnx | ||
| issue_19480.onnx | ||
| memcpyToHost_same_stream_with_transpose.json | ||
| multi_stream_double_stream.json | ||
| multi_stream_single_stream.json | ||
| simplified_ssd.onnx | ||
| simplified_ssd_cpu.json | ||