• Home
  • Raw
  • Download

Lines Matching refs:linalg

1 // RUN: mlir-opt %s -linalg-fusion-for-tensor-ops -split-input-file | FileCheck %s
9 …%0 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
13 linalg.yield %1 : f32
15 // CHECK: linalg.generic {
17 …%2 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
25 // CHECK-NOT: linalg.yield
27 // CHECK: linalg.yield
29 linalg.yield %3 : f32
44 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle…
48 linalg.yield %1 : f32
50 // CHECK: linalg.generic {
52 …%2 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = ["parallel", "paralle…
56 linalg.yield %3 : f32
71 …%0 = linalg.generic {indexing_maps = [#map0, #map1, #map0], iterator_types = ["parallel", "paralle…
75 linalg.yield %1 : f32
77 // CHECK: linalg.generic {
79 …%2 = linalg.generic {indexing_maps = [#map1, #map0, #map0], iterator_types = ["parallel", "paralle…
83 linalg.yield %3 : f32
99 %0 = linalg.generic {indexing_maps = [#map2, #map2, #map2], iterator_types = ["parallel"]}
103 linalg.yield %1 : f32
105 // CHECK: linalg.generic {
107 …%2 = linalg.generic {indexing_maps = [#map1, #map0, #map0], iterator_types = ["parallel", "paralle…
111 linalg.yield %3 : f32
124 %0 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = []}
128 linalg.yield %1 : f32
130 // CHECK: linalg.generic {
133 %1 = linalg.generic {indexing_maps = [#map0, #map0, #map0], iterator_types = []}
137 linalg.yield %1 : f32
150 %1 = linalg.generic {
156 linalg.yield %2 : f32
163 // CHECK: linalg.generic
175 %1 = linalg.indexed_generic {
181 linalg.yield %2 : f32
188 // CHECK: linalg.indexed_generic
204 %1 = linalg.generic {
210 linalg.yield %2 : f32
217 // CHECK: linalg.generic
229 %1 = linalg.indexed_generic {
235 linalg.yield %2 : f32
242 // CHECK: linalg.indexed_generic
255 %0 = linalg.generic {
261 linalg.yield %10 : i32
263 %1 = linalg.indexed_generic {
272 linalg.yield %5 : i32
278 // CHECK-NOT: linalg.generic
279 // CHECK: linalg.indexed_generic
291 // CHECK: linalg.yield %[[VAL3]] : i32
298 %0 = linalg.indexed_generic {
307 linalg.yield %5 : i32
309 %1 = linalg.generic {
315 linalg.yield %10 : i32
321 // CHECK: linalg.indexed_generic
333 // CHECK: linalg.yield %[[VAL3]] : i32
334 // CHECK-NOT: linalg.generic
342 %0 = linalg.indexed_generic {
351 linalg.yield %5 : i32
353 %1 = linalg.indexed_generic {
362 linalg.yield %5 : i32
368 // CHECK: linalg.indexed_generic
382 // CHECK: linalg.yield %[[VAL4]] : i32
383 // CHECK-NOT: linalg.indexed_generic
392 %0 = linalg.indexed_generic
399 linalg.yield %4 : f32
401 %1 = linalg.generic
408 linalg.yield %3 : f32
417 // CHECK: %[[T0:.+]] = linalg.indexed_generic
422 // CHECK: linalg.yield