Home
last modified time | relevance | path

Searched defs:DeviceThreadHandlePool (Results 1 – 1 of 1) 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
31 struct Handle {
56 std::mutex mutex;
86 PoolWindow(std::shared_ptr<DeviceThreadHandlePool> parent): weak_parent(std::move(parent)) {} in PoolWindow() argument
87 ~PoolWindow(){ release(); } in ~PoolWindow()
89 Handle_t reserve(int device) in reserve()
121 std::weak_ptr<DeviceThreadHandlePool> weak_parent; argument
124 void release() { in release()
144 PoolWindow *newPoolWindow() { in newPoolWindow()