pytorch/torch/cuda
2022-11-08 18:11:01 +00:00
..
amp Store autocast_gpu_dtype in custom_fwd and custom_bwd for BFloat16 autocast (#88029) 2022-10-31 22:45:26 +00:00
__init__.py Deprecate TypedStorage, its derived classes, and all of their public methods (#85303) 2022-11-08 18:11:01 +00:00
_dynamo_graphs.py Deprecate TypedStorage, its derived classes, and all of their public methods (#85303) 2022-11-08 18:11:01 +00:00
_memory_viz.py Caching allocator tracing (#86241) 2022-10-07 23:19:54 +00:00
_sanitizer.py Rework printing tensor aliases in CSAN error message (#85008) 2022-09-21 13:41:52 +00:00
_utils.py
comm.py
error.py
graphs.py Disable autocast cache in torch.cuda.make_graphed_callables (#84289) 2022-09-01 21:34:51 +00:00
jiterator.py Fix typos used in documents under torch directory (#88300) 2022-11-02 09:38:13 +00:00
memory.py Fix typos used in documents under torch directory (#88300) 2022-11-02 09:38:13 +00:00
nccl.py
nvtx.py Fix docstring inconsistencies: string -> str, boolean -> bool (#82410) 2022-07-28 21:29:57 +00:00
profiler.py Add __all__ to torch.{autograd, fx, cuda} submodules (#85343) 2022-10-09 14:46:54 +00:00
random.py
sparse.py
streams.py