Home
last modified time | relevance | path

Searched refs:empty_cache (Results 1 – 25 of 34) sorted by relevance

12

/external/pytorch/aten/src/ATen/core/
DCachingHostAllocator.h212 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/
Dtest_cuda.py180 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 …]
Dtest_cuda_multigpu.py149 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/
Dmps.rst16 empty_cache
Dxpu.rst59 empty_cache
Dcuda.rst102 empty_cache
/external/pytorch/aten/src/ATen/cuda/
DCachingHostAllocator.cpp54 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/
DCachingHostAllocator.cpp69 getXPUCachingHostAllocator().empty_cache(); in CachingHostAllocator_emptyCache()
/external/pytorch/torch/mps/
D__init__.py81 def empty_cache() -> None: function
/external/pytorch/torch/xpu/
Dmemory.py13 def empty_cache() -> None: function
D__init__.py440 empty_cache,
/external/pytorch/torch/_inductor/fx_passes/
Dnumeric_utils.py39 torch.cuda.empty_cache()
/external/pytorch/test/distributed/_composable/fsdp/
Dtest_fully_shard_memory.py65 torch.cuda.empty_cache()
/external/pytorch/test/distributed/_tools/
Dtest_fsdp2_mem_tracker.py40 torch.cuda.empty_cache()
Dtest_mem_tracker.py23 torch.cuda.empty_cache()
/external/pytorch/torch/distributed/_tools/
Dmemory_tracker.py108 torch.cuda.empty_cache()
/external/pytorch/torch/cuda/
Dgraphs.py175 torch.cuda.empty_cache()
Dmemory.py180 def empty_cache() -> None: function
/external/pytorch/docs/source/notes/
Dhip.rst94 managed by the caching allocator. Calling :meth:`~torch.cuda.empty_cache`
/external/pytorch/test/inductor/
Dtest_torchinductor_opinfo.py592 torch.cuda.empty_cache()
/external/pytorch/test/jit/
Dtest_cuda.py53 torch.cuda.empty_cache()
/external/pytorch/torch/testing/_internal/
Dcommon_device_type.py1273 torch.cuda.empty_cache()
Dcommon_utils.py311 torch.cuda.empty_cache()
2279 torch.cuda.empty_cache()
2321 torch.cuda.empty_cache()
/external/pytorch/test/distributed/
Dtest_c10d_ops_nccl.py267 torch.cuda.empty_cache()
/external/pytorch/benchmarks/dynamo/
Dcommon.py372 torch.cuda.empty_cache()
374 torch.xpu.empty_cache()

12