diff --git a/compiler/include/concretelang/Bindings/Python/CompilerAPIModule.h b/compiler/include/concretelang/Bindings/Python/CompilerAPIModule.h index db2328afe..f6ff667e6 100644 --- a/compiler/include/concretelang/Bindings/Python/CompilerAPIModule.h +++ b/compiler/include/concretelang/Bindings/Python/CompilerAPIModule.h @@ -18,4 +18,4 @@ void populateCompilerAPISubmodule(pybind11::module &m); } // namespace concretelang } // namespace mlir -#endif // CONCRETELANG_PYTHON_DIALECTMODULES_H +#endif // CONCRETELANG_BINDINGS_PYTHON_COMPILER_API_MODULE_H diff --git a/compiler/include/concretelang/Bindings/Python/DialectModules.h b/compiler/include/concretelang/Bindings/Python/DialectModules.h index 06441216e..42dd73659 100644 --- a/compiler/include/concretelang/Bindings/Python/DialectModules.h +++ b/compiler/include/concretelang/Bindings/Python/DialectModules.h @@ -18,4 +18,4 @@ void populateDialectFHESubmodule(pybind11::module &m); } // namespace concretelang } // namespace mlir -#endif // CONCRETELANG_PYTHON_DIALECTMODULES_H +#endif // CONCRETELANG_BINDINGS_PYTHON_DIALECTMODULES_H diff --git a/compiler/lib/Bindings/Rust/src/fhe.rs b/compiler/lib/Bindings/Rust/src/fhe.rs index 4a35a474a..c718e3d50 100644 --- a/compiler/lib/Bindings/Rust/src/fhe.rs +++ b/compiler/lib/Bindings/Rust/src/fhe.rs @@ -1,5 +1,6 @@ //! FHE dialect module +use crate::mlir::ffi::*; use crate::mlir::*; pub fn create_fhe_add_eint_op( diff --git a/compiler/lib/Bindings/Rust/src/fhelinalg.rs b/compiler/lib/Bindings/Rust/src/fhelinalg.rs index 63471e82f..8cc898b94 100644 --- a/compiler/lib/Bindings/Rust/src/fhelinalg.rs +++ b/compiler/lib/Bindings/Rust/src/fhelinalg.rs @@ -2,6 +2,7 @@ use crate::{ fhe::{convert_eint_to_esint_type, convert_esint_to_eint_type}, + mlir::ffi::*, mlir::*, }; use std::ffi::CString; diff --git a/compiler/lib/Bindings/Rust/src/mlir.rs b/compiler/lib/Bindings/Rust/src/mlir.rs index bd59ec1b0..c5fba68a3 100644 --- a/compiler/lib/Bindings/Rust/src/mlir.rs +++ b/compiler/lib/Bindings/Rust/src/mlir.rs @@ -4,8 +4,11 @@ #![allow(non_camel_case_types)] #![allow(non_snake_case)] -include!(concat!(env!("OUT_DIR"), "/bindings.rs")); +pub mod ffi { + include!(concat!(env!("OUT_DIR"), "/bindings.rs")); +} +use ffi::*; use std::ffi::CString; use std::ops::AddAssign; @@ -53,6 +56,7 @@ pub fn print_mlir_type_to_string(mlir_type: MlirType) -> String { /// # Examples /// ``` /// use concrete_compiler_rust::mlir::*; +/// use concrete_compiler_rust::mlir::ffi::*; /// unsafe{ /// let context = mlirContextCreate(); /// mlirRegisterAllDialects(context);