Mojo function
run_radix_sort_pairs_gpu
run_radix_sort_pairs_gpu[dtype: DType, out_idx_type: DType, ascending: Bool = False, BLOCK_SIZE: Int = 256, NUM_BITS_PER_PASS: Int = 4](ctx: DeviceContext, mut keys: DoubleBuffer[Scalar[dtype]], mut key_ids: DoubleBuffer[Scalar[out_idx_type]], skip_sort: UnsafePointer[Scalar[DType.bool]], 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!