Searched refs:capture_dev_ (Results 1 – 2 of 2) sorted by relevance
106 capture_dev_ = c10::cuda::current_device(); in capture_begin()126 …c10::cuda::CUDACachingAllocator::beginAllocateToPool(capture_dev_, mempool_id_, [this](cudaStream_… in capture_begin()161 c10::cuda::CUDACachingAllocator::endAllocateToPool(capture_dev_, mempool_id_); in capture_end()291 c10::cuda::CUDACachingAllocator::releasePool(capture_dev_, mempool_id_); in reset()
85 int capture_dev_; member