Lines Matching refs:linalg

1 // RUN: mlir-opt %s -linalg-fold-unit-extent-dims -split-input-file | FileCheck %s
16   %0 = linalg.generic #trait
19          linalg.yield %arg1 : f32
31 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[$MAP0]], #[[$MAP1]]]
32 //       CHECK: linalg.generic
35 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[$MAP4]], #[[$MAP5]], #[[$MAP6]]]
53   %0 = linalg.indexed_generic #trait
63          linalg.yield %6 : i32
68 //       CHECK:   linalg.indexed_generic
76 //       CHECK:     linalg.yield %[[T6]] : i32
90   %0 = linalg.generic #trait
93          linalg.yield %arg1 : f32
99 //       CHECK:   linalg.tensor_reshape %{{.*}} []
100 //       CHECK:   linalg.generic
117   %0 = linalg.indexed_generic #trait
123          linalg.yield %3 : i32
129 //       CHECK:   linalg.indexed_generic
131 //       CHECK:     linalg.yield %[[ARG1]] : i32
147   %0 = linalg.generic #trait
150     linalg.yield %arg2 : f32
156 //       CHECK:   linalg.tensor_reshape %{{.*}} [#[[$MAP0]]]
157 //       CHECK:   linalg.generic
177   %0 = linalg.tensor_reshape %arg0 [affine_map<(d0, d1) -> (d0, d1)>] :
179   %1 = linalg.tensor_reshape %arg1 [affine_map<(d0, d1) -> (d0, d1)>] :
181   %2 = linalg.generic #trait
185          linalg.yield %3 : f32
193 //   CHECK-NOT:   linalg.tensor_reshape
194 //       CHECK:   linalg.generic
197 //   CHECK-NOT:   linalg.tensor_reshape
214    %0 = linalg.generic #trait
217         linalg.yield %arg1 : f32
225 //       CHECK:   %[[A:.*]] = linalg.tensor_reshape %[[ARG0]] []
227 //       CHECK:   linalg.generic
236 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[MAP0]]]
240   %0 = linalg.tensor_reshape %arg0
243   %1 = linalg.tensor_reshape %0
254 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[MAP0]]]
258   %0 = linalg.tensor_reshape %arg0
262   %1 = linalg.tensor_reshape %0
273 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[MAP0]], #[[MAP1]]]
277   %0 = linalg.tensor_reshape %arg0
281   %1 = linalg.tensor_reshape %0
295 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[MAP0]], #[[MAP1]], #[[MAP2]]]
299   %0 = linalg.tensor_reshape %arg0
304   %1 = linalg.tensor_reshape %0
318 //       CHECK: linalg.tensor_reshape %{{.*}} [#[[MAP0]]
322 …%0 = linalg.tensor_reshape %arg0 [affine_map<(d0, d1, d2) -> (d0, d1, d2)>] : tensor<2xf32> into t…
323   %1 = linalg.tensor_reshape %0