pytorch/torch/cuda
Nikita Shulga 43f0ccd1ec torch.cuda.memory_allocated to return {} if not initialized (#51179)
Summary:
Fixes https://github.com/pytorch/pytorch/issues/49952

Pull Request resolved: https://github.com/pytorch/pytorch/pull/51179

Reviewed By: ngimel

Differential Revision: D26094932

Pulled By: malfet

fbshipit-source-id: 0ec28ef9b0604245753d3f2b0e3536286700668d
2021-01-28 20:38:17 -08:00
..
amp [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00
__init__.py Back out "Revert D26077905: Back out "Revert D25850783: Add torch::deploy, an embedded torch-python interpreter"" (#51267) 2021-01-28 19:30:45 -08:00
_utils.py torch.cuda.memory_allocated to return {} if not initialized (#51179) 2021-01-28 20:38:17 -08:00
comm.py Add torch.cuda.comm to typechecking CI (#45350) 2020-09-25 12:13:43 -07:00
error.py remediation of S205607 2020-07-17 17:19:47 -07:00
memory.py torch.cuda.memory_allocated to return {} if not initialized (#51179) 2021-01-28 20:38:17 -08:00
nccl.py Enable torch.cuda.nccl typechecking (#45344) 2020-09-25 17:02:47 -07:00
nvtx.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00
profiler.py Replace map(lambda constructs (#46462) 2020-10-22 09:50:22 -07:00
random.py doc string fix for torch.cuda.set_rng_state_all (#40544) 2020-06-25 08:37:14 -07:00
sparse.py
streams.py [*.py] Rename "Arguments:" to "Args:" (#49736) 2020-12-28 09:34:47 -08:00