flashinfer.quantizationΒΆ

Quantization related kernels.

packbits(x[, bitorder])

Pack the elements of a binary-valued array into bits in a uint8 array.

segment_packbits(x, indptr[, bitorder])

Pack a batch of binary-valued segments into bits in a uint8 array.