Skip to content

Instantly share code, notes, and snippets.

View mundya's full-sized avatar

Andrew Mundy mundya

View GitHub Profile
@bjacob
bjacob / README.md
Last active June 30, 2023 14:56
Example of data-tiling and microkernels on a matmul.

Example of data-tiling and microkernels on a matmul.

Basic setup.

Source: matmul_i8.mlir:

func.func @matmul_i8(%lhs: tensor<?x?xi8>, %rhs: tensor<?x?xi8>, %acc: tensor<?x?xi32>) -> tensor<?x?xi32> {
  %result = linalg.matmul ins(%lhs, %rhs: tensor<?x?xi8>, tensor<?x?xi8>) outs(%acc: tensor<?x?xi32>) -> tensor<?x?xi32>
 return %result: tensor