Searched refs:raw_alloc_with_stream (Results 1 – 7 of 7) sorted by relevance
203 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() function311 return get()->raw_alloc_with_stream(nbytes, stream); in raw_alloc_with_stream()
843 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function
3548 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override { in raw_alloc_with_stream() function in c10::cuda::CUDACachingAllocator::Native::NativeCachingAllocator
107 void* raw_alloc_with_stream(size_t nbytes, cudaStream_t stream) override;
137 void* CUDAPluggableAllocator::raw_alloc_with_stream( in raw_alloc_with_stream() function in torch::cuda::CUDAPluggableAllocator::CUDAPluggableAllocator
313 mem = c10::cuda::CUDACachingAllocator::raw_alloc_with_stream(size, stream); in THCPModule_cudaCachingAllocator_raw_alloc()
626 void* raw_alloc_with_stream(size_t size, XPUStream stream) { in raw_alloc_with_stream() function in c10::xpu::XPUCachingAllocator::XPUAllocator