| /aosp_15_r20/external/pytorch/c10/cuda/ |
| H A D | CUDAMallocAsyncAllocator.cpp | 151 cudaMemPool_t mempool = nullptr; in lazy_init_device() local
152 C10_CUDA_CHECK(cudaDeviceGetDefaultMemPool(&mempool, device)); in lazy_init_device()
155 mempool, cudaMemPoolAttrReleaseThreshold, &threshold)); in lazy_init_device()
161 mempool, cudaMemPoolReuseFollowEventDependencies, &enable)); in lazy_init_device()
163 mempool, cudaMemPoolReuseAllowOpportunistic, &enable)); in lazy_init_device()
165 mempool, cudaMemPoolReuseAllowInternalDependencies, &enable)); in lazy_init_device()
491 cudaMemPool_t mempool = nullptr; in emptyCache() local
492 cudaDeviceGetDefaultMemPool(&mempool, dev); in emptyCache()
494 cudaMemPoolTrimTo(mempool, 0); in emptyCache()
686 cudaMemPool_t mempool = nullptr; in getDeviceStats() local
[all …]
|
| H A D | CUDACachingAllocator.h | 480 MemPoolContext(MemPool* mempool);
|
| H A D | CUDACachingAllocator.cpp | 3830 MemPoolContext::MemPoolContext(MemPool* mempool) in MemPoolContext() argument
3832 active_mempool_ = mempool; in MemPoolContext()
|
| /aosp_15_r20/external/mesa3d/src/util/ |
| H A D | slab.c | 299 slab_alloc_st(struct slab_mempool *mempool) in slab_alloc_st() argument
301 return slab_alloc(&mempool->child); in slab_alloc_st()
308 slab_free_st(struct slab_mempool *mempool, void *ptr) in slab_free_st() argument
310 slab_free(&mempool->child, ptr); in slab_free_st()
314 slab_destroy(struct slab_mempool *mempool) in slab_destroy() argument
316 slab_destroy_child(&mempool->child); in slab_destroy()
317 slab_destroy_parent(&mempool->parent); in slab_destroy()
327 slab_create(struct slab_mempool *mempool, in slab_create() argument
331 slab_create_parent(&mempool->parent, item_size, num_items); in slab_create()
332 slab_create_child(&mempool->child, &mempool->parent); in slab_create()
|
| H A D | slab.h | 93 void slab_create(struct slab_mempool *mempool,
96 void slab_destroy(struct slab_mempool *mempool);
97 void *slab_alloc_st(struct slab_mempool *mempool);
98 void slab_free_st(struct slab_mempool *mempool, void *ptr);
|
| /aosp_15_r20/external/mesa3d/src/freedreno/decode/ |
| H A D | crashdec-mempool.c | 74 dump_cp_mem_pool(uint32_t *mempool) in dump_cp_mem_pool() argument
116 small_mem_pool ? &mempool[0x800] : &mempool[0x1000]; in dump_cp_mem_pool()
220 small_mem_pool ? &mempool[0xc00] : &mempool[0x1800]; in dump_cp_mem_pool()
227 small_mem_pool ? &mempool[0x1000] : &mempool[0x2000]; in dump_cp_mem_pool()
284 uint32_t *chunk_ptr = &mempool[cur_block * 0x20 + cur_chunk * 4]; in dump_cp_mem_pool()
|
| H A D | crashdec.h | 82 void dump_cp_mem_pool(uint32_t *mempool);
|
| H A D | meson.build | 218 'crashdec-mempool.c',
|
| /aosp_15_r20/external/pytorch/torch/cuda/ |
| H A D | graphs.py | 311 mempool = graph_pool_handle() if pool is None else pool
350 with torch.cuda.graph(fwd_graph, pool=mempool):
375 with torch.cuda.graph(bwd_graph, pool=mempool):
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDAGraph.cpp | 118 auto mempool = c10::cuda::MemPool({}, false); in capture_begin() local
119 mempool_id_ = mempool.id(); in capture_begin()
|
| /aosp_15_r20/external/sandboxed-api/sandboxed_api/bazel/external/ |
| H A D | libunwind.BUILD | 38 "include/mempool.h",
87 "src/mi/mempool.c",
|
| /aosp_15_r20/external/sandboxed-api/cmake/ |
| H A D | libunwind.cmake | 127 ${libunwind_SOURCE_DIR}/include/mempool.h
163 ${libunwind_SOURCE_DIR}/src/mi/mempool.c
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_cuda.py | 2859 mempool = torch.cuda.graph_pool_handle()
2865 graphed_model, (x,), pool=mempool
|
| /aosp_15_r20/external/rust/android-crates-io/crates/regex/ |
| D | CHANGELOG.md | 1106 Use of the `mempool` crate (which used thread local storage) was replaced
|
| /aosp_15_r20/external/pytorch/torch/_C/ |
| H A D | __init__.pyi.in | 1818 def _cuda_getCheckpointState(device: _int, mempool: Tuple[_int, _int]) -> _cuda_CUDAAllocator_Alloc…
|
| /aosp_15_r20/external/cronet/third_party/rust/chromium_crates_io/vendor/regex-1.10.4/ |
| H A D | CHANGELOG.md | 1595 Use of the `mempool` crate (which used thread local storage) was replaced
|
| /aosp_15_r20/external/mesa3d/docs/relnotes/ |
| H A D | 24.2.0.rst | 3383 - panvk: Make mempool detect NULL BOs
3387 - panvk: Fix device mempool leaks
|
| H A D | 19.0.0.rst | 1140 - util/slab: Rename slab_mempool typed parameters to mempool
|
| H A D | 22.0.0.rst | 3799 - freedreno/crashdec: Split out mempool decoding
|
| /aosp_15_r20/external/bcc/libbpf-tools/powerpc/ |
| H A D | vmlinux.h | 97164 mempool_t *mempool; member
|
| H A D | vmlinux_600.h | 97164 mempool_t *mempool; member
|
| /aosp_15_r20/external/bcc/libbpf-tools/loongarch/ |
| H A D | vmlinux.h | 93137 mempool_t *mempool; member
|
| H A D | vmlinux_602.h | 93137 mempool_t *mempool; member
|
| /aosp_15_r20/external/bcc/libbpf-tools/x86/ |
| H A D | vmlinux.h | 76661 mempool_t *mempool; member
|
| H A D | vmlinux_518.h | 76661 mempool_t *mempool; member
|