01a31cee56
Introduces a SubscriptOp that allows to write IR like
```
func.func @load_store(%arg0: !emitc.array<4x8xf32>, %arg1: !emitc.array<3x5xf32>, %arg2: index, %arg3: index) {
%0 = emitc.subscript %arg0[%arg2, %arg3] : <4x8xf32>, index, index
%1 = emitc.subscript %arg1[%arg2, %arg3] : <3x5xf32>, index, index
emitc.assign %0 : f32 to %1 : f32
return
}
```
which gets translated into the C++ code
```
v1[v2][v3] = v0[v1][v2];
```
To make this happen, this
- adds the SubscriptOp
- allows the subscript op as rhs of emitc.assign
- updates the emitter to print SubscriptOps
The emitter prints emitc.subscript in a delayed fashing to allow it
being used as lvalue.
I.e. while processing
```
%0 = emitc.subscript %arg0[%arg2, %arg3] : <4x8xf32>, index, index
```
it will not emit any text, but record in the `valueMapper` that the name
for `%0` is `v0[v1][v2]`, see `CppEmitter::getSubscriptName`. Only when
that result is then used (here in `emitc.assign`), that name is inserted
into the text.
33 lines
1.4 KiB
MLIR
33 lines
1.4 KiB
MLIR
// RUN: mlir-translate -mlir-to-cpp %s | FileCheck %s
|
|
// RUN: mlir-translate -mlir-to-cpp -declare-variables-at-top %s | FileCheck %s
|
|
|
|
func.func @load_store(%arg0: !emitc.array<4x8xf32>, %arg1: !emitc.array<3x5xf32>, %arg2: index, %arg3: index) {
|
|
%0 = emitc.subscript %arg0[%arg2, %arg3] : <4x8xf32>, index, index
|
|
%1 = emitc.subscript %arg1[%arg2, %arg3] : <3x5xf32>, index, index
|
|
emitc.assign %0 : f32 to %1 : f32
|
|
return
|
|
}
|
|
// CHECK: void load_store(float [[ARR1:[^ ]*]][4][8], float [[ARR2:[^ ]*]][3][5],
|
|
// CHECK-SAME: size_t [[I:[^ ]*]], size_t [[J:[^ ]*]])
|
|
// CHECK-NEXT: [[ARR2]][[[I]]][[[J]]] = [[ARR1]][[[I]]][[[J]]];
|
|
|
|
emitc.func @func1(%arg0 : f32) {
|
|
emitc.return
|
|
}
|
|
|
|
emitc.func @call_arg(%arg0: !emitc.array<4x8xf32>, %i: i32, %j: i16,
|
|
%k: i8) {
|
|
%0 = emitc.subscript %arg0[%i, %j] : <4x8xf32>, i32, i16
|
|
%1 = emitc.subscript %arg0[%j, %k] : <4x8xf32>, i16, i8
|
|
|
|
emitc.call @func1 (%0) : (f32) -> ()
|
|
emitc.call_opaque "func2" (%1) : (f32) -> ()
|
|
emitc.call_opaque "func3" (%0, %1) { args = [1 : index, 0 : index] } : (f32, f32) -> ()
|
|
emitc.return
|
|
}
|
|
// CHECK: void call_arg(float [[ARR1:[^ ]*]][4][8], int32_t [[I:[^ ]*]],
|
|
// CHECK-SAME: int16_t [[J:[^ ]*]], int8_t [[K:[^ ]*]])
|
|
// CHECK-NEXT: func1([[ARR1]][[[I]]][[[J]]]);
|
|
// CHECK-NEXT: func2([[ARR1]][[[J]]][[[K]]]);
|
|
// CHECK-NEXT: func3([[ARR1]][[[J]]][[[K]]], [[ARR1]][[[I]]][[[J]]]);
|