Home
last modified time | relevance | path

Searched refs:map2 (Results 1 – 25 of 90) sorted by relevance

1234

/llvm-project/mlir/lib/Dialect/Utils/
H A DStructuredOpsUtils.cpp26 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isRowMajorMatmul() local
29 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isRowMajorMatmul()
30 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isRowMajorMatmul()
35 AffineExpr m = map2.getResult(0); in isRowMajorMatmul()
36 AffineExpr n = map2.getResult(1); in isRowMajorMatmul()
52 AffineMap map2 = cast<AffineMapAttr>(indexingMaps[2]).getValue(); in isColumnMajorMatmul() local
55 map2.getNumResults() != 2 || map0.getNumInputs() != 3 || in isColumnMajorMatmul()
56 map1.getNumInputs() != 3 || map2.getNumInputs() != 3) { in isColumnMajorMatmul()
61 AffineExpr n = map2.getResult(0); in isColumnMajorMatmul()
62 AffineExpr m = map2.getResult(1); in isColumnMajorMatmul()
[all …]
/llvm-project/polly/lib/External/isl/
H A Disl_map_subtract.c515 __isl_take isl_map *map2) in replace_pair_by_empty() argument
521 isl_map_free(map2); in replace_pair_by_empty()
534 __isl_take isl_map *map2) in isl_map_subtract() argument
540 if (isl_map_align_params_bin(&map1, &map2) < 0) in isl_map_subtract()
542 if (isl_map_check_equal_space(map1, map2) < 0) in isl_map_subtract()
545 equal = isl_map_plain_is_equal(map1, map2); in isl_map_subtract()
549 return replace_pair_by_empty(map1, map2); in isl_map_subtract()
551 disjoint = isl_map_is_disjoint(map1, map2); in isl_map_subtract()
555 isl_map_free(map2); in isl_map_subtract()
560 map2 = isl_map_compute_divs(map2); in isl_map_subtract()
[all …]
H A Disl_map.c295 __isl_keep isl_map *map2) in isl_map_has_equal_params() argument
300 space2 = isl_map_peek_space(map2); in isl_map_has_equal_params()
1608 __isl_keep isl_map *map2, in isl_map_align_params_map_map_and_test() argument
1609 isl_bool (*fn)(__isl_keep isl_map *map1, __isl_keep isl_map *map2)) in isl_map_align_params_map_map_and_test()
1613 if (!map1 || !map2) in isl_map_align_params_map_map_and_test()
1615 if (isl_map_has_equal_params(map1, map2)) in isl_map_align_params_map_map_and_test()
1616 return fn(map1, map2); in isl_map_align_params_map_map_and_test()
1619 if (isl_map_check_named_params(map2) < 0) in isl_map_align_params_map_map_and_test()
1622 map2 = isl_map_copy(map2); in isl_map_align_params_map_map_and_test()
1623 map1 = isl_map_align_params(map1, isl_map_get_space(map2)); in isl_map_align_params_map_map_and_test()
[all …]
H A Disl_test.c77 isl_map *map, *map2; in test_parse_map_equal() local
81 map2 = isl_map_read_from_str(ctx, str2); in test_parse_map_equal()
82 equal = isl_map_is_equal(map, map2); in test_parse_map_equal()
84 isl_map_free(map2); in test_parse_map_equal()
361 const char *map2; member
480 isl_map *map, *map2; in test_parse() local
512 str2 = parse_map_equal_tests[i].map2; in test_parse()
523 map2 = isl_map_read_from_str(ctx, str); in test_parse()
524 assert(isl_map_is_equal(map, map2)); in test_parse()
526 isl_map_free(map2); in test_parse()
[all …]
H A Disl_union_map.c913 __isl_take isl_map *map2);
939 __isl_keep isl_map *map1, __isl_keep isl_map *map2, in bin_add_pair() argument
945 map = data->control->fn_map(isl_map_copy(map1), isl_map_copy(map2)); in bin_add_pair()
1586 isl_map *map2 = *entry; in apply_range_entry() local
1590 map2, isl_dim_in); in apply_range_entry()
1596 map2 = isl_map_apply_range(isl_map_copy(data->map), isl_map_copy(map2)); in apply_range_entry()
1598 empty = isl_map_is_empty(map2); in apply_range_entry()
1600 isl_map_free(map2); in apply_range_entry()
1604 isl_map_free(map2); in apply_range_entry()
1608 data->res = isl_union_map_add_map(data->res, map2); in apply_range_entry()
[all …]
H A Disl_power_templ.c16 __isl_keep TYPE *map2) in FN()
20 res = FN(TYPE,apply_range)(map1, FN(TYPE,copy)(map2)); in FN()
/llvm-project/mlir/test/Dialect/Linalg/
H A Dinline-scalar-operands.mlir4 #map2 = affine_map<(d0) -> (d0)>
12 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
28 #map2 = affine_map<(d0) -> (d0)>
37 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
53 #map2 = affine_map<(d0) -> (d0)>
61 %1 = linalg.generic {indexing_maps = [#map2, #map3, #map2],
H A Dtransform-op-specialize-matmul.mlir5 #map2 = affine_map<(d0, d1, d2) -> (d0, d1)>
8 … {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "reduction"]}
35 #map2 = affine_map<(d0, d1, d2) -> (d0, d1)>
38 {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "reduction"]}
65 #map2 = affine_map<(d0, d1, d2) -> (d0, d1)>
68 … {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "reduction"]}
95 #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
98 …{indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "parallel", "redu…
124 #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
127 …{indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "parallel", "redu…
H A Dspecialize-generic-ops.mlir43 #map2 = affine_map<(d0, d1, d2) -> (d0, d1)>
46 … {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "reduction"]}
65 #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d1, d2)>
68 …{indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "parallel", "parallel", "redu…
113 #map2 = affine_map<(d0, d1) -> (d0)>
116 {indexing_maps = [#map, #map1, #map2], iterator_types = ["parallel", "reduction"]}
H A Derase-unused-operands-and-results.mlir149 #map2 = affine_map<(d0, d1, d2) -> (d1, d2, d0)>
154 indexing_maps = [#map0, #map1, #map2, #map3, #map4],
265 #map2 = affine_map<(d0, d1) -> (d0)>
270 indexing_maps = [#map3, #map0, #map0, #map2, #map1, #map1, #map2],
345 #map2 = affine_map<(d0, d1, d2) -> (d1, d2, d0)>
357 indexing_maps = [#map0, #map1, #map2, #map3, #map4],
384 #map2 = affine_map<(d0, d1, d2) -> (d1, d2, d0)>
396 indexing_maps = [#map0, #map1, #map2, #map3, #map4],
426 #map2 = affine_map<(d0, d1, d2) -> (d1, d2, d0)>
437 indexing_maps = [#map0, #map1, #map2, #map3],
[all …]
H A Dtransform-op-fuse-into-containing.mlir5 #map2 = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
22 %4 = affine.min #map2(%arg3)[%d0, %arg0]
61 #map2 = affine_map<(d0)[s0] -> (-(d0 * s0) + 64, s0)>
76 %4 = affine.min #map2(%arg3)[%arg0]
156 #map2 = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
173 %4 = affine.min #map2(%arg3)[%d0, %arg0]
207 #map2 = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
236 %4 = affine.min #map2(%i)[%d0, %idx]
308 #map2 = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
343 %4 = affine.min #map2(%i)[%d0, %idx]
[all …]
H A Dtile-to-forall.mlir99 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
145 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0) -> (d0 * 10)>
158 // CHECK: %[[LB0:.+]] = affine.apply #[[$map2]](%[[IV0]])
184 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0)[s0] -> (d0 * -10 + s0, 10)>
199 // CHECK-DAG: %[[TS0:.+]] = affine.min #[[$map2]](%[[IV0]])[%[[M]]]
227 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0) -> (d0 * 10)>
237 // CHECK-DAG: %[[LB0:.+]] = affine.apply #[[$map2]](%[[IV0]])
300 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0)[s0, s1] -> (-(d0 * s1) + s0, s1)>
343 // CHECK-DAG: #[[$map2:.+]] = affine_map<(d0) -> (d0 * 15)>
358 // CHECK: %[[LB:.+]] = affine.apply #[[$map2]](
[all...]
H A Ddetensorize_while_impure_cf.mlir6 #map2 = affine_map<(i) -> (i)>
14 indexing_maps = [#map2, #map1],
20 indexing_maps = [#map1, #map2],
/llvm-project/mlir/test/Dialect/Vector/
H A Dvector-reduce-to-contract.mlir44 // CHECK-DAG: #[[$map2:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
49 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
70 // CHECK-DAG: #[[$map2:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
75 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
96 #map2 = affine_map<(d0, d1, d2, d3) -> (d1, d2)>
100 // CHECK-DAG: #[[$map2:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
105 // CHECK-SAME: indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]]
112 indexing_maps = [#map0, #map1, #map2],
128 #map2 = affine_map<(d0, d1, d2, d3) -> (d0, d2)>
132 // CHECK-DAG: #[[$map2
[all...]
H A Dfold-arith-extf-into-vector-contract.mlir6 // CHECK-DAG: #[[$map2:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
9 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
31 // CHECK-DAG: #[[$map2:.*]] = affine_map<(d0, d1, d2) -> (d0, d1)>
34 // CHECK-NEXT: %[[R:.+]] = vector.contract {indexing_maps = [#[[$map0]], #[[$map1]], #[[$map2]]],
/llvm-project/mlir/test/python/ir/
H A Daffine_map.py45 map2 = AffineMap.get(0, 0, [c2])
46 print(map2)
62 assert map2 == AffineMap.get_constant(2)
130 map2 = AffineMap.get(3, 0, [d2, d0, d1])
137 print(map2.is_permutation)
139 print(map2.is_projected_permutation)
/llvm-project/llvm/unittests/Support/
H A DYAMLIOTest.cpp137 FooBar& map2 = seq[1]; in TEST() local
140 EXPECT_EQ(map2.foo, 7); in TEST()
141 EXPECT_EQ(map2.bar, 9); in TEST()
251 FooBar& map2 = seq2[1]; in TEST() local
254 EXPECT_EQ(map2.foo, 257); in TEST()
255 EXPECT_EQ(map2.bar, 0); in TEST()
1031 FlagsMap map2; in TEST() local
1032 yin >> map2; in TEST()
1035 EXPECT_EQ(flagBig, map2.f1); in TEST()
1036 EXPECT_EQ(flagRound|flagFlat, map2 in TEST()
1122 MyCustomTypeMap map2; TEST() local
1211 MultilineStringTypeMap map2; TEST() local
1388 NameAndNumbers map2; TEST() local
1435 NameAndNumbersFlow map2; TEST() local
1745 FooBarMap& map2 = docList2[1]; TEST() local
[all...]
/llvm-project/mlir/test/Conversion/VectorToGPU/
H A Dvector-to-mma-ops.mlir5 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
23 %D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} %A, %B, %C : vector<16x16xf16>, vector<16x16xf16> into vector<16x16xf16>
32 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
50 %D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} %A, %B, %cst_0 : vector<16x16xf16>, vector<16x16xf16> into vector<16x16xf16>
59 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
77 %D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} %A, %B, %C : vector<16x16xf16>, vector<16x16xf16> into vector<16x16xf16>
86 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
109 %19 = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parallel", "reduction"], kind = #vector.kind<add>} %17, %18, %arg18 : vector<16x16xf16>, vector<16x16xf16> into vector<16x16xf16>
120 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
142 %D = vector.contract {indexing_maps = [#map1, #map2, #map
[all...]
H A Dvector-to-mma-ops-mma-sync.mlir27 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
92 …%D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parall…
120 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
157 …%D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parall…
174 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
195 …%D = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "parall…
211 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
240 …%D0 = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "paral…
248 …%D1 = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "paral…
264 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
[all …]
H A Dfold-arith-vector-to-mma-ops-mma-sync.mlir9 #map2 = affine_map<(d0, d1, d2) -> (d1, d2)>
33 …%D0 = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "paral…
42 …%D1 = vector.contract {indexing_maps = [#map1, #map2, #map3], iterator_types = ["parallel", "paral…
/llvm-project/polly/lib/External/isl/include/isl/
H A Dmap.h235 __isl_take isl_map *map2);
333 __isl_take isl_map *map2);
335 __isl_take isl_map *map1, __isl_take isl_map *map2);
365 __isl_take isl_map *map2);
369 __isl_take isl_map *map2);
389 __isl_take isl_map *map2);
396 __isl_take isl_map *map2);
399 __isl_take isl_map *map2);
403 __isl_take isl_map *map2);
407 __isl_take isl_map *map2);
[all …]
/llvm-project/compiler-rt/lib/sanitizer_common/
H A Dsanitizer_flat_map.h102 T *map2 = GetOrCreate(idx / kSize2); variable
103 return *AddressSpaceView::Load(&map2[idx % kSize2]);
108 T *map2 = GetOrCreate(idx / kSize2); variable
109 return *AddressSpaceView::LoadWritable(&map2[idx % kSize2]);
/llvm-project/mlir/test/Dialect/Affine/
H A Daffine-expand-index-ops.mlir
/llvm-project/mlir/utils/tree-sitter-mlir/test/highlight/
H A Dattribute.mlir70 #map2 = affine_map<(d0) -> (d0)>
81 %2 = linalg.generic {indexing_maps = [#map2, #map2, #map2], iterator_types = ["parallel"]}
/llvm-project/mlir/test/Dialect/SparseTensor/
H A Dconstant_index_map.mlir6 #map2 = affine_map<(d0) -> (d0)>
32 indexing_maps = [#map1, #map1, #map2],

1234