Searched defs:cuda_allocator (Results 1 – 2 of 2) sorted by relevance
144 auto cuda_allocator = at::cuda::getCUDADeviceAllocator(); in segmented_sort_pairs_by_full_sort() local192 auto cuda_allocator = at::cuda::getCUDADeviceAllocator(); in segmented_sort_pairs() local
33 auto cuda_allocator = at::cuda::getCUDADeviceAllocator(); in operator ()() local