Natalia Gimelshein f01e628e3b Resubmit Remove MemPoolContext (#154042) (#154746) ... Summary: Per title Test Plan: Added tests + existing tests Differential Revision: D75695030 Pull Request resolved: https://github.com/pytorch/pytorch/pull/154746 Approved by: https://github.com/malfet		2025-05-31 01:21:54 +00:00
..
amp
__init__.py	Resubmit Remove MemPoolContext (#154042) (#154746)	2025-05-31 01:21:54 +00:00
_gpu_trace.py
_memory_viz.py
_sanitizer.py
_utils.py	Add torch.cuda._compile_kernel() (#151484)	2025-04-24 07:14:31 +00:00
comm.py
error.py
gds.py	[BE] Upgrade to mypy 1.14 (#145966)	2025-03-04 20:58:26 +00:00
graphs.py	Revert "Implement cuda graphs implementation of torch.cond and torch.while_loop (#140979)"	2025-02-13 18:04:26 +00:00
jiterator.py
memory.py	Resubmit Remove MemPoolContext (#154042) (#154746)	2025-05-31 01:21:54 +00:00
nccl.py
nvtx.py
profiler.py
random.py	Avoid unnecessary clone in torch.cuda.set_rng_state (#149283)	2025-03-18 20:47:57 +00:00
sparse.py
streams.py
tunable.py	[ROCm][TunableOp] Support submatrices in offline tuning (#151138)	2025-04-19 04:14:27 +00:00