Skip to content

v2.7.0-rc8

@pytorchbot pytorchbot tagged this 09 Apr 00:35
Fix inplacing with multiple, fused uses (#150845)

We had `can_inplace` defined on a single use. When that buffer has multiple uses inside a fused node, we need to check if the other accesses have the same index. Otherwise we may read memory that has already been written to from inplacing.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/150845
Approved by: https://github.com/zou3519, https://github.com/exclamaforte, https://github.com/atalman, https://github.com/jansel

(cherry picked from commit 27ded359a5dcbe8f92e01a24bec258bbfe1a73d6)

Co-authored-by: eellison <elias.ellison@gmail.com>
Assets 2
Loading