Mojo function
quantize_fp8_kernel
quantize_fp8_kernel[out_type: DType, scales_type: DType, in_type: DType, input_fn: def[width: Int, alignment: Int](row: Int, col: Int) capturing -> SIMD[in_type, width], num_threads: Int, group_size: Int, simd_width: Int, output_layout: TensorLayout, scales_layout: TensorLayout](output: TileTensor[out_type, output_layout, MutAnyOrigin], scales: TileTensor[scales_type, scales_layout, MutAnyOrigin], scale_ub: Scalar[scales_type])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!