Skip to main content

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[Scalar[type]], 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?