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

feat: support GPU (bootstrapping)

Browse Source
This commit is contained in:
youben11
2022-07-21 14:45:28 +01:00
committed by Ayoub Benaissa
parent a487b03699
commit d169a27fc0
26 changed files with 715 additions and 47 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
compiler/lib/Conversion/Tools.cpp
View File

@@ -6,6 +6,7 @@
#include "mlir/Dialect/Func/IR/FuncOps.h"
#include "concretelang/Conversion/Tools.h"
#include "concretelang/Dialect/Concrete/IR/ConcreteTypes.h"
mlir::LogicalResult insertForwardDeclaration(mlir::Operation *op,
mlir::OpBuilder &rewriter,
@@ -35,3 +36,27 @@ mlir::LogicalResult insertForwardDeclaration(mlir::Operation *op,
mlir::SymbolTable::lookupSymbolIn(module, funcName)));
return mlir::success();
}
/// Returns the value of the context argument from the enclosing func
mlir::Value getContextArgument(mlir::Operation *op) {
mlir::Block *block = op->getBlock();
while (block != nullptr) {
if (llvm::isa<mlir::func::FuncOp>(block->getParentOp())) {
auto context = std::find_if(
block->getArguments().rbegin(), block->getArguments().rend(),
[](mlir::BlockArgument &arg) {
return arg.getType()
.isa<mlir::concretelang::Concrete::ContextType>();
});
assert(context != block->getArguments().rend() &&
"Cannot find the Concrete.context");
return *context;
}
block = block->getParentOp()->getBlock();
}
assert("can't find a function that enclose the op");
return nullptr;
}