Mojo function
run_radix_sort_pairs_gpu
run_radix_sort_pairs_gpu[type: DType, out_idx_type: DType, ascending: Bool = False, BLOCK_SIZE: Int = 256, NUM_BITS_PER_PASS: Int = 4](ctx: DeviceContext, mut keys: DoubleBuffer[SIMD[type, 1]], mut key_ids: DoubleBuffer[SIMD[out_idx_type, 1]], skip_sort: UnsafePointer[SIMD[bool, 1]], in_shape: IndexList[size, element_type=element_type])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!