Home
last modified time | relevance | path

Searched refs:createCustomAllocator (Results 1 – 4 of 4) sorted by relevance

/external/pytorch/aten/src/ATen/test/
Dcuda_allocator_test.cpp29 …auto dummy_allocator_0 = torch::cuda::CUDAPluggableAllocator::createCustomAllocator(dummy_alloc_0,… in TEST()
34 …auto dummy_allocator_1 = torch::cuda::CUDAPluggableAllocator::createCustomAllocator(dummy_alloc_0,… in TEST()
/external/pytorch/torch/csrc/cuda/
DCUDAPluggableAllocator.h51 createCustomAllocator(
DCUDAPluggableAllocator.cpp377 createCustomAllocator( in createCustomAllocator() function
DModule.cpp1207 return createCustomAllocator(malloc_fn, free_fn); in registerCudaPluggableAllocator()