Skip to main content

Mojo function

matmul_qint4

matmul_qint4[group_size: Int, b_static_shape: DimList = DimList.create_unknown[2](), elementwise_lambda_fn: OptionalReg[fn[DType, Int, Int](IndexList[2], SIMD[$0, $1]) capturing -> None] = OptionalReg[fn[DType, Int, Int](IndexList[2], SIMD[$0, $1]) 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?