| Safe Haskell | None |
|---|---|
| Language | Haskell2010 |
MLIR.AST.Dialect.ControlFlow
Synopsis
- pattern Branch :: Location -> BlockName -> [Name] -> Operation
- br :: MonadBlockBuilder m => BlockName -> [Value] -> m EndOfBlock
- cond_br :: MonadBlockBuilder m => Value -> BlockName -> [Value] -> BlockName -> [Value] -> m EndOfBlock
- pattern Assert :: Location -> operand -> ByteString -> AbstractOperation operand
- assert :: MonadBlockBuilder m => Value -> ByteString -> m ()
Documentation
br :: MonadBlockBuilder m => BlockName -> [Value] -> m EndOfBlock Source #
cond_br :: MonadBlockBuilder m => Value -> BlockName -> [Value] -> BlockName -> [Value] -> m EndOfBlock Source #
assert
Assert operation at runtime with single boolean operand and an error
message attribute.
If the argument is true this operation has no effect. Otherwise, the
program execution will abort. The provided error message may be used by a
runtime to propagate the error to the user.
Example:
cf.assert %b, "Expected ... to be true"
pattern Assert :: Location -> operand -> ByteString -> AbstractOperation operand Source #
A pattern for cf.assert.
assert :: MonadBlockBuilder m => Value -> ByteString -> m () Source #
A builder for cf.assert.
br
The cf.br operation represents a direct branch operation to a given
block. The operands of this operation are forwarded to the successor block,
and the number and type of the operands must match the arguments of the
target block.
Example:
^bb2: %2 = call @someFn() cf.br ^bb3(%2 : tensor<*xf32>) ^bb3(%3: tensor<*xf32>):
cond_br
The cf.cond_br terminator operation represents a conditional branch on a
boolean (1-bit integer) value. If the bit is set, then the first destination
is jumped to; if it is false, the second destination is chosen. The count
and types of operands must align with the arguments in the corresponding
target blocks.
The MLIR conditional branch operation is not allowed to target the entry block for a region. The two destinations of the conditional branch operation are allowed to be the same.
The following example illustrates a function with a conditional branch operation that targets the same block.
Example:
func.func @select(%a: i32, %b: i32, %flag: i1) -> i32 {
// Both targets are the same, operands differ
cf.cond_br %flag, ^bb1(%a : i32), ^bb1(%b : i32)
^bb1(%x : i32) :
return %x : i32
}
switch
The cf.switch terminator operation represents a switch on a signless integer
value. If the flag matches one of the specified cases, then the
corresponding destination is jumped to. If the flag does not match any of
the cases, the default destination is jumped to. The count and types of
operands must align with the arguments in the corresponding target blocks.
Example:
cf.switch %flag : i32, [ default: ^bb1(%a : i32), 42: ^bb1(%b : i32), 43: ^bb3(%c : i32) ]