Home
last modified time | relevance | path

Searched refs:memref (Results 1 – 25 of 633) sorted by relevance

12345678910>>...26

/llvm-project-15.0.7/mlir/test/Dialect/MemRef/
H A Dops.mlir43 : (memref<*xf32>, memref<1xi32>) -> memref<?xf32>
45 : (memref<?xf32>, memref<2xi32>) -> memref<?x?xf32>
47 : (memref<?x?xf32>, memref<?xi32>) -> memref<*xf32>
52 memref.global @memref0 : memref<2xf32>
82 %0 = memref.alloc() : memref<2xf32>
83 %1 = memref.cast %0 : memref<2xf32> to memref<*xf32>
84 %2 = memref.alloc() : memref<2xf32>
85 %3 = memref.cast %0 : memref<2xf32> to memref<*xf32>
86 memref.copy %1, %3 : memref<*xf32> to memref<*xf32>
93 %1 = memref.cast %0 : memref<2xf32> to memref<*xf32>
[all …]
H A Dcanonicalize.mlir10 %0 = memref.cast %arg : memref<4x6x16x32xi8> to memref<?x?x16x32xi8>
165 %0 = memref.cast %arg : memref<32xi8> to memref<?xi8>
166 memref.store %0, %holder[] : memref<memref<?xi8>>
213 : (memref<*xf32>, memref<?xindex>) -> memref<*xf32>
235 : (memref<*xf32>, memref<?xi32>) -> memref<*xf32>
299 memref.store %0, %arg0[] : memref<memref<?xi32>>
309 %1 = memref.collapse_shape %0 [] : memref<1xf32> into memref<f32>
402 %0 = memref.cast %arg0 : memref<?x?xf32> to memref<8x4xf32>
541 %v = memref.cast %arg1 : memref<4xf32> to memref<?xf32>
567 memref.copy %m1, %m1 : memref<?xf32> to memref<?xf32>
[all …]
H A Dinvalid.mlir252 memref.reshape %buf(%shape) : (memref<*xf32>, memref<1xi32>) -> memref<?xi32>
260 memref.reshape %buf(%shape) : (memref<*xf32>, memref<?xi32>) -> memref<?xf32>
269 : (memref<*xf32>, memref<1xi32>) -> memref<?x?xf32>
305 memref.global @foo : memref<*xf32>
364 memref.global @gv : memref<3xi32>
381 memref.copy %arg0, %arg1 : memref<2xf32> to memref<3xf32>
389 memref.copy %arg0, %arg1 : memref<2xf32> to memref<2xf16>
769 %1 = memref.cast %0 : memref<2x5xf32, 0> to memref<*xi32>
806 %1 = memref.cast %0 : memref<2x5xf32, 0> to memref<*xf32, 1>
815 %1 = memref.cast %0 : memref<2x5xf32, 0> to memref<*xf32, 0>
[all …]
H A Dsubview.mlir59 %7 = memref.alloc(%arg1, %arg2) : memref<?x?xf32>
66 %9 = memref.alloc() : memref<16x4xf32>
97 %18 = memref.alloc() : memref<1x8xf32>
98 // CHECK: memref.subview %18[0, 0] [1, 8] [1, 1] : memref<1x8xf32> to memref<8xf32>
99 %19 = memref.subview %18[0, 0][1, 8][1, 1] : memref<1x8xf32> to memref<8xf32>
101 %20 = memref.alloc() : memref<8x16x4xf32>
103 %21 = memref.subview %20[0, 0, 0][1, 16, 4][1, 1, 1] : memref<8x16x4xf32> to memref<16x4xf32>
107 %23 = memref.alloc() : memref<f32>
108 %78 = memref.subview %23[] [] [] : memref<f32> to memref<f32>
111 %24 = memref.alloc() : memref<5x3xf32>
[all …]
H A Dmultibuffer.mlir8 // CHECK-DAG: %[[A:.*]] = memref.alloc() : memref<5x4x128xf32>
11 %0 = memref.alloc() : memref<4x128xf32>
21 // CHECK: memref.copy %{{.*}}, %[[SV]] : memref<4x128xf32, #{{.*}}> to memref<4x128xf32, #[[$MAP0]]>
22 …memref.copy %1, %0 : memref<4x128xf32, affine_map<(d0, d1)[s0] -> (d0 * 1024 + s0 + d1)>> to memre…
35 // CHECK-DAG: %[[A:.*]] = memref.alloc() : memref<5x4x128xf32>
38 %0 = memref.alloc() : memref<4x128xf32>
48 // CHECK: memref.copy %{{.*}}, %[[SV]] : memref<4x128xf32, #{{.*}}> to memref<4x128xf32, #[[$MAP0]]>
68 %0 = memref.alloc() : memref<4x128xf32>
78 // CHECK: memref.copy %{{.*}}, %[[SV]] : memref<4x128xf32, #{{.*}}> to memref<4x128xf32, #[[$MAP0]]>
96 // CHECK: %{{.*}} = memref.alloc() : memref<4x128xf32>
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/
H A Dconversion_sparse2dense.mlir21 // CHECK-DAG: %[[AttrsD:.*]] = memref.cast %[[AttrsS]] : memref<1xi8> to memref<?xi8>
25 // CHECK-DAG: %[[SizesD:.*]] = memref.cast %[[SizesS]] : memref<1xindex> to memref<?xindex>
28 // CHECK-DAG: %[[PermD:.*]] = memref.cast %[[PermS]] : memref<1xindex> to memref<?xindex>
34 // CHECK-DAG: %[[IndD:.*]] = memref.cast %[[IndS]] : memref<1xindex> to memref<?xindex>
58 // CHECK-DAG: %[[AttrsD:.*]] = memref.cast %[[AttrsS]] : memref<1xi8> to memref<?xi8>
66 // CHECK-DAG: %[[PermD:.*]] = memref.cast %[[PermS]] : memref<1xindex> to memref<?xindex>
72 // CHECK-DAG: %[[IndD:.*]] = memref.cast %[[IndS]] : memref<1xindex> to memref<?xindex>
99 // CHECK-DAG: %[[AttrsD:.*]] = memref.cast %[[AttrsS]] : memref<2xi8> to memref<?xi8>
114 // CHECK-DAG: %[[IndD:.*]] = memref.cast %[[IndS]] : memref<2xindex> to memref<?xindex>
142 // CHECK-DAG: %[[AttrsD:.*]] = memref.cast %[[AttrsS]] : memref<2xi8> to memref<?xi8>
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/Linalg/
H A Dpromote.mlir15 %3 = memref.view %A[%c0][%M, %K] : memref<?xi8> to memref<?x?xf32>
16 %4 = memref.view %A[%c0][%K, %N] : memref<?xi8> to memref<?x?xf32>
17 %5 = memref.view %A[%c0][%M, %N] : memref<?xi8> to memref<?x?xf32>
18 %6 = memref.dim %3, %c0 : memref<?x?xf32>
19 %7 = memref.dim %3, %c1 : memref<?x?xf32>
20 %8 = memref.dim %4, %c1 : memref<?x?xf32>
88 %3 = memref.view %A[%c0][%M, %K] : memref<?xi8> to memref<?x?xf64>
89 %4 = memref.view %A[%c0][%K, %N] : memref<?xi8> to memref<?x?xf64>
90 %5 = memref.view %A[%c0][%M, %N] : memref<?xi8> to memref<?x?xf64>
91 %6 = memref.dim %3, %c0 : memref<?x?xf64>
[all …]
H A Dforward-vector-transfers.mlir13 %alloc = memref.alloc() : memref<32 x f32>
15 memref.copy %in, %subview : memref<? x f32> to memref<16 x f32>
17 memref.dealloc %alloc : memref<32 x f32>
34 memref.copy %in, %subview : memref<? x f32> to memref<16 x f32>
51 %view = memref.view %alloc[%c0][] : memref<128 x i8> to memref<32 x f32>
53 memref.copy %in, %subview : memref<? x f32> to memref<16 x f32>
73 memref.copy %in, %subview : memref<? x f32> to memref<16 x f32>
92 memref.copy %subview, %out : memref<16 x f32> to memref<? x f32>
111 memref.copy %subview, %out : memref<16 x f32> to memref<? x f32>
134 memref.copy %in, %subview : memref<? x f32> to memref<16 x f32>
[all …]
H A Dtransform-promotion.mlir47 // CHECK: %[[v0:.*]] = memref.view %[[a0]]{{.*}} : memref<32000000xi8> to memref<?x?x…
51 // CHECK: %[[v1:.*]] = memref.view %[[a1]]{{.*}} : memref<48000000xi8> to memref<?x?x…
55 // CHECK: %[[v2:.*]] = memref.view %[[a2]]{{.*}} : memref<24000000xi8> to memref<?x?x…
58 // CHECK: memref.copy %[[s0]], %[[l0]] : memref<?x?xf32, #map{{.*}}> to memref<?x?xf3…
59 // CHECK: memref.copy %[[s1]], %[[l1]] : memref<?x?xf32, #map{{.*}}> to memref<?x?xf3…
60 // CHECK: memref.copy %[[s2]], %[[l2]] : memref<?x?xf32, #map{{.*}}> to memref<?x?xf3…
117 // CHECK: %[[v0:.*]] = memref.view %[[a0]]{{.*}} : memref<32000000xi8> to memref<?x?xf32>
122 // CHECK: memref.copy %[[s0]], %[[l0]] : memref<?x?xf32, #map{{.*}}> to memref<?x?xf32, #ma…
155 // CHECK: %[[v0:.*]] = memref.view %[[a0]]{{.*}} : memref<32000000xi8> to memref<?x?xf32>
158 // CHECK: memref.copy %[[s0]], %[[l0]] : memref<?x?xf32, #map{{.*}}> to memref<?x?xf32, #ma…
[all …]
H A Dfusion.mlir479 // CHECK: (%[[A:.*]]: memref{{.*}}, %[[B:.*]]: memref{{.*}}, %[[C:.*]]: memref{{.*}}, %[[D:.*]]: m…
557 %A = memref.alloc (%M, %N): memref<?x?xf32>
558 %B = memref.alloc (%M, %N): memref<?x?xf32>
559 %C = memref.alloc (%M, %N): memref<?x?xf32>
560 %D = memref.alloc (%M, %N): memref<?x?xf32>
561 %E = memref.alloc (%M, %N): memref<?x?xf32>
569 %0 = memref.dim %B, %c0 : memref<?x?xf32>
570 %1 = memref.dim %B, %c1 : memref<?x?xf32>
634 memref.dealloc %0 : memref<100x10xf32>
635 %2 = memref.dim %1, %c0 : memref<100x10xf32>
[all …]
/llvm-project-15.0.7/mlir/test/Transforms/
H A Dnormalize-memrefs-ops.mlir19 %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
21 memref.dealloc %0 : memref<1x16x14x14xf32, #map0>
34 %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
36 memref.dealloc %0 : memref<1x16x14x14xf32, #map0>
49 %0 = memref.alloc() : memref<1x16x14x14xf32, #map0>
51 memref.dealloc %0 : memref<1x16x14x14xf32, #map0>
68 %1 = memref.alloc() : memref<1x16x14x14xf32>
86 memref.dealloc %0 : memref<1x16x14x14xf32, #map_tile>
88 memref.dealloc %1 : memref<1x16x14x14xf32>
106 memref.dealloc %0 : memref<1x16x14x14xf32, #map_tile>
[all …]
H A Dpromote-buffers-to-stack.mlir23 %0 = memref.alloc() : memref<2xf32>
58 %1 = memref.alloc(%0) : memref<?xf32>
80 %0 = memref.rank %memref : memref<*xf32>
92 %0 = memref.rank %memref : memref<*xf32>
119 %0 = memref.alloc() : memref<4xf32>
140 %0 = memref.alloc() : memref<2xf32>
166 %0 = memref.alloc() : memref<2xf32>
193 %0 = memref.alloc() : memref<2xf32>
203 %7 = memref.alloc() : memref<2xf32>
228 %0 = memref.alloc() : memref<2xf32>
[all …]
H A Dpipeline-data-transfer.mlir14 %tag = memref.alloc() : memref<1 x f32>
29 memref.dealloc %tag : memref<1 x f32>
79 : memref<512xf32>, memref<4xf32, 1>, memref<1xi32>
82 memref.dealloc %2 : memref<1xi32>
117 %3 = memref.alloc() : memref<2xi32>
184 memref.dealloc %5 : memref<2xi32>
185 memref.dealloc %4 : memref<2xi32>
186 memref.dealloc %3 : memref<2xi32>
227 memref.dealloc %5 : memref<2xi32>
228 memref.dealloc %4 : memref<2xi32>
[all …]
H A Dbuffer-results-to-out-params.mlir6 // CHECK: memref.copy %[[RESULT]], %[[ARG]] : memref<f32> to memref<f32>
18 // CHECK: memref.copy %[[RESULT]], %[[ARG1]] : memref<2xf32> to memref<2xf32>
30 // CHECK: memref.copy %[[RESULTS]]#0, %[[ARG0]] : memref<1xf32> to memref<1xf32>
31 // CHECK: memref.copy %[[RESULTS]]#1, %[[ARG1]] : memref<2xf32> to memref<2xf32>
34 func.func @multiple_results() -> (memref<1xf32>, memref<2xf32>) {
36 return %0, %1 : memref<1xf32>, memref<2xf32>
42 // CHECK: memref.copy %[[RESULT1]]#1, %[[OUTPARAM]] : memref<f32> to memref<f32>
54 // CHECK: func private @mixed_result_attrs(memref<1xf32>, memref<2xf32> {test.some_attr}, memref<3x…
55 func.func private @mixed_result_attrs() -> (memref<1xf32>, memref<2xf32> {test.some_attr}, memref<3…
77 func.func private @callee() -> (memref<1xf32>, memref<2xf32>)
[all …]
H A Dnormalize-memrefs.mlir32 // CHECK-NEXT: memref.alloc() : memref<65xf32>
73 // CHECK: memref.alloc() : memref<8x32x8x16xf32>
117 // CHECK: memref.alloc() : memref<100xf32>
174 %a = memref.alloc(): memref<8xf64, #tile>
175 %b = memref.alloc(): memref<16xf64, #tile>
177 %e = memref.alloc(): memref<24xf64>
228 // CHECK-SAME: (%[[C:arg[0-9]+]]: memref<2x4xf64>) -> (memref<4x4xf64>, memref<2x4xf64>)
230 %a = memref.alloc() : memref<8xf64, #tile>
231 %b = memref.alloc() : memref<16xf64, #tile>
307 %a = memref.alloc() : memref<16xf64, #tile>
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/Transforms/
H A Dbuffer-hoisting.mlir21 %0 = memref.alloc() : memref<2xf32>
54 %1 = memref.alloc(%0) : memref<?xf32>
96 %1 = memref.alloc(%0) : memref<?xf32>
133 %0 = memref.alloc() : memref<2xf32>
156 %0 = memref.alloc() : memref<2xf32>
166 %7 = memref.alloc() : memref<2xf32>
194 %0 = memref.alloc() : memref<2xf32>
226 %0 = memref.alloc() : memref<2xf32>
240 %9 = memref.alloc() : memref<2xf32>
262 %0 = memref.alloc() : memref<2xf32>
[all …]
H A Dbuffer-loop-hoisting.mlir20 %0 = memref.alloc() : memref<2xf32>
53 %1 = memref.alloc(%0) : memref<?xf32>
84 %0 = memref.alloc() : memref<2xf32>
87 %1 = memref.alloc() : memref<2xf32>
141 %0 = memref.alloc() : memref<2xf32>
145 %3 = memref.alloc() : memref<2xf32>
168 %0 = memref.alloc() : memref<2xf32>
201 %0 = memref.alloc() : memref<2xf32>
290 %0 = memref.alloc() : memref<2xf32>
334 %0 = memref.alloc() : memref<2xf32>
[all …]
H A Dbuffer-deallocation.mlir24 %0 = memref.alloc() : memref<2xf32>
69 %1 = memref.alloc(%0) : memref<?xf32>
105 %1 = memref.alloc(%0) : memref<?xf32>
106 %2 = memref.cast %1 : memref<?xf32> to memref<*xf32>
207 %0 = memref.alloc() : memref<4xf32>
230 %0 = memref.alloc() : memref<2xf32>
261 %0 = memref.alloc() : memref<2xf32>
488 memref.dealloc %1 : memref<2xf32>
528 memref.dealloc %0 : memref<2xf32>
595 %arg2: memref<5xf32>) -> (memref<10xf32>, memref<15xf32>) {
[all …]
/llvm-project-15.0.7/mlir/test/mlir-cpu-runner/
H A Dcopy.mlir14 %input = memref.alloc() : memref<2x3xf32>
24 %unranked_input = memref.cast %input : memref<2x3xf32> to memref<*xf32>
30 %copy = memref.alloc() : memref<2x3xf32>
31 memref.copy %input, %copy : memref<2x3xf32> to memref<2x3xf32>
32 %unranked_copy = memref.cast %copy : memref<2x3xf32> to memref<*xf32>
42 %unranked_copy_two = memref.cast %copy_two : memref<3x2xf32> to memref<*xf32>
60 %scalar = memref.alloc() : memref<f32>
63 memref.copy %scalar, %scalar_copy : memref<f32> to memref<f32>
73 memref.dealloc %copy : memref<2x3xf32>
74 memref.dealloc %input : memref<2x3xf32>
[all …]
H A Dmemref-reshape.mlir14 %input = memref.alloc() : memref<2x3xf32>
24 %unranked_input = memref.cast %input : memref<2x3xf32> to memref<*xf32>
46 memref.dealloc %input : memref<2x3xf32>
47 memref.dealloc %shape : memref<2xindex>
54 : (memref<2x3xf32>, memref<2xindex>) -> memref<?x?xf32>
56 %unranked_output = memref.cast %output : memref<?x?xf32> to memref<*xf32>
67 %unranked_input = memref.cast %input : memref<2x3xf32> to memref<*xf32>
69 : (memref<2x3xf32>, memref<2xindex>) -> memref<?x?xf32>
84 : (memref<2x3xf32>, memref<?xindex>) -> memref<*xf32>
96 %unranked_input = memref.cast %input : memref<2x3xf32> to memref<*xf32>
[all …]
H A Dunranked-memref.mlir44 %V = memref.cast %A : memref<10x3xf32, 0> to memref<?x?xf32>
46 %U = memref.cast %A : memref<10x3xf32, 0> to memref<*xf32>
49 %V2 = memref.cast %U : memref<*xf32> to memref<?x?xf32>
51 %U2 = memref.cast %V2 : memref<?x?xf32, 0> to memref<*xf32>
54 %V3 = memref.cast %V2 : memref<?x?xf32> to memref<*xf32>
55 %V4 = memref.cast %V3 : memref<*xf32> to memref<?x?xf32>
57 %U3 = memref.cast %V2 : memref<?x?xf32> to memref<*xf32>
64 %U4 = memref.cast %I8 : memref<i8> to memref<*xi8>
90 %0 = memref.cast %arg0 : memref<4x3xf32> to memref<*xf32>
104 %0 = memref.cast %arg0: memref<4x3xf32> to memref<*xf32>
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/Bufferization/
H A Dcanonicalize.mlir118 memref.dealloc %arg0 : memref<?xf32>
130 memref.dealloc %0 : memref<?xf32>
141 %0 = memref.cast %arg0 : memref<?xf32> to memref<32xf32>
154 memref.dealloc %arg0 : memref<?xf32>
166 %0 = memref.cast %arg0 : memref<?xf32> to memref<32xf32>
168 memref.dealloc %arg0 : memref<?xf32>
170 memref.dealloc %1 : memref<32xf32>
184 memref.dealloc %arg0 : memref<?xf32>
186 memref.dealloc %arg0 : memref<?xf32>
208 memref.dealloc %0 : memref<?xf32>
[all …]
/llvm-project-15.0.7/mlir/test/IR/
H A Dmemory-ops.mlir9 // CHECK: %0 = memref.alloc() : memref<1024x64xf32, 1>
16 // CHECK: %1 = memref.alloc(%c0, %c1) : memref<?x?xf32, 1>
29 // CHECK: %4 = memref.alloc() : memref<2xi32>
30 %4 = memref.alloc() : memref<2 x i32>
40 // CHECK: %0 = memref.alloca() : memref<1024x64xf32, 1>
60 %4 = memref.alloca() {alignment = 64} : memref<2 x i32>
68 // CHECK: %0 = memref.alloc() : memref<1024x64xf32>
71 // CHECK: memref.dealloc %0 : memref<1024x64xf32>
79 // CHECK: %0 = memref.alloc() : memref<1024x64xf32, 1>
102 %tag = memref.alloc() : memref<1 x f32>
[all …]
/llvm-project-15.0.7/mlir/test/Integration/GPU/ROCM/
H A Dvecadd.mlir12 func.func @vecadd(%arg0 : memref<?xf32>, %arg1 : memref<?xf32>, %arg2 : memref<?xf32>) {
18 %a = memref.load %arg0[%tx] : memref<?xf32>
33 %0 = memref.alloc() : memref<5xf32>
34 %1 = memref.alloc() : memref<5xf32>
35 %2 = memref.alloc() : memref<5xf32>
36 %3 = memref.cast %0 : memref<5xf32> to memref<?xf32>
37 %4 = memref.cast %1 : memref<5xf32> to memref<?xf32>
38 %5 = memref.cast %2 : memref<5xf32> to memref<?xf32>
43 %6 = memref.cast %3 : memref<?xf32> to memref<*xf32>
44 %7 = memref.cast %4 : memref<?xf32> to memref<*xf32>
[all …]
/llvm-project-15.0.7/mlir/test/Dialect/GPU/
H A Dcanonicalize.mlir23 return %memref, %memref_0 : memref<5xf16>, memref<5xf16>
34 %1 = memref.alloc() : memref<2xf16>
40 %4 = gpu.memcpy async [%3] %1, %memref : memref<2xf16>, memref<2xf16>
89 %1 = memref.view %arg0[%0][] : memref<1xi8> to memref<i1>
99 %0 = memref.cast %arg0 : memref<10xf32> to memref<?xf32>
100 %1 = memref.cast %arg1 : memref<10xf32> to memref<?xf32>
109 %0 = memref.cast %arg0 : memref<10xf32> to memref<?xf32>
160 %12 = memref.load %memref[%11, %c0, %8] : memref<2x16x16xf32>
162 %14 = memref.load %memref[%13, %c0, %8] : memref<2x16x16xf32>
163 memref.store %12, %memref[%11, %c0, %8] : memref<2x16x16xf32>
[all …]

12345678910>>...26