ciflow/trunk/201326: [Inductor] Reuse identical custom-op layout constraints
- PyTorch: 2491 events in the last 90 days
- PyTorch: 2458th Release in the last 90 days
- Previous: earlier the same day · trunk/b7fb781ec19c6908e73becce6d3a902b5a2c9bef
What happened
Summary: Summary Reuse prepared inputs when the same IR value needs identical exact strides. We observed lower peak memory in a workload using this pattern. Problem Some attention workloads pass the same tensor as both K and V. Independent argument preparation can materialize it twice. Minimal CUDA reproducer on this branch: torch.Tensor: print("shared buffer:", k.data_ptr() == v.data_ptr()) return k + v pair.regist…
Summary assembled by rule from the sources below