Mojo function
convert_ref_scales_to_mxfp8_format
convert_ref_scales_to_mxfp8_format[ref_scales_type: DType, scales_type: DType, *, REF_BLOCK_SIZE: Int, SF_VECTOR_SIZE: Int](m: ValOrDim[dim], n: ValOrDim[dim], k: ValOrDim[dim], ref_a_scales: NDBuffer[ref_scales_type, 2, origin, shape, strides], ref_b_scales: NDBuffer[ref_scales_type, 2, origin, shape, strides], a_scales: NDBuffer[scales_type, 5, origin, shape, strides], b_scales: NDBuffer[scales_type, 5, origin, shape, strides])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!