mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
85ebc0cb7a
Rebase to llvm-project at 3f81841474fe with a pending upstream patch for arbitrary element types in linalg named operations. Co-authored-by: Ayoub Benaissa <ayoub.benaissa@zama.ai>
29 lines
1022 B
MLIR
29 lines
1022 B
MLIR
// RUN: concretecompiler --action=dump-fhe %s 2>&1| FileCheck %s
|
|
|
|
// CHECK-LABEL: func.func @add_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2>
|
|
func.func @add_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2> {
|
|
// CHECK-NEXT: return %arg0 : !FHE.eint<2>
|
|
|
|
%0 = arith.constant 0 : i3
|
|
%1 = "FHE.add_eint_int"(%arg0, %0): (!FHE.eint<2>, i3) -> (!FHE.eint<2>)
|
|
return %1: !FHE.eint<2>
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @sub_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2>
|
|
func.func @sub_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2> {
|
|
// CHECK-NEXT: return %arg0 : !FHE.eint<2>
|
|
|
|
%0 = arith.constant 0 : i3
|
|
%1 = "FHE.sub_eint_int"(%arg0, %0): (!FHE.eint<2>, i3) -> (!FHE.eint<2>)
|
|
return %1: !FHE.eint<2>
|
|
}
|
|
|
|
// CHECK-LABEL: func.func @mul_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2>
|
|
func.func @mul_eint_int(%arg0: !FHE.eint<2>) -> !FHE.eint<2> {
|
|
// CHECK-NEXT: return %arg0 : !FHE.eint<2>
|
|
|
|
%0 = arith.constant 1 : i3
|
|
%1 = "FHE.mul_eint_int"(%arg0, %0): (!FHE.eint<2>, i3) -> (!FHE.eint<2>)
|
|
return %1: !FHE.eint<2>
|
|
}
|