/llvm-project/mlir/test/Dialect/Linalg/ |
H A D | block-pack-matmul-layout.mlir | 44 // MMT4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red… 50 // MMT4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red… 56 // MMT4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red… 66 // MM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "redu… 72 // MM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "redu… 78 // MM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "redu… 88 // MTM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red… 94 // MTM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red… 100 // MTM4D-SAME: iterator_types = ["parallel", "parallel", "reduction", "parallel", "parallel", "red…
|
H A D | conv-interface-invalid.mlir | 20 iterator_types = [#test.iterator_type<parallel>]} 37 iterator_types = [#test.iterator_type<parallel>, 56 iterator_types = [#test.iterator_type<parallel>, 75 iterator_types = [#test.iterator_type<parallel>, 94 iterator_types = [#test.iterator_type<parallel>, 115 iterator_types = [#test.iterator_type<parallel>, 135 iterator_types = [#test.iterator_type<parallel>, 156 iterator_types = [#test.iterator_type<parallel>, 177 iterator_types = [#test.iterator_type<parallel>, 198 iterator_types = [#test.iterator_type<parallel>,
|
H A D | fusion-elementwise-ops.mlir | 14 …%3 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle… 23 …%4 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle… 56 …%3 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle… 65 …%4 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle… 98 …%3 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle… 107 …%4 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle… 132 …%3 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle… 141 …%4 = linalg.generic {indexing_maps = [#map1, #map0, #map0], iterator_types = ["parallel", "paralle… 166 %2 = linalg.generic {indexing_maps = [#map2, #map2, #map2], iterator_types = ["parallel"]} 177 …%5 = linalg.generic {indexing_maps = [#map1, #map0, #map0], iterator_types = ["parallel", "paralle… [all …]
|
H A D | transform-op-split-reduction.mlir | 21 // CHECK-SAME: , iterator_types = ["parallel", "parallel", "parallel", "reduction"]} 28 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"]} ins(%[[G]] : tensor<16x32x… 49 iterator_types = ["reduction"]} 73 // CHECK: iterator_types = ["parallel", "reduction"]} ins(%[[I1]], %{{.*}} : tensor<4x8xf32>… 79 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP3]], #[[$MAP4]]], iterator_types… 105 iterator_types = ["parallel", "reduction", "parallel"] 126 …inalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]], iterator_types = ["parallel",… 132 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP3]], #[[$MAP4]]], iterator_types… 163 iterator_types = ["parallel", "reduction", "parallel"] 184 …inalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]], iterator_types = ["parallel",… [all …]
|
H A D | drop-unit-extent-dims.mlir | 11 iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel"], 37 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel"] 53 // CHECK-SLICES-SAME: iterator_types = ["parallel", "parallel", "parallel"] 66 iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel"], 90 // CHECK-SAME: iterator_types = ["parallel"] 103 iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel"], 150 iterator_types = ["parallel", "parallel"], 170 // CHECK-SAME: iterator_types = [] 177 iterator_types = ["parallel", "parallel"], 212 iterator_types [all...] |
H A D | transform-op-specialize.mlir | 11 iterator_types = ["parallel", "parallel"]} 24 iterator_types = ["parallel", "parallel"]} 36 iterator_types = ["parallel", "parallel"]} 49 iterator_types = ["parallel", "parallel"]} 74 iterator_types = ["parallel", "parallel"]} 93 iterator_types = ["parallel", "parallel", "parallel"]} 139 %0 = linalg.generic {indexing_maps = [#map, #map1], iterator_types = ["parallel", "parallel"]} ins(%cst : f32) outs(%arg0 : tensor<7x7xf32>) {
|
H A D | collapse-dim.mlir | 9 iterator_types = ["parallel", "parallel", "reduction", "reduction"]} 24 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"]} 36 iterator_types = ["parallel", "parallel", "parallel", "parallel"]} 52 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel"]} 63 … indexing_maps = [#map, #map1], iterator_types = ["parallel", "parallel", "parallel", "parallel"]} 72 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel"] 83 // CHECK: linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel… 93 …d0, d1, d2, d3)>, affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>], iterator_types = ["parallel",… 105 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel"] 112 …d0, d1, d2, d3)>, affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>], iterator_types = ["parallel",…
|
H A D | invalid.mlir | 38 iterator_types = []} 52 iterator_types = []} 66 iterator_types = [] 76 iterator_types = []} 89 iterator_types = ["parallel"]} 102 iterator_types = ["random"]} 115 iterator_types = []} 129 iterator_types = []} 143 iterator_types = ["parallel"]} 160 iterator_types [all...] |
H A D | generalize-named-ops.mlir | 21 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"] 49 // CHECK: linalg.generic {indexing_maps = [#[[$ATTR_0]], #[[$ATTR_1]], #[[$ATTR_2]]], iterator_types = ["parallel", "parallel", "reduction"]} ins(%[[VAL_0]], %[[VAL_1]] : memref<5xf32>, memref<5x7xf32>) outs(%[[VAL_2]] : memref<3x7xf32>) { 119 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel", "reduction", "reduction"]} 146 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel", "reduction", "reduction"]} 172 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel", "reduction", "reduction"]} 198 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "reduction", "reduction"]} 224 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "reduction", "reduction"]} 251 // CHECK-SAME: iterator_types = ["parallel", "parallel", "parallel", "parallel", "parallel", "reduction", "reduction", "reduction"]} 279 // CHECK-SAME: iterator_types = ["parallel", "parallel"]} 301 // CHECK-SAME: iterator_types [all...] |
H A D | detensorize_0d.mlir | 7 %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} 26 %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} 35 %4 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} 44 %7 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} 66 %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []} 87 %1 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = []}
|
H A D | constant-fold.mlir | 10 iterator_types = ["parallel", "parallel"] 28 iterator_types = ["parallel", "parallel"] 52 iterator_types = ["parallel", "parallel", "parallel", "parallel"] 76 iterator_types = ["parallel", "parallel", "parallel", "parallel"] 92 iterator_types = ["parallel", "parallel"] 109 iterator_types = ["parallel", "parallel"] 125 iterator_types = ["parallel", "parallel"]
|
H A D | inline-scalar-operands.mlir | 11 // CHECK-SAME: iterator_types = ["parallel"]} ins(%[[ARG]] : tensor<4xf32>) 13 iterator_types = ["parallel"]} 36 // CHECK-SAME: iterator_types = ["parallel"]} ins(%[[ARG]] : tensor<4xf32>) 38 iterator_types = ["parallel"]} 60 // CHECK-SAME: iterator_types = ["parallel"]} ins(%[[ARG]] : tensor<4xf32>) 62 iterator_types = ["parallel"]}
|
H A D | erase-unused-operands-and-results.mlir | 16 …%3 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types=["parallel"]} ins(%arg0… 42 …%5 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types=["parallel","parallel"]… 63 %0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel"]} 86 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle… 111 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0, #map0], iterator_types = ["parallel", "… 133 …%0 = linalg.generic {indexing_maps = [#map, #map, #map, #map, #map], iterator_types = ["parallel"]} 155 iterator_types = ["parallel", "parallel", "parallel"]} 187 iterator_types = ["reduction"]} 220 iterator_types = ["reduction"]} 245 iterator_types = ["parallel"]} [all …]
|
H A D | int-narrowing.mlir |
|
H A D | specialize-generic-ops.mlir | 6 {indexing_maps = [#umap, #umap], iterator_types = ["parallel", "parallel","parallel"]} 25 {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]} 46 … {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "reduction"]} 68 …{indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "parallel", "redu… 94 iterator_types = ["parallel", "parallel", "reduction", "reduction"]} 116 {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "reduction"]}
|
H A D | bubble-up-extract-slice-op.mlir | 8 iterator_types = ["parallel", "parallel"] 24 …[GENERIC:.+]] = linalg.generic {indexing_maps = [#map, #map1, #map], iterator_types = ["parallel",… 35 iterator_types = ["parallel", "parallel"] 51 …[GENERIC:.+]] = linalg.generic {indexing_maps = [#map, #map1, #map], iterator_types = ["parallel",… 62 iterator_types = ["parallel", "parallel"] 78 …[GENERIC:.+]] = linalg.generic {indexing_maps = [#map, #map1, #map], iterator_types = ["parallel",… 89 iterator_types = ["parallel", "parallel"] 105 …[GENERIC:.+]] = linalg.generic {indexing_maps = [#map, #map1, #map], iterator_types = ["parallel",…
|
H A D | transform-op-specialize-elemwise-binary.mlir | 5 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]… 18 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]… 31 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]… 44 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]… 57 …%0 = linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel", "parallel"]…
|
/llvm-project/mlir/test/Integration/Dialect/Vector/CPU/ |
H A D | contraction.mlir | 13 iterator_types = ["reduction"] 23 iterator_types = ["parallel", "reduction"] 33 iterator_types = ["parallel", "reduction"] 43 iterator_types = ["parallel", "parallel", "reduction"] 53 iterator_types = ["parallel", "parallel", "reduction"] 63 iterator_types = ["parallel", "parallel", "reduction"] 73 iterator_types = ["parallel", "parallel", "reduction"] 83 iterator_types = ["parallel", "parallel", "reduction"] 93 iterator_types = ["reduction", "reduction"] 103 iterator_types [all...] |
/llvm-project/mlir/test/Dialect/Vector/ |
H A D | vector-contract-matmul-transforms.mlir | 11 iterator_types = ["reduction"], 26 iterator_types = ["parallel", "parallel", "reduction"], 40 iterator_types = ["parallel", "parallel", "reduction"], 58 iterator_types = ["parallel", "parallel", "reduction"], 73 iterator_types = ["parallel", "parallel", "reduction"], 91 iterator_types = ["parallel", "parallel", "reduction"], 105 iterator_types = ["parallel", "parallel", "reduction"], 120 iterator_types = ["parallel", "parallel", "reduction"], 139 iterator_types = ["parallel", "parallel", "reduction"], 152 iterator_types = ["parallel", "parallel", "reduction"], [all …]
|
H A D | fold-arith-extf-into-vector-contract.mlir | 10 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} 21 iterator_types = ["parallel", "parallel", "reduction"], 35 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} 46 iterator_types = ["parallel", "parallel", "reduction"], 57 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} 68 iterator_types = ["parallel", "parallel", "reduction"],
|
H A D | vector-reduce-to-contract.mlir | 9 // CHECK-SAME: iterator_types = ["parallel", "reduction", "parallel"], kind = #vector.kind<add>} 27 // CHECK-SAME: iterator_types = ["parallel", "reduction", "parallel"], kind = #vector.kind<add>} 50 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} 58 iterator_types = ["parallel", "parallel", "reduction"], 76 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} 84 iterator_types = ["parallel", "parallel", "reduction"], 106 // CHECK-SAME: iterator_types = ["parallel", "parallel", "reduction"] 113 iterator_types = ["reduction", "parallel", "parallel", "reduction"], 140 // CHECK-SAME: iterator_types = ["parallel", "reduction", "parallel", "reduction"] 147 iterator_types [all...] |
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
H A D | LinalgNamedStructuredOps.yaml | 34 iterator_types: [] 79 iterator_types: [] 120 iterator_types: [] 155 iterator_types: [] 190 iterator_types: [] 225 iterator_types: [] 260 iterator_types: [] 295 iterator_types: [] 330 iterator_types: [] 365 iterator_types [all...] |
/llvm-project/mlir/test/Integration/Dialect/Transform/ |
H A D | match_reduction.mlir | 139 iterator_types = ["parallel", "parallel"]} 154 iterator_types = ["parallel", "reduction"]} 177 iterator_types = ["parallel", "reduction"]} 189 iterator_types = ["parallel"]} 209 iterator_types = ["parallel", "parallel"]} 224 iterator_types = ["parallel", "reduction"]} 236 iterator_types = ["parallel"]} 255 iterator_types = ["parallel", "parallel"]} 273 iterator_types = ["parallel", "reduction"]} 285 iterator_types = ["parallel"]} [all …]
|
/llvm-project/mlir/docs/Tutorials/transform/ |
H A D | Ch0.md | 71 iterator_types = ["reduction"] 84 …d which ones are being preserved. This can be achieved by using the `iterator_types` attribute tha… 91 iterator_types = ["parallel", "parallel", "reduction"] 115 iterator_types = ["parallel", "parallel", "reduction"] 125 This looks more complicated, so let us unpack. The `indexing_maps` and `iterator_types` are _exactl… 142 iterator_types = ["parallel"] 165 iterator_types = ["parallel", "parallel", "reduction"] 216 iterator_types = ["parallel", "parallel", "reduction"] 269 iterator_types = ["parallel", "parallel", "reduction"] 288 iterator_types = ["parallel", "parallel"]
|
/llvm-project/mlir/test/Integration/Dialect/Linalg/CPU/ |
H A D | runtime-verification.mlir | 116 iterator_types = ["parallel"] 151 iterator_types = ["parallel", "parallel"] 169 iterator_types = ["parallel", "parallel"] 187 iterator_types = ["parallel", "parallel"] 205 iterator_types = ["parallel", "parallel"] 219 iterator_types = ["parallel", "parallel"] 234 iterator_types = ["parallel", "parallel", "reduction"], 267 iterator_types = ["parallel", "parallel", "reduction", "reduction"] 287 iterator_types = ["parallel"]
|