mirror of
https://github.com/zama-ai/concrete.git
synced 2026-04-17 03:00:54 -04:00
refactor: redesign GPU support
- unify CPU and GPU bootstrapping operations - remove operations to build GLWE from table: this is now done in wrapper functions - remove GPU memory management operations: done in wrappers now, but we will have to think about how to deal with it later in MLIR
This commit is contained in:
@@ -42,6 +42,9 @@ mlir::Value getContextArgument(mlir::Operation *op) {
|
||||
mlir::Block *block = op->getBlock();
|
||||
while (block != nullptr) {
|
||||
if (llvm::isa<mlir::func::FuncOp>(block->getParentOp())) {
|
||||
block = &mlir::cast<mlir::func::FuncOp>(block->getParentOp())
|
||||
.getBody()
|
||||
.front();
|
||||
|
||||
auto context = std::find_if(
|
||||
block->getArguments().rbegin(), block->getArguments().rend(),
|
||||
|
||||
Reference in New Issue
Block a user