Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/
DCuSparseHandlePool.cpp24 using CuSparsePoolType = DeviceThreadHandlePool<cusparseHandle_t, createCusparseHandle, destroyCusp… typedef
37 static auto pool = std::make_shared<CuSparsePoolType>(); in getCurrentCUDASparseHandle()
38 thread_local std::unique_ptr<CuSparsePoolType::PoolWindow> myPoolWindow( in getCurrentCUDASparseHandle()