mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
018e9826a2
Summary: pretty simple. if planner exists, which implies that planning is enabled, create a manager for each frame. the associated serial executor will use the withMemoryPlannner fn to ensure the deallocation is done after execution completes. Test Plan: CI Differential Revision: D73635809 Pull Request resolved: https://github.com/pytorch/pytorch/pull/157053 Approved by: https://github.com/henryoier, https://github.com/georgiaphillips |
||
|---|---|---|
| .. | ||
| CMakeLists.txt | ||
| test_c10_kernel.cpp | ||
| test_execution_frame.cpp | ||
| test_execution_planner.cpp | ||
| test_file_util.cpp | ||
| test_function_schema.cpp | ||
| test_graph_signature.cpp | ||
| test_graph.cpp | ||
| test_itree.cpp | ||
| test_layout_planner_algorithm.cpp | ||
| test_mpmc_queue.cpp | ||
| test_op_kernel.cpp | ||
| test_placement.cpp | ||
| test_serialization.cpp | ||
| test_tensor_meta.cpp | ||
| test_weights.cpp | ||