ciflow/torchtitan/196847: [Inductor] Account NonOwningLayout storage in peak-memory planning
- PyTorch: 814 events in the last 90 days
- PyTorch: 803th Release in the last 90 days
- Previous: earlier the same day · trunk/84e524623ea4754a748936bf1ba6ecaaa92c3ae6: add kernel shared memory get_graph_data() (#196527)
What happened
Summary: NonOwningLayout buffers do not allocate independent storage. Count scheduler-owned aliases as zero bytes, transfer their consumers to the ultimate storage owner, and keep owners of graph-output aliases live. Preserve conservative sizing when the owner is a graph input. Test Plan: python3 -m py_compile fbcode/caffe2/torch/_inductor/memory.py fbcode/caffe2/test/inductor/test_memory.py arc lint fbcode/caffe2/t…
Summary assembled by rule from the sources below