/external/pytorch/aten/src/ATen/core/ |
D | CachingHostAllocator.h | 212 virtual void empty_cache() { in empty_cache() function 370 void empty_cache() { in empty_cache() function 371 impl_->empty_cache(); in empty_cache()
|
/external/pytorch/test/ |
D | test_cuda.py | 180 torch.cuda.empty_cache() 251 torch.cuda.empty_cache() 266 torch.cuda.empty_cache() 280 torch.cuda.empty_cache() 288 torch.cuda.empty_cache() 783 torch.cuda.empty_cache() 1589 torch.cuda.empty_cache() 1687 torch.cuda.empty_cache() 1766 torch.cuda.empty_cache() 1798 torch.cuda.empty_cache() [all …]
|
D | test_cuda_multigpu.py | 149 def assert_change(comp=1, empty_cache=False, reset_peak=False): argument 178 if empty_cache: 188 torch.cuda.empty_cache() 214 assert_change(0, empty_cache=True) 280 assert_change(0, empty_cache=True) 287 torch.cuda.empty_cache() 304 torch.cuda.empty_cache() 315 torch.cuda.empty_cache() 1010 torch.cuda.empty_cache()
|
/external/pytorch/docs/source/ |
D | mps.rst | 16 empty_cache
|
D | xpu.rst | 59 empty_cache
|
D | cuda.rst | 102 empty_cache
|
/external/pytorch/aten/src/ATen/cuda/ |
D | CachingHostAllocator.cpp | 54 void empty_cache() { in empty_cache() function in at::cuda::__anondde4d04d0111::EventPool 261 getCUDACachingHostAllocator().empty_cache(); in CachingHostAllocator_emptyCache()
|
/external/pytorch/aten/src/ATen/xpu/ |
D | CachingHostAllocator.cpp | 69 getXPUCachingHostAllocator().empty_cache(); in CachingHostAllocator_emptyCache()
|
/external/pytorch/torch/mps/ |
D | __init__.py | 81 def empty_cache() -> None: function
|
/external/pytorch/torch/xpu/ |
D | memory.py | 13 def empty_cache() -> None: function
|
D | __init__.py | 440 empty_cache,
|
/external/pytorch/torch/_inductor/fx_passes/ |
D | numeric_utils.py | 39 torch.cuda.empty_cache()
|
/external/pytorch/test/distributed/_composable/fsdp/ |
D | test_fully_shard_memory.py | 65 torch.cuda.empty_cache()
|
/external/pytorch/test/distributed/_tools/ |
D | test_fsdp2_mem_tracker.py | 40 torch.cuda.empty_cache()
|
D | test_mem_tracker.py | 23 torch.cuda.empty_cache()
|
/external/pytorch/torch/distributed/_tools/ |
D | memory_tracker.py | 108 torch.cuda.empty_cache()
|
/external/pytorch/torch/cuda/ |
D | graphs.py | 175 torch.cuda.empty_cache()
|
D | memory.py | 180 def empty_cache() -> None: function
|
/external/pytorch/docs/source/notes/ |
D | hip.rst | 94 managed by the caching allocator. Calling :meth:`~torch.cuda.empty_cache`
|
/external/pytorch/test/inductor/ |
D | test_torchinductor_opinfo.py | 592 torch.cuda.empty_cache()
|
/external/pytorch/test/jit/ |
D | test_cuda.py | 53 torch.cuda.empty_cache()
|
/external/pytorch/torch/testing/_internal/ |
D | common_device_type.py | 1273 torch.cuda.empty_cache()
|
D | common_utils.py | 311 torch.cuda.empty_cache() 2279 torch.cuda.empty_cache() 2321 torch.cuda.empty_cache()
|
/external/pytorch/test/distributed/ |
D | test_c10d_ops_nccl.py | 267 torch.cuda.empty_cache()
|
/external/pytorch/benchmarks/dynamo/ |
D | common.py | 372 torch.cuda.empty_cache() 374 torch.xpu.empty_cache()
|