Tobias Ringwald 6753ee127c Allow torch.cuda.memory.mem_get_info to take a device str argument with an unspecified device index. (#132616) ... `torch.cuda.memory.mem_get_info` allows device strings given the current type hints. However, `device = torch.device('cuda')` leads to `device.index = None`, which results in downstream problems. Setting `optional=True` will insert the default device index in such cases. Fixes #132583 Pull Request resolved: https://github.com/pytorch/pytorch/pull/132616 Approved by: https://github.com/soulitzer		2024-08-06 13:19:46 +00:00
..
amp	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
__init__.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
_gpu_trace.py
_memory_viz.py
_sanitizer.py	Add None return type to init (#132335)	2024-08-01 15:26:45 +00:00
_utils.py
comm.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
error.py
graphs.py	fixes #124582 (#128483)	2024-07-02 08:45:59 +00:00
jiterator.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
memory.py	Allow torch.cuda.memory.mem_get_info to take a device str argument with an unspecified device index. (#132616)	2024-08-06 13:19:46 +00:00
nccl.py
nvtx.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
profiler.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
random.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
sparse.py
streams.py	[BE][Easy][17/19] enforce style for empty lines in import segments in torch/[a-c]*/ and torch/[e-n]*/ (#129769)	2024-08-04 10:24:09 +00:00
tunable.py