Skip to content

Commit

Permalink
feat(compiler): Copy tile sizes for all matmul ops when lowering to l…
Browse files Browse the repository at this point in the history
…inalg.generic

For now, the attribute "tile-sizes" is copied from the FHELinalg
operation to the corresponding `linalg.generic` operation only for
multiplications of encrypted matrices with plaintext matrices.

The changes of this commit also cause the attribute to be copied for
multiplications between plaintext ad ciphertext matrices, as well as
for multiplications between two ciphertext matrices.
  • Loading branch information
andidr committed Apr 9, 2024
1 parent e6ae713 commit 278c9dc
Showing 1 changed file with 3 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -188,7 +188,9 @@ class FHELinalgTilingMarkerPass
op->walk([&](mlir::Operation *op) {
if (llvm::isa<mlir::concretelang::FHELinalg::ApplyLookupTableEintOp>(
op) ||
llvm::isa<mlir::concretelang::FHELinalg::MatMulEintIntOp>(op) ||
llvm::isa<mlir::concretelang::FHELinalg::MatMulEintIntOp,
mlir::concretelang::FHELinalg::MatMulIntEintOp,
mlir::concretelang::FHELinalg::MatMulEintEintOp>(op) ||
llvm::isa<mlir::concretelang::FHELinalg::AddEintOp,
mlir::concretelang::FHELinalg::AddEintIntOp,
mlir::concretelang::FHELinalg::SubIntEintOp,
Expand Down

0 comments on commit 278c9dc

Please sign in to comment.