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 [Static Runtime] Fix & run gen_static_runtime_ops (#128299) 2024-06-11 16:27:39 +00:00
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 [9/N] Replace c10::optional with std::optional (#130674) 2024-07-15 00:48:43 +00:00
test_utils.cc Re-enable some C++ warnings (#142332) 2024-12-12 04:02:12 +00:00
test_utils.h