608a663c8e
Before the change `test-loop-fusion` and `affine-super-vectorizer-test`
options were in their own category. This was because they used the
standard llvm command line parsing with `llvm::cl::opt`. This PR moves
them over to the mlir `Pass::Option` class.
Before the change
```
$ mlir-opt --help
...
General options:
...
Compiler passes to run
Passes:
...
Pass Pipelines:
...
Generic Options:
....
affine-super-vectorizer-test options:
--backward-slicing
...
--vectorize-affine-loop-nest
test-loop-fusion options:
--test-loop-fusion-dependence-check
...
--test-loop-fusion-transformation
```
After the change
```
$ mlir-opt --help
...
General options:
...
Compiler passes to run
Passes:
...
--affine-super-vectorizer-test
--backward-slicing
...
--vectorize-affine-loop-nest
...
--test-loop-fusion options:
--test-loop-fusion-dependence-check
...
--test-loop-fusion-transformation
...
Pass Pipelines:
...
Generic Options:
...
```
---------
Signed-off-by: philass <plassen@groq.com>
170 lines
6.0 KiB
MLIR
170 lines
6.0 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect %s -affine-super-vectorizer-test=compose-maps -split-input-file 2>&1 | FileCheck %s
|
|
|
|
// For all these cases, the test traverses the `test_affine_map` ops and
|
|
// composes them in order one-by-one.
|
|
// For instance, the pseudo-sequence:
|
|
// "test_affine_map"() { affine_map = f } : () -> ()
|
|
// "test_affine_map"() { affine_map = g } : () -> ()
|
|
// "test_affine_map"() { affine_map = h } : () -> ()
|
|
// will produce the sequence of compositions: f, g(f), h(g(f)) and print the
|
|
// AffineMap h(g(f)), which is what FileCheck checks against.
|
|
|
|
func.func @simple1() {
|
|
// CHECK: Composed map: (d0) -> (d0)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 1)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple2() {
|
|
// CHECK: Composed map: (d0)[s0, s1] -> (d0 - s0 + s1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0)[s0] -> (d0 + s0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0)[s0] -> (d0 - s0 + 1)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple3a() {
|
|
// CHECK: Composed map: (d0, d1)[s0, s1, s2, s3] -> ((d0 ceildiv s2) * s0, (d1 ceildiv s3) * s1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> (d0 ceildiv s0, d1 ceildiv s1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> (d0 * s0, d1 * s1)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple3b() {
|
|
// CHECK: Composed map: (d0, d1)[s0, s1] -> (d0 mod s0, d1 mod s1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> (d0 mod s0, d1 mod s1)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple3c() {
|
|
// CHECK: Composed map: (d0, d1)[s0, s1, s2, s3, s4, s5] -> ((d0 ceildiv s4) * s4 + d0 mod s2, (d1 ceildiv s5) * s5 + d1 mod s3)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> ((d0 ceildiv s0) * s0, (d1 ceildiv s1) * s1, d0, d1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1, d2, d3)[s0, s1, s2, s3] -> (d0 + d2 mod s2, d1 + d3 mod s3)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple4() {
|
|
// CHECK: Composed map: (d0, d1)[s0, s1] -> (d1 * s1, d0 ceildiv s0)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1) -> (d1, d0)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> (d0 * s1, d1 ceildiv s0)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5a() {
|
|
// CHECK: Composed map: (d0) -> (d0 * 3 + 18)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 7)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 * 24)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 ceildiv 8)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5b() {
|
|
// CHECK: Composed map: (d0) -> ((d0 + 6) ceildiv 2)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 7)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 * 4)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 ceildiv 8)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5c() {
|
|
// CHECK: Composed map: (d0) -> (d0 * 8 + 48)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 7)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 * 24)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 floordiv 3)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5d() {
|
|
// CHECK: Composed map: (d0) -> ((d0 * 4) floordiv 3 + 8)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 7)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 * 4)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 floordiv 3)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5e() {
|
|
// CHECK: Composed map: (d0) -> ((d0 + 6) ceildiv 8)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 + 7)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 ceildiv 8)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @simple5f() {
|
|
// CHECK: Composed map: (d0) -> ((d0 * 4 - 4) floordiv 3)
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 - 1)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 * 4)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0) -> (d0 floordiv 3)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @perm_and_proj() {
|
|
// CHECK: Composed map: (d0, d1, d2, d3) -> (d1, d3, d0)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1, d2, d3) -> (d3, d1, d2, d0)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1, d2, d3) -> (d1, d0, d3)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @symbols1() {
|
|
// CHECK: Composed map: (d0)[s0] -> (d0 + s0 + 1, d0 - s0 - 1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0)[s0] -> (d0 + s0, d0 - s0)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1) -> (d0 + 1, d1 - 1)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @drop() {
|
|
// CHECK: Composed map: (d0, d1, d2)[s0, s1] -> (d0 * 2 + d1 + d2 + s1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1, d2)[s0, s1] -> (d0 + s1, d1 + s0, d0 + d1 + d2)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1, d2) -> (d0 + d2)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
func.func @multi_symbols() {
|
|
// CHECK: Composed map: (d0)[s0, s1, s2] -> (d0 + s1 + s2 + 1, d0 - s0 - s2 - 1)
|
|
"test_affine_map"() { affine_map = affine_map<(d0)[s0] -> (d0 + s0, d0 - s0)> } : () -> ()
|
|
"test_affine_map"() { affine_map = affine_map<(d0, d1)[s0, s1] -> (d0 + 1 + s1, d1 - 1 - s0)> } : () -> ()
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: @no_affine_maps
|
|
func.func @no_affine_maps() {
|
|
// CHECK: return
|
|
return
|
|
}
|