Home
last modified time | relevance | path

Searched refs:raw_alloc_with_stream (Results 1 – 7 of 7) sorted by relevance

/external/pytorch/c10/cuda/
DCUDACachingAllocator.h203 virtual void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) = 0;
310 inline void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) { in raw_alloc_with_stream() function
311 return get()->raw_alloc_with_stream(nbytes, stream); in raw_alloc_with_stream()
DCUDAMallocAsyncAllocator.cpp843 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function
DCUDACachingAllocator.cpp3548 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
/external/pytorch/torch/csrc/cuda/
DCUDAPluggableAllocator.h107 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override;
DCUDAPluggableAllocator.cpp137 void* CUDAPluggableAllocator::raw_alloc_with_stream( in raw_alloc_with_stream() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
DModule.cpp313 mem = c10::cuda::CUDACachingAllocator::raw_alloc_with_stream(size, stream); in THCPModule_cudaCachingAllocator_raw_alloc()
/external/pytorch/c10/xpu/
DXPUCachingAllocator.cpp626 void* raw_alloc_with_stream(size_t size, XPUStream stream) { in raw_alloc_with_stream() function in c10::xpu::XPUCachingAllocator::XPUAllocator