Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/detail/
H A DDeviceThreadHandles.h29 struct DeviceThreadHandlePool : public std::enable_shared_from_this<DeviceThreadHandlePool<Handle_t… struct
86 PoolWindow(std::shared_ptr<DeviceThreadHandlePool> parent): weak_parent(std::move(parent)) {} in PoolWindow() argument
121 std::weak_ptr<DeviceThreadHandlePool> weak_parent; argument
/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCublasHandlePool.cpp50 using CuBlasLtPoolType = DeviceThreadHandlePool<cublasLtHandle_t, createCublasLtHandle, destroyCubl…
75 using CuBlasPoolType = DeviceThreadHandlePool<cublasHandle_t, createCublasHandle, destroyCublasHand…
H A DCuSparseHandlePool.cpp24 using CuSparsePoolType = DeviceThreadHandlePool<cusparseHandle_t, createCusparseHandle, destroyCusp…
/aosp_15_r20/external/pytorch/aten/src/ATen/miopen/
H A DHandle.cpp31 using MIOpenPoolType = at::cuda::DeviceThreadHandlePool<miopenHandle_t, createMIOpenHandle, destroy…
/aosp_15_r20/external/pytorch/aten/src/ATen/cudnn/
H A DHandle.cpp32 using CudnnPoolType = at::cuda::DeviceThreadHandlePool<
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/linalg/
H A DCusolverDnHandlePool.cpp27 using CuSolverDnPoolType = DeviceThreadHandlePool<cusolverDnHandle_t, createCusolverDnHandle, destr…
H A DCudssHandlePool.cpp27 using CudssPoolType = DeviceThreadHandlePool<cudssHandle_t, createCudssHandle, destroyCudssHandle>;