Mojo function
matmul_qint4
matmul_qint4[group_size: Int, b_static_shape: DimList = DimList.create_unknown[2](), elementwise_lambda_fn: OptionalReg[fn[dtype: DType, width: Int, *, alignment: Int = 1](IndexList[2], SIMD[dtype, width]) capturing -> None] = OptionalReg[fn[dtype: DType, width: Int, *, alignment: Int = 1](IndexList[2], SIMD[dtype, width]) capturing -> None]({:i1 0, 1})](a: NDBuffer[DType.float32, 2, origin], b: NDBuffer[DType.uint8, 2, origin, b_static_shape], c: NDBuffer[DType.float32, 2, origin])
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!