Home
last modified time | relevance | path

Searched full:indexing_maps (Results 1 – 25 of 283) sorted by relevance

12345678910>>...12

/llvm-project/mlir/test/Dialect/Linalg/
H A Dfusion-elementwise-ops.mlir14 …%3 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
22 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP0]], [[$MAP0]], [[$MAP0]], [[$MAP0]]{{\]}}
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…
64 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP0]], [[$MAP1]], [[$MAP1]], [[$MAP0]]{{\]}}
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…
106 // CHECK-SAME: indexing_maps = {{\[}}[[$MAP0]], [[$MAP1]], [[$MAP0]], [[$MAP0]]{{\]}}
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…
[all …]
H A Dblock-pack-matmul-layout.mlir43 // MMT4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
49 // MMT4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
55 // MMT4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
65 // MM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
71 // MM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
77 // MM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
87 // MTM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
93 // MTM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
99 // MTM4D-SAME: indexing_maps = [#[[$MAP]], #[[$MAP1]], #[[$MAP2]]]
H A Dconv-interface-invalid.mlir19 indexing_maps = [#map, #map],
34 indexing_maps = [affine_map<(d0, d1) -> (d0 * 2)>,
53 indexing_maps = [affine_map<(d0, d1) -> (d0 + d1, d0)>,
72 indexing_maps = [affine_map<(d0, d1) -> (d1)>,
91 indexing_maps = [affine_map<(d0, d1) -> (d0)>,
112 indexing_maps = [affine_map<(d0, d1) -> (d0 + d1)>,
132 indexing_maps = [affine_map<(d0, d1, d2) -> (d0 + d1)>,
153 indexing_maps = [affine_map<(d0, d1, d2) -> (d0 + d1)>,
174 indexing_maps = [affine_map<(d0, d1, d2) -> (d0 + d1, d2)>,
195 indexing_maps = [affine_map<(d0, d1) -> (d0 + d1)>,
H A Dtransform-op-split-reduction.mlir20 // CHECK: %[[G:.*]] = linalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]]
27 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP3]], #[[$MAP4]]],
46 %red = linalg.generic {indexing_maps = [affine_map<(d0) -> (d0)>,
72 // CHECK: {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]],
79 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP3]], #[[$MAP4]]], iterator_types…
100 indexing_maps = [
126 // CHECK: %[[G:.*]] = linalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]], it…
132 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP3]], #[[$MAP4]]], iterator_types…
158 indexing_maps = [
184 // CHECK: %[[G:.*]] = linalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]], #[[$MAP2]]], it…
[all …]
H A Derase-unused-operands-and-results.mlir16 …%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"]}
154 indexing_maps = [#map0, #map1, #map2, #map3, #map4],
171 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
186 indexing_maps = [#map0, #map1, #map1],
219 indexing_maps = [affine_map<(d0) -> (d0)>, affine_map<(d0) -> ()>],
[all …]
H A Ddrop-unit-extent-dims.mlir12 indexing_maps = #accesses,
36 // CHECK-SAME: indexing_maps = [#[[$MAP1]], #[[$MAP2]], #[[$MAP3]]]
52 // CHECK-SLICES-SAME: indexing_maps = [#[[$MAP1]], #[[$MAP2]], #[[$MAP3]]]
67 indexing_maps = #accesses,
89 // CHECK-SAME: indexing_maps = [#[[$MAP1]], #[[$MAP1]], #[[$MAP2]]]
104 indexing_maps = #accesses,
151 indexing_maps = #access,
169 // CHECK-SAME: indexing_maps = [#[[$MAP0]], #[[$MAP0]]]
178 indexing_maps = #access,
211 indexing_maps
[all...]
H A Dtransform-op-specialize.mlir10 indexing_maps = [#map, #map],
23 indexing_maps = [#map, #map2],
35 indexing_maps = [#map, #map],
48 indexing_maps = [#map, #map],
73 indexing_maps = [#map, #map],
92 indexing_maps = [#map1, #map1],
139 %0 = linalg.generic {indexing_maps = [#map, #map1], iterator_types = ["parallel", "parallel"]} ins(%cst : f32) outs(%arg0 : tensor<7x7xf32>) {
H A Dgeneralize-named-ops.mlir20 // CHECK-SAME: indexing_maps = [#[[A_MAP]], #[[B_MAP]], #[[C_MAP]]]
33 linalg.matmul indexing_maps = [
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>) {
118 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
145 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
171 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
197 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
223 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
250 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP2]], #[[MAP2]], #[[MAP3]]]
278 // CHECK-SAME: indexing_maps
[all...]
H A Ddetensorize_0d.mlir7 %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 Dconstant-fold.mlir9 indexing_maps = [affine_map<(d0, d1) -> (d1, d0)>, affine_map<(d0, d1) -> (d0, d1)>],
27 indexing_maps = [affine_map<(d0, d1) -> (d1, d0)>, affine_map<(d0, d1) -> (d0, d1)>],
51 …indexing_maps = [affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>, affine_map<(d0, d1, d2, d3) -> …
75 …indexing_maps = [affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>, affine_map<(d0, d1, d2, d3) -> …
91 indexing_maps = [affine_map<(d0, d1) -> (d1, d0)>, affine_map<(d0, d1) -> (d0, d1)>],
108 indexing_maps = [affine_map<(d0, d1) -> (d1, d0)>, affine_map<(d0, d1) -> (d0, d1)>],
124 indexing_maps = [affine_map<(d0, d1) -> (d1, d0)>, affine_map<(d0, d1) -> (d0, d1)>],
H A Dinline-scalar-operands.mlir10 // CHECK: linalg.generic {indexing_maps = [#[[MAP]], #[[MAP]]],
12 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
35 // CHECK: linalg.generic {indexing_maps = [#[[MAP]], #[[MAP]]],
37 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
59 // CHECK: linalg.generic {indexing_maps = [#[[MAP]], #[[MAP]]],
61 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
H A Dinvalid.mlir37 indexing_maps = [ affine_map<() -> ()> ],
51 indexing_maps = [ affine_map<() -> ()> ],
65 indexing_maps = [ affine_map<() -> (0)> ],
75 indexing_maps = [ affine_map<() -> ()> ],
88 indexing_maps = [ affine_map<() -> (0)> ],
101 indexing_maps = [ affine_map<(i) -> (i)> ],
114 indexing_maps = [ affine_map<() -> (0, 0)> ],
128 indexing_maps = [ affine_map<() -> (0)>, affine_map<() -> (0, 0)> ],
142 indexing_maps = [ affine_map<(i) -> (i)> ],
156 indexing_maps
[all...]
H A Dint-narrowing.mlir
H A Dspecialize-generic-ops.mlir6 {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…
93 {indexing_maps = [#mapA, #mapB, #mapC],
116 {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "reduction"]}
H A Dtransform-pack-greedily.mlir15 // CHECK-SAME: indexing_maps = [#[[$mk_kkmm]], #[[$kn_kknn]], #[[$mn_mmnn]]]
46 indexing_maps = #mkn_accesses,
58 // CHECK-SAME: indexing_maps = [#[[$mk_kkmm]], #[[$kn_kknn]], #[[$mn_mmnn]]]
93 indexing_maps = #mkn_accesses,
105 // CHECK-SAME: indexing_maps = [#[[$mk_kkmm]], #[[$kn_kknn]], #[[$mn_mmnn]]]
140 indexing_maps = #mkn_accesses,
152 // CHECK-SAME: indexing_maps = [#[[$bmkm2_kkmm]], #[[$nkb_kknn]], #[[$nbm_mmnn]]]
190 // CHECK-SAME: indexing_maps = [#[[$M1]], #[[$M2]], #[[$M3]]]
254 indexing_maps = #mkn_accesses,
266 // CHECK-SAME: indexing_maps = [#[[$mk_kkmm]], #[[$kn_kknn]], #[[$mn_mmnn]]]
[all …]
H A Dcollapse-dim.mlir6 indexing_maps = [
23 // CHECK: linalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]]],
33 indexing_maps = [
51 // CHECK: %[[R:.*]] = linalg.generic {indexing_maps = [#[[$MAP0]], #[[$MAP1]]],
63 … indexing_maps = [#map, #map1], iterator_types = ["parallel", "parallel", "parallel", "parallel"]}
83 // CHECK: linalg.generic {indexing_maps = [#map, #map, #map], iterator_types = ["parallel…
93 …linalg.generic {indexing_maps = [affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>, affine_map<(d0,…
112 …linalg.generic {indexing_maps = [affine_map<(d0, d1, d2, d3) -> (d0, d1, d2, d3)>, affine_map<(d0,…
H A Dreshape_control_fusion.mlir11 …indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>, affine_map<(d0, d1) -> (d1)>, affine_map<(d0, d…
31 // CHECK-SAME: indexing_maps = [#[[MAP0]], #[[MAP1]], #[[MAP0]]]
45 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>],
59 // CHECK-SAME: indexing_maps = [#[[MAP]]]
/llvm-project/mlir/python/mlir/dialects/linalg/opdsl/lang/
H A Demitter.py103 indexing_maps = [] # type: Sequence[AffineMap]
104 for curr in op_config.indexing_maps:
107 indexing_maps.append(curr)
111 indexing_maps = AffineMap.compress_unused_symbols(indexing_maps, Context.current)
112 if any(indexing_map.n_symbols != 0 for indexing_map in indexing_maps):
114 f"Expected indexing_maps to use no symbols after "
115 f"replacement and compression but got {indexing_maps}"
140 indexing_maps_attr = ArrayAttr.get([AffineMapAttr.get(am) for am in indexing_maps])
238 indexing_maps = []
241 indexing_maps.append(scalar_map)
[all …]
/llvm-project/mlir/test/Integration/Dialect/Vector/CPU/
H A Dcontraction.mlir12 indexing_maps = #dotp_accesses,
22 indexing_maps = #matvec_accesses,
32 indexing_maps = #mattransvec_accesses,
42 indexing_maps = #matmat_accesses,
52 indexing_maps = #mattransmat_accesses,
62 indexing_maps = #matmattrans_accesses,
72 indexing_maps = #mattransmattrans_accesses,
82 indexing_maps = #matmat_then_trans_accesses,
92 indexing_maps = #contract2d_accesses,
102 indexing_maps
[all...]
/llvm-project/mlir/test/Dialect/Vector/
H A Dvector-contract-matmul-transforms.mlir8 %0 = vector.contract {indexing_maps = [affine_map<(d0) -> (d0)>,
23 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
37 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
55 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
70 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
88 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
102 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d2, d0)>,
117 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d2, d0)>,
136 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d2, d0)>,
149 %res = vector.contract {indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>,
[all …]
H A Dvector-reduce-to-contract.mlir8 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map0]], #[[$map1]]],
26 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map0]], #[[$map1]]],
49 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
57 %1 = vector.contract {indexing_maps = [#map0, #map0, #map1],
75 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
83 %1 = vector.contract {indexing_maps = [#map0, #map0, #map1],
105 // CHECK-SAME: indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]]
112 indexing_maps = [#map0, #map1, #map2],
139 // CHECK-SAME: indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]]
146 indexing_maps
[all...]
H A Dfold-arith-extf-into-vector-contract.mlir9 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
20 …indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>, affine_map<(d0, d1, d2) -> (d2, d1)>, affin…
34 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
45 …indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>, affine_map<(d0, d1, d2) -> (d2, d1)>, affin…
67 …indexing_maps = [affine_map<(d0, d1, d2) -> (d0, d2)>, affine_map<(d0, d1, d2) -> (d2, d1)>, affin…
/llvm-project/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgNamedStructuredOps.yaml30 indexing_maps: !LinalgIndexingMapsConfig
75 indexing_maps: !LinalgIndexingMapsConfig
116 indexing_maps: !LinalgIndexingMapsConfig
151 indexing_maps: !LinalgIndexingMapsConfig
186 indexing_maps: !LinalgIndexingMapsConfig
221 indexing_maps: !LinalgIndexingMapsConfig
256 indexing_maps: !LinalgIndexingMapsConfig
291 indexing_maps: !LinalgIndexingMapsConfig
326 indexing_maps: !LinalgIndexingMapsConfig
361 indexing_maps
[all...]
/llvm-project/mlir/test/Integration/Dialect/Transform/
H A Dmatch_reduction.mlir137 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
152 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
175 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
187 indexing_maps = [affine_map<(d0) -> (d0)>,
207 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
222 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
234 indexing_maps = [affine_map<(d0) -> (d0)>,
253 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
271 indexing_maps = [affine_map<(d0, d1) -> (d0, d1)>,
283 indexing_maps = [affine_map<(d0) -> (d0)>,
[all …]
/llvm-project/mlir/test/Integration/Dialect/Linalg/CPU/
H A Druntime-verification.mlir115 indexing_maps = [#identity1D, #identity1D, #identity1D],
150 indexing_maps = [#broadcastD0, #identity2D],
168 indexing_maps = [#broadcastD1, #identity2D],
186 indexing_maps = [#broadcastD0, #identity2D],
204 indexing_maps = [#broadcastD1, #identity2D],
218 indexing_maps = [#identity2D, #identity2D, #identity2D],
235 indexing_maps = #matmul_accesses
266 indexing_maps = [affine_map<(d0, d1, d2, d3) -> (d0 * 3 + d2, d1 * 4 + d3)>, affine_map<(d0, d1, d2, d3) -> (d2, d3)>, affine_map<(d0, d1, d2, d3) -> (d0, d1)>],
283 indexing_maps = [

12345678910>>...12