| //===- Intrinsics.cpp - MLIR Operations for Declarative Builders ----------===// |
| // |
| // Part of the MLIR Project, under the Apache License v2.0 with LLVM Exceptions. |
| // See https://llvm.org/LICENSE.txt for license information. |
| // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception |
| // |
| //===----------------------------------------------------------------------===// |
| |
| #include "mlir/EDSC/Intrinsics.h" |
| #include "mlir/EDSC/Builders.h" |
| #include "mlir/IR/AffineExpr.h" |
| |
| using namespace mlir; |
| using namespace mlir::edsc; |
| |
| OperationHandle mlir::edsc::intrinsics::br(BlockHandle bh, |
| ArrayRef<ValueHandle> operands) { |
| assert(bh && "Expected already captured BlockHandle"); |
| for (auto &o : operands) { |
| (void)o; |
| assert(o && "Expected already captured ValueHandle"); |
| } |
| SmallVector<Value, 4> ops(operands.begin(), operands.end()); |
| return OperationHandle::create<BranchOp>(bh.getBlock(), ops); |
| } |
| static void enforceEmptyCapturesMatchOperands(ArrayRef<ValueHandle *> captures, |
| ArrayRef<ValueHandle> operands) { |
| assert(captures.size() == operands.size() && |
| "Expected same number of captures as operands"); |
| for (auto it : llvm::zip(captures, operands)) { |
| (void)it; |
| assert(!std::get<0>(it)->hasValue() && |
| "Unexpected already captured ValueHandle"); |
| assert(std::get<1>(it) && "Expected already captured ValueHandle"); |
| assert(std::get<0>(it)->getType() == std::get<1>(it).getType() && |
| "Expected the same type for capture and operand"); |
| } |
| } |
| |
| OperationHandle mlir::edsc::intrinsics::br(BlockHandle *bh, |
| ArrayRef<ValueHandle *> captures, |
| ArrayRef<ValueHandle> operands) { |
| assert(!*bh && "Unexpected already captured BlockHandle"); |
| enforceEmptyCapturesMatchOperands(captures, operands); |
| BlockBuilder(bh, captures)(/* no body */); |
| SmallVector<Value, 4> ops(operands.begin(), operands.end()); |
| return OperationHandle::create<BranchOp>(bh->getBlock(), ops); |
| } |
| |
| OperationHandle |
| mlir::edsc::intrinsics::cond_br(ValueHandle cond, BlockHandle trueBranch, |
| ArrayRef<ValueHandle> trueOperands, |
| BlockHandle falseBranch, |
| ArrayRef<ValueHandle> falseOperands) { |
| SmallVector<Value, 4> trueOps(trueOperands.begin(), trueOperands.end()); |
| SmallVector<Value, 4> falseOps(falseOperands.begin(), falseOperands.end()); |
| return OperationHandle::create<CondBranchOp>( |
| cond, trueBranch.getBlock(), trueOps, falseBranch.getBlock(), falseOps); |
| } |
| |
| OperationHandle mlir::edsc::intrinsics::cond_br( |
| ValueHandle cond, BlockHandle *trueBranch, |
| ArrayRef<ValueHandle *> trueCaptures, ArrayRef<ValueHandle> trueOperands, |
| BlockHandle *falseBranch, ArrayRef<ValueHandle *> falseCaptures, |
| ArrayRef<ValueHandle> falseOperands) { |
| assert(!*trueBranch && "Unexpected already captured BlockHandle"); |
| assert(!*falseBranch && "Unexpected already captured BlockHandle"); |
| enforceEmptyCapturesMatchOperands(trueCaptures, trueOperands); |
| enforceEmptyCapturesMatchOperands(falseCaptures, falseOperands); |
| BlockBuilder(trueBranch, trueCaptures)(/* no body */); |
| BlockBuilder(falseBranch, falseCaptures)(/* no body */); |
| SmallVector<Value, 4> trueOps(trueOperands.begin(), trueOperands.end()); |
| SmallVector<Value, 4> falseOps(falseOperands.begin(), falseOperands.end()); |
| return OperationHandle::create<CondBranchOp>( |
| cond, trueBranch->getBlock(), trueOps, falseBranch->getBlock(), falseOps); |
| } |