pytorch/benchmarks/static_runtime
Zhou Fang fc5913b6bf [StaticRuntime] Fix a bug that memory planner ignores subblocks (#146728) (#146855)
Summary:

When Static Runtime graph node has sub-blocks, the memory planner does not consider sub-blocks' inputs as a node's input in memory planner. As the result, such nodes' inputs' lifetime is incorrect and corresponding tensor memory is released earlier than required and causes errors.

Differential Revision: D69195886

Pull Request resolved: https://github.com/pytorch/pytorch/pull/146855
Approved by: https://github.com/swolchok
2025-02-11 13:59:54 +00:00
..
CMakeLists.txt
deep_wide_pt_bench.cc
deep_wide_pt.cc
deep_wide_pt.h
test_cpu_fusion.cc
test_generated_ops.cc
test_static_module.cc [StaticRuntime] Fix a bug that memory planner ignores subblocks (#146728) (#146855) 2025-02-11 13:59:54 +00:00
test_static_runtime.cc
test_utils.cc Re-enable some C++ warnings (#142332) 2024-12-12 04:02:12 +00:00
test_utils.h