fix(compiler): CRT multiplication with negative value (close: #834)

2026-02-09 12:15:09 -05:00 · 2022-12-09 15:47:54 +01:00
parent b08ccbbfb5
commit 607457d34a
2 changed files with 21 additions and 1 deletions
--- a/compiler/lib/Dialect/BConcrete/Transforms/EliminateCRTOps.cpp
+++ b/compiler/lib/Dialect/BConcrete/Transforms/EliminateCRTOps.cpp
@@ -464,7 +464,7 @@ struct MulCleartextCRTLweTensorOpPattern
    mlir::Value init = rewriter.create<bufferization::AllocTensorOp>(
        op.getLoc(), resultTy, mlir::ValueRange{});

-    auto rhs = rewriter.create<arith::ExtUIOp>(op.getLoc(),
+    auto rhs = rewriter.create<arith::ExtSIOp>(op.getLoc(),
                                               rewriter.getI64Type(), op.rhs());

    // %0 = scf.for %i = %c0 to %cB step %c1 iter_args(%acc = %init) ->
--- a/compiler/tests/end_to_end_fixture/tests_cpu/end_to_end_fhe.yaml
+++ b/compiler/tests/end_to_end_fixture/tests_cpu/end_to_end_fhe.yaml