pytorch/torch/masked
albanD bf77f52895 Fix memory leak on masked Tensor (#137890)
Note that this reverts the change from https://github.com/pytorch/pytorch/pull/137815 as well which is not needed anymore!

Without this, you create an unbeakable reference cycle. It is unbreakable because part of the cycle is through the autograd graph which we cannot traverse.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/137890
Approved by: https://github.com/atalman, https://github.com/huydhn, https://github.com/Skylion007
2024-10-15 18:37:55 +00:00
..
maskedtensor Fix memory leak on masked Tensor (#137890) 2024-10-15 18:37:55 +00:00
__init__.py Add functions from torch.masked._ops to __all__ for torch.masked (#131288) 2024-08-01 05:45:08 +00:00
_docs.py Fix global flake8 issues (#124771) 2024-04-26 15:35:53 +00:00
_ops.py [BE] typing for decorators - masked/_ops (#135108) 2024-09-12 01:34:09 +00:00