github/concrete
1
1
Fork 0
mirror of https://github.com/zama-ai/concrete.git synced 2026-02-09 03:55:04 -05:00
Code Issues Packages Projects Releases Wiki Activity

feature(compiler): Introduce the BConcrete Dialect

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-02-11 11:40:27 +01:00
committed by Quentin Bourgerie
parent b950a0010d
commit 1f3944047f
15 changed files with 246 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
compiler/include/concretelang/Dialect/BConcrete/CMakeLists.txt Normal file
View File

@@ -0,0 +1 @@
add_subdirectory(IR)

18
compiler/include/concretelang/Dialect/BConcrete/IR/BConcreteDialect.h Normal file
View File

@@ -0,0 +1,18 @@
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
// Exceptions. See
// https://github.com/zama-ai/concrete-compiler-internal/blob/master/LICENSE.txt
// for license information.
#ifndef ZAMALANG_DIALECT_BConcrete_IR_BConcreteDIALECT_H
#define ZAMALANG_DIALECT_BConcrete_IR_BConcreteDIALECT_H
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Dialect.h"
#include "mlir/IR/BuiltinOps.h"
#include "mlir/IR/BuiltinTypes.h"
#include "mlir/IR/Dialect.h"
#include "concretelang/Dialect/BConcrete/IR/BConcreteOpsDialect.h.inc"
#endif

15
compiler/include/concretelang/Dialect/BConcrete/IR/BConcreteDialect.td Normal file
View File

@@ -0,0 +1,15 @@
#ifndef ZAMALANG_DIALECT_BConcrete_IR_BConcrete_DIALECT
#define ZAMALANG_DIALECT_BConcrete_IR_BConcrete_DIALECT
include "mlir/IR/OpBase.td"
def BConcrete_Dialect : Dialect {
let name = "BConcrete";
let summary = "Bufferized concrete dialect";
let description = [{
A dialect for representation of bufferized concrete operations on fully homomorphic ciphertext.
}];
let cppNamespace = "::mlir::concretelang::BConcrete";
}
#endif

20
compiler/include/concretelang/Dialect/BConcrete/IR/BConcreteOps.h Normal file
View File

@@ -0,0 +1,20 @@
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
// Exceptions. See
// https://github.com/zama-ai/concrete-compiler-internal/blob/master/LICENSE.txt
// for license information.
#ifndef ZAMALANG_DIALECT_BConcrete_BConcrete_OPS_H
#define ZAMALANG_DIALECT_BConcrete_BConcrete_OPS_H
#include <mlir/IR/Builders.h>
#include <mlir/IR/BuiltinOps.h>
#include <mlir/IR/BuiltinTypes.h>
#include <mlir/Interfaces/ControlFlowInterfaces.h>
#include <mlir/Interfaces/SideEffectInterfaces.h>
#include "concretelang/Dialect/Concrete/IR/ConcreteTypes.h"
#define GET_OP_CLASSES
#include "concretelang/Dialect/BConcrete/IR/BConcreteOps.h.inc"
#endif

65
compiler/include/concretelang/Dialect/BConcrete/IR/BConcreteOps.td Normal file
View File

@@ -0,0 +1,65 @@
#ifndef ZAMALANG_DIALECT_BConcrete_IR_BConcrete_OPS
#define ZAMALANG_DIALECT_BConcrete_IR_BConcrete_OPS
include "mlir/Interfaces/SideEffectInterfaces.td"
include "mlir/Interfaces/ControlFlowInterfaces.td"
include "mlir/IR/BuiltinTypes.td"
include "mlir/Dialect/MemRef/IR/MemRefBase.td"
include "concretelang/Dialect/BConcrete/IR/BConcreteDialect.td"
include "concretelang/Dialect/Concrete/IR/ConcreteTypes.td"
class BConcrete_Op<string mnemonic, list<OpTrait> traits = []> :
Op<BConcrete_Dialect, mnemonic, traits>;
def AddLweBuffersOp : BConcrete_Op<"add_lwe_buffer"> {
let arguments = (ins
1DTensorOf<[I64]>:$result,
1DTensorOf<[I64]>:$lhs,
1DTensorOf<[I64]>:$rhs
);
let results = (outs);
}
def AddPlaintextLweBufferOp : BConcrete_Op<"add_plaintext_lwe_buffer"> {
let arguments = (ins 1DTensorOf<[I64]>:$result, 1DTensorOf<[I64]>:$lhs, PlaintextType:$rhs);
let results = (outs);
}
def MulCleartextLweBufferOp : BConcrete_Op<"mul_cleartext_lwe_buffer"> {
let arguments = (ins 1DTensorOf<[I64]>:$result, 1DTensorOf<[I64]>:$lhs, CleartextType:$rhs);
let results = (outs);
}
def NegateLweBufferOp : BConcrete_Op<"negate_lwe_buffer"> {
let arguments = (ins 1DTensorOf<[I64]>:$result, 1DTensorOf<[I64]>:$ciphertext);
let results = (outs);
}
def KeySwitchLweBufferOp : BConcrete_Op<"keyswitch_lwe_buffer"> {
let arguments = (ins
1DTensorOf<[I64]>:$result,
// LweKeySwitchKeyType:$keyswitch_key,
1DTensorOf<[I64]>:$ciphertext,
I32Attr:$level,
I32Attr:$baseLog
);
let results = (outs);
}
def BootstrapLweBufferOp : BConcrete_Op<"bootstrap_lwe_buffer"> {
let arguments = (ins
1DTensorOf<[I64]>:$result,
// LweBootstrapKeyType:$bootstrap_key,
1DTensorOf<[I64]>:$input_ciphertext,
GlweCiphertextType:$accumulator,
I32Attr:$glweDimension,
I32Attr:$polynomialSize,
I32Attr:$level,
I32Attr:$baseLog
);
let results = (outs);
}
#endif

9
compiler/include/concretelang/Dialect/BConcrete/IR/CMakeLists.txt Normal file
View File

@@ -0,0 +1,9 @@
set(LLVM_TARGET_DEFINITIONS BConcreteOps.td)
mlir_tablegen(BConcreteOps.h.inc -gen-op-decls)
mlir_tablegen(BConcreteOps.cpp.inc -gen-op-defs)
mlir_tablegen(BConcreteOpsTypes.h.inc -gen-typedef-decls -typedefs-dialect=BConcrete)
mlir_tablegen(BConcreteOpsTypes.cpp.inc -gen-typedef-defs -typedefs-dialect=BConcrete)
mlir_tablegen(BConcreteOpsDialect.h.inc -gen-dialect-decls -dialect=BConcrete)
mlir_tablegen(BConcreteOpsDialect.cpp.inc -gen-dialect-defs -dialect=BConcrete)
add_public_tablegen_target(MLIRBConcreteOpsIncGen)
add_dependencies(mlir-headers MLIRBConcreteOpsIncGen)

1
compiler/include/concretelang/Dialect/CMakeLists.txt
View File

@@ -2,4 +2,5 @@ add_subdirectory(FHE)
add_subdirectory(FHELinalg)
add_subdirectory(TFHE)
add_subdirectory(Concrete)
add_subdirectory(BConcrete)
add_subdirectory(RT)

1
compiler/lib/Dialect/BConcrete/CMakeLists.txt Normal file
View File

@@ -0,0 +1 @@
add_subdirectory(IR)

26
compiler/lib/Dialect/BConcrete/IR/BConcreteDialect.cpp Normal file
View File

@@ -0,0 +1,26 @@
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
// Exceptions. See
// https://github.com/zama-ai/concrete-compiler-internal/blob/master/LICENSE.txt
// for license information.
#include "concretelang/Dialect/BConcrete/IR/BConcreteDialect.h"
#include "concretelang/Dialect/BConcrete/IR/BConcreteOps.h"
#define GET_TYPEDEF_CLASSES
#include "concretelang/Dialect/BConcrete/IR/BConcreteOpsTypes.cpp.inc"
#include "concretelang/Dialect/BConcrete/IR/BConcreteOpsDialect.cpp.inc"
using namespace mlir::concretelang::BConcrete;
void BConcreteDialect::initialize() {
addOperations<
#define GET_OP_LIST
#include "concretelang/Dialect/BConcrete/IR/BConcreteOps.cpp.inc"
>();
addTypes<
#define GET_TYPEDEF_LIST
#include "concretelang/Dialect/BConcrete/IR/BConcreteOpsTypes.cpp.inc"
>();
}

9
compiler/lib/Dialect/BConcrete/IR/BConcreteOps.cpp Normal file
View File

@@ -0,0 +1,9 @@
// Part of the Concrete Compiler Project, under the BSD3 License with Zama
// Exceptions. See
// https://github.com/zama-ai/concrete-compiler-internal/blob/master/LICENSE.txt
// for license information.
#include "concretelang/Dialect/BConcrete/IR/BConcreteOps.h"
#define GET_OP_CLASSES
#include "concretelang/Dialect/BConcrete/IR/BConcreteOps.cpp.inc"

14
compiler/lib/Dialect/BConcrete/IR/CMakeLists.txt Normal file
View File

@@ -0,0 +1,14 @@
add_mlir_dialect_library(BConcreteDialect
BConcreteDialect.cpp
BConcreteOps.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/concretelang/Dialect/BConcrete
DEPENDS
MLIRBConcreteOpsIncGen
LINK_LIBS PUBLIC
MLIRIR)
target_link_libraries(BConcreteDialect PUBLIC MLIRIR)

1
compiler/lib/Dialect/CMakeLists.txt
View File

@@ -2,4 +2,5 @@ add_subdirectory(FHELinalg)
add_subdirectory(FHE)
add_subdirectory(TFHE)
add_subdirectory(Concrete)
add_subdirectory(BConcrete)
add_subdirectory(RT)

3
compiler/lib/Support/CompilerEngine.cpp
View File

@@ -19,6 +19,7 @@
#include <mlir/ExecutionEngine/OptUtils.h>
#include <mlir/Parser.h>
#include <concretelang/Dialect/BConcrete/IR/BConcreteDialect.h>
#include <concretelang/Dialect/Concrete/IR/ConcreteDialect.h>
#include <concretelang/Dialect/FHE/IR/FHEDialect.h>
#include <concretelang/Dialect/FHELinalg/IR/FHELinalgDialect.h>
@@ -61,6 +62,8 @@ mlir::MLIRContext *CompilationContext::getMLIRContext() {
->getOrLoadDialect<mlir::concretelang::FHELinalg::FHELinalgDialect>();
this->mlirContext
->getOrLoadDialect<mlir::concretelang::Concrete::ConcreteDialect>();
this->mlirContext
->getOrLoadDialect<mlir::concretelang::BConcrete::BConcreteDialect>();
this->mlirContext->getOrLoadDialect<mlir::StandardOpsDialect>();
this->mlirContext->getOrLoadDialect<mlir::memref::MemRefDialect>();
this->mlirContext->getOrLoadDialect<mlir::linalg::LinalgDialect>();

2
compiler/src/CMakeLists.txt
View File

@@ -12,6 +12,7 @@ if(CONCRETELANG_PARALLEL_EXECUTION_ENABLED)
${conversion_libs}
MLIRTransforms
BConcreteDialect
ConcreteDialect
TFHEDialect
FHEDialect
@@ -37,6 +38,7 @@ else()
${conversion_libs}
MLIRTransforms
BConcreteDialect
ConcreteDialect
TFHEDialect
FHEDialect

61
compiler/tests/Dialect/BConcrete/ops.mlir Normal file
View File

@@ -0,0 +1,61 @@
// RUN: concretecompiler --action=roundtrip %s 2>&1| FileCheck %s
// CHECK-LABEL: func @add_lwe_ciphertexts(%arg0: tensor<2049xi64>, %arg1: tensor<2049xi64>) -> tensor<2049xi64>
func @add_lwe_ciphertexts(%arg0: tensor<2049xi64>, %arg1: tensor<2049xi64>) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.add_lwe_buffer"(%[[V0]], %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, tensor<2049xi64>) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.add_lwe_buffer"(%0, %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, tensor<2049xi64>) -> ()
return %0 : tensor<2049xi64>
}
// CHECK-LABEL: func @add_plaintext_lwe_ciphertext(%arg0: tensor<2049xi64>, %arg1: !Concrete.plaintext<5>) -> tensor<2049xi64>
func @add_plaintext_lwe_ciphertext(%arg0: tensor<2049xi64>, %arg1: !Concrete.plaintext<5>) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.add_plaintext_lwe_buffer"(%[[V0]], %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.plaintext<5>) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.add_plaintext_lwe_buffer"(%0, %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.plaintext<5>) -> ()
return %0 : tensor<2049xi64>
}
// CHECK-LABEL: func @mul_cleartext_lwe_ciphertext(%arg0: tensor<2049xi64>, %arg1: !Concrete.cleartext<7>) -> tensor<2049xi64>
func @mul_cleartext_lwe_ciphertext(%arg0: tensor<2049xi64>, %arg1: !Concrete.cleartext<7>) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.mul_cleartext_lwe_buffer"(%[[V0]], %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.cleartext<7>) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.mul_cleartext_lwe_buffer"(%0, %arg0, %arg1) : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.cleartext<7>) -> ()
return %0 : tensor<2049xi64>
}
// CHECK-LABEL: func @negate_lwe_ciphertext(%arg0: tensor<2049xi64>) -> tensor<2049xi64>
func @negate_lwe_ciphertext(%arg0: tensor<2049xi64>) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.negate_lwe_buffer"(%[[V0]], %arg0) : (tensor<2049xi64>, tensor<2049xi64>) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.negate_lwe_buffer"(%0, %arg0) : (tensor<2049xi64>, tensor<2049xi64>) -> ()
return %0 : tensor<2049xi64>
}
// CHECK-LABEL: func @bootstrap_lwe(%arg0: tensor<2049xi64>, %arg1: !Concrete.glwe_ciphertext) -> tensor<2049xi64>
func @bootstrap_lwe(%arg0: tensor<2049xi64>, %arg1: !Concrete.glwe_ciphertext) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.bootstrap_lwe_buffer"(%[[V0]], %arg0, %arg1) {baseLog = -1 : i32, glweDimension = 1 : i32, level = -1 : i32, polynomialSize = 1024 : i32} : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.glwe_ciphertext) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.bootstrap_lwe_buffer"(%0, %arg0, %arg1) {baseLog = -1 : i32, glweDimension = 1 : i32, level = -1 : i32, polynomialSize = 1024 : i32} : (tensor<2049xi64>, tensor<2049xi64>, !Concrete.glwe_ciphertext) -> ()
return %0 : tensor<2049xi64>
}
// CHECK-LABEL: func @keyswitch_lwe(%arg0: tensor<2049xi64>) -> tensor<2049xi64>
func @keyswitch_lwe(%arg0: tensor<2049xi64>) -> tensor<2049xi64> {
// CHECK-NEXT: %[[V0:.*]] = linalg.init_tensor [2049] : tensor<2049xi64>
// CHECK-NEXT: "BConcrete.keyswitch_lwe_buffer"(%[[V0]], %arg0) {baseLog = 2 : i32, inputLweDimension = 1 : i32, level = 3 : i32, outputLweDimension = 1 : i32} : (tensor<2049xi64>, tensor<2049xi64>) -> ()
// CHECK-NEXT: return %[[V0]] : tensor<2049xi64>
%0 = linalg.init_tensor [2049] : tensor<2049xi64>
"BConcrete.keyswitch_lwe_buffer"(%0, %arg0) {baseLog = 2 : i32, inputLweDimension = 1 : i32, level = 3 : i32, outputLweDimension = 1 : i32} : (tensor<2049xi64>, tensor<2049xi64>) -> ()
return %0 : tensor<2049xi64>
}