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

enhance(compiler): Lower linalg to parallel loops and instantiare parallel loops with openmp

Browse Source
This commit is contained in:
Quentin Bourgerie
2022-01-28 07:21:53 +01:00
committed by Antoniu Pop
parent 260768e9af
commit f52965e22f
4 changed files with 21 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -13,6 +13,7 @@
#include <mlir/Dialect/LLVMIR/LLVMDialect.h>
#include <mlir/Dialect/Linalg/IR/LinalgOps.h>
#include <mlir/Dialect/MemRef/IR/MemRef.h>
#include <mlir/Dialect/OpenMP/OpenMPDialect.h>
#include <mlir/Dialect/SCF/SCF.h>
#include <mlir/Dialect/StandardOps/IR/Ops.h>
#include <mlir/ExecutionEngine/OptUtils.h>
@@ -65,6 +66,7 @@ mlir::MLIRContext *CompilationContext::getMLIRContext() {
this->mlirContext->getOrLoadDialect<mlir::linalg::LinalgDialect>();
this->mlirContext->getOrLoadDialect<mlir::LLVM::LLVMDialect>();
this->mlirContext->getOrLoadDialect<mlir::scf::SCFDialect>();
this->mlirContext->getOrLoadDialect<mlir::omp::OpenMPDialect>();
}
return this->mlirContext;