Home
last modified time | relevance | path

Searched defs:PerDevicePool (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/cuda/
H A DCachingHostAllocator.cpp62 struct PerDevicePool { struct in at::cuda::__anon812ef5cd0111::EventPool
63 alignas(64) std::mutex mutex_;
64 std::vector<std::unique_ptr<at::cuda::CUDAEvent>> event_pool_;
/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDACachingAllocator.cpp807 struct PerDevicePool { struct in c10::cuda::CUDACachingAllocator::Native::__anon8687b97a0111::EventPool
808 alignas(64) std::mutex mutex_;
809 std::vector<std::unique_ptr<cudaEvent_t>> event_pool_;