Home
last modified time | relevance | path

Searched refs:sparse_tensor (Results 1 – 25 of 128) sorted by relevance

123456

/llvm-project-15.0.7/mlir/test/Dialect/SparseTensor/
H A Droundtrip.mlir143 // CHECK: sparse_tensor.expand %[[A]]
157 // CHECK: sparse_tensor.compress %[[A]]
174 // CHECK: sparse_tensor.out %[[A]], %[[B]] : tensor<?x?xf64, #sparse_tensor.encoding<{{.*}}>>…
205 sparse_tensor.yield %x : f64
210 sparse_tensor.yield %cf0 : f64
233 %r = sparse_tensor.unary %arg0 : f64 to f64
236 sparse_tensor.yield %x : f64
240 sparse_tensor.yield %cf1 : f64
262 %r = sparse_tensor.unary %arg0 : f64 to i64
266 sparse_tensor.yield %ret : i64
[all …]
H A Dinvalid.mlir185 sparse_tensor.yield %x : f64
201 sparse_tensor.yield %y : f64
213 sparse_tensor.yield %x : f64
229 sparse_tensor.yield %1 : f32
264 %r = sparse_tensor.unary %arg0 : f64 to f64
267 sparse_tensor.yield %x : index
281 sparse_tensor.yield %x : f64
294 sparse_tensor.yield %1 : f32
320 sparse_tensor.yield %x : f64
333 sparse_tensor.yield %cst : i64
[all …]
H A Drewriting.mlir3 #SparseVector = #sparse_tensor.encoding<{
7 #SparseMatrix = #sparse_tensor.encoding<{
23 // CHECK: %[[C:.*]] = sparse_tensor.convert %[[A]] : tensor<12xf64, #sparse_tensor.encoding…
35 // CHECK: %[[C:.*]] = sparse_tensor.convert %[[E]] : tensor<3x4xf64> to tensor<3x4xf64, #sp…
36 // CHECK: return %[[C]] : tensor<3x4xf64, #sparse_tensor.encoding<{{{.*}}}>>
47 …CHECK-SAME: %[[A:.*]]: tensor<12xf64, #sparse_tensor.encoding<{{{.*}}}>>) -> tensor<3x4xf64, #s…
48 …and_shape %[[A]] {{.*}} : tensor<12xf64, #sparse_tensor.encoding<{{{.*}}}>> into tensor<3x4xf64, #…
68 // CHECK: %[[C:.*]] = sparse_tensor.convert %[[A]] : tensor<3x4xf64, #sparse_tensor.encodin…
80 // CHECK: %[[C:.*]] = sparse_tensor.convert %[[R]] : tensor<12xf64> to tensor<12xf64, #spar…
92 …CHECK-SAME: %[[A:.*]]: tensor<3x4xf64, #sparse_tensor.encoding<{{{.*}}}>>) -> tensor<12xf64, #s…
[all …]
H A Dfold.mlir3 #SparseVector = #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>
6 // CHECK-SAME: %[[A:.*]]: tensor<64xf32, #sparse_tensor.encoding<{{{.*}}}>>)
7 // CHECK-NOT: sparse_tensor.convert
8 // CHECK: return %[[A]] : tensor<64xf32, #sparse_tensor.encoding<{{{.*}}}>>
16 // CHECK-NOT: sparse_tensor.convert
19 %0 = sparse_tensor.convert %arg0 : tensor<64xf32> to tensor<64xf32, #SparseVector>
24 // CHECK-SAME: %[[A:.*]]: tensor<64xf32, #sparse_tensor.encoding<{{{.*}}}>>)
25 // CHECK-NOT: sparse_tensor.pointers
26 // CHECK-NOT: sparse_tensor.indices
27 // CHECK-NOT: sparse_tensor.values
[all …]
H A Dsparse_kernels.mlir17 // CHECK-DAG: %[[VAL_6:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_3]] : tensor<10x20xf3…
18 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<10x20xf32…
19 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_4]] : tensor<10x20xf3…
20 // CHECK-DAG: %[[VAL_9:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_4]] : tensor<10x20xf32…
21 // CHECK-DAG: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<10x20xf32, #sparse_te…
141 // CHECK: %[[VAL_77:.*]] = sparse_tensor.load %[[VAL_8]] hasInserts : tensor<4x4xf64, #sp…
163 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.indices %[[VAL_1]], %[[VAL_3]] : tensor<3x3xi32, …
166 // CHECK-DAG: %[[VAL_11:.*]] = sparse_tensor.values %[[VAL_1]] : tensor<3x3xi32, #sparse_tens…
212 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.pointers %[[VAL_1]], %[[VAL_4]] : tensor<3x6xi8, …
213 // CHECK-DAG: %[[VAL_9:.*]] = sparse_tensor.indices %[[VAL_1]], %[[VAL_4]] : tensor<3x6xi8, #…
[all …]
H A Dsparse_index.mlir4 #DenseMatrix = #sparse_tensor.encoding<{
8 #SparseMatrix = #sparse_tensor.encoding<{
28 // CHECK-DAG: %[[VAL_6:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<?x?xi64, #sparse_tenso…
31 // CHECK-DAG: %[[VAL_9:.*]] = sparse_tensor.values %[[VAL_5]] : tensor<?x?xi64, #sparse_tenso…
46 // CHECK: %[[VAL_21:.*]] = sparse_tensor.load %[[VAL_5]] : tensor<?x?xi64, #sparse_tensor…
79 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_1]] : tensor<?x?xi64,…
80 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_1]] : tensor<?x?xi64, …
81 // CHECK-DAG: %[[VAL_9:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_2]] : tensor<?x?xi64,…
82 // CHECK-DAG: %[[VAL_10:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_2]] : tensor<?x?xi64,…
83 // CHECK-DAG: %[[VAL_11:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<?x?xi64, #sparse_tens…
[all …]
H A Dsparse_transpose.mlir3 #DCSR = #sparse_tensor.encoding<{
19 … #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ], pointerBitWidth = 0, ind…
24 …sparse_tensor.convert %[[VAL_0]] : tensor<3x4xf64, #sparse_tensor.encoding<{ dimLevelType = [ "com…
25 // CHECK: %[[VAL_6:.*]] = sparse_tensor.pointers %[[VAL_5]], %[[VAL_1]] : tensor<3x4xf64,…
26 // CHECK: %[[VAL_7:.*]] = sparse_tensor.indices %[[VAL_5]], %[[VAL_1]] : tensor<3x4xf64, …
27 // CHECK: %[[VAL_8:.*]] = sparse_tensor.pointers %[[VAL_5]], %[[VAL_2]] : tensor<3x4xf64,…
28 // CHECK: %[[VAL_9:.*]] = sparse_tensor.indices %[[VAL_5]], %[[VAL_2]] : tensor<3x4xf64, …
29 // CHECK: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_5]] : tensor<3x4xf64, #sparse_tens…
45 // CHECK: sparse_tensor.lex_insert %[[VAL_4]], %[[VAL_11]], %[[VAL_12]] : tensor<4x3x…
48 // CHECK: %[[VAL_23:.*]] = sparse_tensor.load %[[VAL_4]] hasInserts : tensor<4x3xf64, #sp…
[all …]
H A Dsparse_out.mlir4 #CSR = #sparse_tensor.encoding<{
9 #DCSR = #sparse_tensor.encoding<{
14 #SparseTensor = #sparse_tensor.encoding<{
33 // CHECK: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
46 // CHECK: %[[VAL_18:.*]] = sparse_tensor.load %[[VAL_0]] : tensor<32x16xf32, #sparse_tens…
64 // CHECK: %[[VAL_3:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_1]] : tensor<32x16xf3…
65 // CHECK: %[[VAL_4:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_2]] : tensor<32x16xf3…
66 // CHECK: %[[VAL_5:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
80 // CHECK: %[[VAL_16:.*]] = sparse_tensor.load %[[VAL_0]] : tensor<32x16xf32, #sparse_tens…
112 // CHECK: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<10x20xf32, #sparse_te…
[all …]
H A Dsparse_fp_ops.mlir41 // CHECK: %[[VAL_6:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
73 // CHECK: %[[VAL_6:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
105 // CHECK: %[[VAL_6:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
170 // CHECK: %[[VAL_7:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_4]] : tensor<32xf64, #s…
171 // CHECK: %[[VAL_8:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_4]] : tensor<32xf64, #sp…
172 // CHECK: %[[VAL_9:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
291 // CHECK: %[[VAL_6:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xf64, #sp…
292 // CHECK: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
327 // CHECK: %[[VAL_6:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xf64, #sp…
328 // CHECK: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf64, #sparse_tensor.e…
[all …]
H A Dsparse_int_ops.mlir33 // CHECK: %[[VAL_7:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_4]] : tensor<32xi64, …
34 // CHECK: %[[VAL_8:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_4]] : tensor<32xi64, #…
35 // CHECK: %[[VAL_9:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xi64, #sparse_tensor…
94 // CHECK: %[[VAL_8:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_4]] : tensor<32xi64, …
95 // CHECK: %[[VAL_9:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_4]] : tensor<32xi64, #…
96 // CHECK: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xi64, #sparse_tenso…
155 // CHECK: %[[VAL_6:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xi64, #…
156 // CHECK: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xi64, #sparse_tensor…
191 // CHECK: %[[VAL_6:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xi64, #…
192 // CHECK: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xi64, #sparse_tensor…
[all …]
H A Dinvalid_encoding.mlir3 #a = #sparse_tensor.encoding<{dimLevelType = []}>
8 #a = #sparse_tensor.encoding<{dimLevelType = ["dense", "compressed"]}>
18 #a = #sparse_tensor.encoding<{dimLevelType = [1]}> // expected-error {{expected a string value in d…
23 #a = #sparse_tensor.encoding<{dimLevelType = ["strange"]}> // expected-error {{unexpected dimension…
28 #a = #sparse_tensor.encoding<{dimOrdering = "wrong"}> // expected-error {{expected an affine map fo…
33 #a = #sparse_tensor.encoding<{dimOrdering = affine_map<(i,j) -> (i,i)>}> // expected-error {{expect…
38 #a = #sparse_tensor.encoding<{pointerBitWidth = "x"}> // expected-error {{expected an integral poin…
43 #a = #sparse_tensor.encoding<{pointerBitWidth = 42}> // expected-error {{unexpected pointer bitwidt…
48 #a = #sparse_tensor.encoding<{indexBitWidth = "not really"}> // expected-error {{expected an integr…
53 #a = #sparse_tensor.encoding<{indexBitWidth = 128}> // expected-error {{unexpected index bitwidth: …
[all …]
H A Dsparse_2d.mlir4 #Tdd = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "dense" ] }>
5 #Tds = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>
6 #Tsd = #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "dense" ] }>
7 #Tss = #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ] }>
27 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
63 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
102 // CHECK-DAG: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_te…
163 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
204 // CHECK-DAG: %[[VAL_10:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_te…
270 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32x16xf32, #sparse_ten…
[all …]
H A Done_shot_bufferize_tensor_copy_insertion.mlir4 #DCSR = #sparse_tensor.encoding<{
15 %1 = sparse_tensor.load %0 : tensor<20x40xf32, #DCSR>
23 // CHECK: sparse_tensor.new {{.*}} {bufferization.escape = [false]}
24 // CHECK-FUNC: sparse_tensor.new {{.*}} {bufferization.escape = [true]}
25 %0 = sparse_tensor.new %file : !Filename to tensor<20x40xf32, #DCSR>
35 // CHECK: %[[loaded:.*]] = sparse_tensor.load %[[alloc]]
36 // CHECK-FUNC: %[[loaded:.*]] = sparse_tensor.load %[[alloc]]
37 %1 = sparse_tensor.load %0 : tensor<20x40xf32, #DCSR>
38 // CHECK: sparse_tensor.convert %[[loaded]] {bufferization.escape = [false]}
40 %2 = sparse_tensor.convert %1 : tensor<20x40xf32, #DCSR> to tensor<20x40xf32>
[all …]
H A Dsparse_affine.mlir4 #SpVec = #sparse_tensor.encoding<{ dimLevelType = [ "compressed" ] }>
5 #CSR = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ] }>
24 // CHECK-DAG: %[[VAL_6:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_3]] : tensor<32xf32, …
25 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xf32, #…
26 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xf32, #sparse_tensor…
75 // CHECK-DAG: %[[VAL_6:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_3]] : tensor<32xi32, …
76 // CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32xi32, #…
77 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<32xi32, #sparse_tensor…
126 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_3]] : tensor<32x16xf6…
127 // CHECK-DAG: %[[VAL_9:.*]] = sparse_tensor.indices %[[VAL_0]], %[[VAL_3]] : tensor<32x16xf64…
[all …]
H A Dsparse_3d.mlir4 #Td = #sparse_tensor.encoding<{ dimLevelType = [ "dense" ] }>
6 #Tddd = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "dense", "dense" ] }>
7 #Tdds = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "dense", "compressed" ] }>
8 #Tdsd = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed", "dense" ] }>
9 #Tdss = #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed", "compressed" ] }>
1200 …/ CHECK-DAG: %[[VAL_5:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_2]] : tensor<10x20x30…
1201 …/ CHECK-DAG: %[[VAL_6:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_3]] : tensor<10x20x30…
1202 …/ CHECK-DAG: %[[VAL_7:.*]] = sparse_tensor.pointers %[[VAL_0]], %[[VAL_4]] : tensor<10x20x30…
1203 // CHECK-DAG: %[[VAL_8:.*]] = sparse_tensor.values %[[VAL_0]] : tensor<10x20x30xf32, #sparse_…
1252 // CHECK-SAME: %[[VAL_1:.*]]: tensor<?xf32, #sparse_tensor.encoding<{{{.*}}}>>
[all …]
H A Dsparse_reshape.mlir4 #SparseVector = #sparse_tensor.encoding<{ dimLevelType = [ "compressed" ] }>
5 #SparseMatrix = #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ] }>
11 …ROUND-SAME: %[[A:.*]]: tensor<100xf64, #sparse_tensor.encoding<{{{.*}}}>>) -> tensor<10x10xf64, #…
12 …%[[A]] {{\[\[}}0, 1]] : tensor<100xf64, #sparse_tensor.encoding<{{{.*}}}>> into tensor<10x10xf64, …
13 // CHECK-ROUND: return %[[E]] : tensor<10x10xf64, #sparse_tensor.encoding<{{{.*}}}>>
50 …OUND-SAME: %[[A:.*]]: tensor<10x10xf64, #sparse_tensor.encoding<{{{.*}}}>>) -> tensor<100xf64, #s…
51 …[[A]] {{\[\[}}0, 1]] : tensor<10x10xf64, #sparse_tensor.encoding<{{{.*}}}>> into tensor<100xf64, #…
52 // CHECK-ROUND: return %[[C]] : tensor<100xf64, #sparse_tensor.encoding<{{{.*}}}>>
/llvm-project-15.0.7/mlir/test/Integration/Dialect/SparseTensor/CPU/
H A Dsparse_conversion.mlir7 #Tensor1 = #sparse_tensor.encoding<{
12 #Tensor2 = #sparse_tensor.encoding<{
17 #Tensor3 = #sparse_tensor.encoding<{
105 %v1 = sparse_tensor.values %1 : tensor<2x3x4xf64, #Tensor1> to memref<?xf64>
106 %v2 = sparse_tensor.values %2 : tensor<2x3x4xf64, #Tensor2> to memref<?xf64>
107 %v3 = sparse_tensor.values %3 : tensor<2x3x4xf64, #Tensor3> to memref<?xf64>
108 %av = sparse_tensor.values %a : tensor<2x3x4xf64, #Tensor1> to memref<?xf64>
109 %bv = sparse_tensor.values %b : tensor<2x3x4xf64, #Tensor1> to memref<?xf64>
110 %cv = sparse_tensor.values %c : tensor<2x3x4xf64, #Tensor1> to memref<?xf64>
111 %dv = sparse_tensor.values %d : tensor<2x3x4xf64, #Tensor2> to memref<?xf64>
[all …]
H A Dsparse_conversion_sparse2dense.mlir8 #Tensor1 = #sparse_tensor.encoding<{
13 #Tensor2 = #sparse_tensor.encoding<{
18 #Tensor3 = #sparse_tensor.encoding<{
23 #Tensor4 = #sparse_tensor.encoding<{
28 #Tensor5 = #sparse_tensor.encoding<{
33 #Tensor6 = #sparse_tensor.encoding<{
111 %s2341 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor1>
112 %s2342 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor2>
113 %s2343 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor3>
114 %s2344 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor4>
[all …]
H A Dsparse_storage.mlir11 #Dense = #sparse_tensor.encoding<{
15 #CSR = #sparse_tensor.encoding<{
19 #DCSR = #sparse_tensor.encoding<{
23 #CSC = #sparse_tensor.encoding<{
28 #DCSC = #sparse_tensor.encoding<{
33 #BlockRow = #sparse_tensor.encoding<{
37 #BlockCol = #sparse_tensor.encoding<{
77 %1 = sparse_tensor.convert %t : tensor<10x8xf64> to tensor<10x8xf64, #CSR>
93 %5 = sparse_tensor.values %0 : tensor<10x8xf64, #Dense> to memref<?xf64>
114 %11 = sparse_tensor.values %1 : tensor<10x8xf64, #CSR> to memref<?xf64>
[all …]
H A Dsparse_conversion_ptr.mlir7 #DCSR = #sparse_tensor.encoding<{
13 #DCSC = #sparse_tensor.encoding<{
20 #CSC = #sparse_tensor.encoding<{
75 %1 = sparse_tensor.convert %t1 : tensor<32x64xf64> to tensor<32x64xf64, #DCSR>
77 %3 = sparse_tensor.convert %t1 : tensor<32x64xf64> to tensor<32x64xf64, #CSC>
94 %m1 = sparse_tensor.values %1 : tensor<32x64xf64, #DCSR> to memref<?xf64>
95 %m2 = sparse_tensor.values %2 : tensor<32x64xf64, #DCSC> to memref<?xf64>
96 %m3 = sparse_tensor.values %3 : tensor<32x64xf64, #CSC> to memref<?xf64>
97 %m4 = sparse_tensor.values %4 : tensor<32x64xf64, #DCSC> to memref<?xf64>
98 %m5 = sparse_tensor.values %5 : tensor<32x64xf64, #DCSR> to memref<?xf64>
[all …]
H A Dsparse_conversion_dyn.mlir7 #DCSR = #sparse_tensor.encoding<{
11 #DCSC = #sparse_tensor.encoding<{
42 %1 = sparse_tensor.convert %t1 : tensor<32x64xf64> to tensor<?x?xf64, #DCSR>
44 %3 = sparse_tensor.convert %t2 : tensor<?x?xf64> to tensor<?x?xf64, #DCSR>
45 %4 = sparse_tensor.convert %t2 : tensor<?x?xf64> to tensor<?x?xf64, #DCSC>
61 %m1 = sparse_tensor.values %1 : tensor<?x?xf64, #DCSR> to memref<?xf64>
62 %m2 = sparse_tensor.values %2 : tensor<?x?xf64, #DCSC> to memref<?xf64>
63 %m3 = sparse_tensor.values %3 : tensor<?x?xf64, #DCSR> to memref<?xf64>
64 %m4 = sparse_tensor.values %4 : tensor<?x?xf64, #DCSC> to memref<?xf64>
65 %m5 = sparse_tensor.values %5 : tensor<?x?xf64, #DCSC> to memref<?xf64>
[all …]
H A Dsparse_conversion_sparse2sparse.mlir7 #Tensor1 = #sparse_tensor.encoding<{
13 #Tensor2 = #sparse_tensor.encoding<{
17 #Tensor3 = #sparse_tensor.encoding<{
57 %s1 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor1>
58 %s2 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor2>
59 %s3 = sparse_tensor.convert %src : tensor<2x3x4xf64> to tensor<2x3x4xf64, #Tensor3>
64 %t13 = sparse_tensor.convert %s1 : tensor<2x3x4xf64, #Tensor1> to tensor<2x3x4xf64, #Tensor3>
72 %d13 = sparse_tensor.convert %t13 : tensor<2x3x4xf64, #Tensor3> to tensor<2x3x4xf64>
73 %d21 = sparse_tensor.convert %t21 : tensor<2x3x4xf64, #Tensor1> to tensor<2x3x4xf64>
74 %d23 = sparse_tensor.convert %t23 : tensor<2x3x4xf64, #Tensor3> to tensor<2x3x4xf64>
[all …]
H A Dsparse_binary.mlir60 sparse_tensor.yield %2 : f64
84 sparse_tensor.yield %ret : f64
127 sparse_tensor.yield %ret : i32
153 sparse_tensor.yield %ret : f64
179 sparse_tensor.yield %ret : f64
201 sparse_tensor.yield %ret : f64
205 sparse_tensor.yield %x : f64
209 sparse_tensor.yield %y : f64
234 sparse_tensor.yield %ret : f64
242 sparse_tensor.yield %ret : f64
[all …]
/llvm-project-15.0.7/mlir/lib/Dialect/SparseTensor/Transforms/
H A DCodegenUtils.cpp15 using namespace mlir::sparse_tensor;
21 OverheadType mlir::sparse_tensor::overheadTypeEncoding(unsigned width) { in overheadTypeEncoding()
37 OverheadType mlir::sparse_tensor::overheadTypeEncoding(Type tp) { in overheadTypeEncoding()
61 OverheadType mlir::sparse_tensor::pointerOverheadTypeEncoding( in pointerOverheadTypeEncoding()
66 OverheadType mlir::sparse_tensor::indexOverheadTypeEncoding( in indexOverheadTypeEncoding()
71 Type mlir::sparse_tensor::getPointerOverheadType( in getPointerOverheadType()
76 Type mlir::sparse_tensor::getIndexOverheadType( in getIndexOverheadType()
98 StringRef mlir::sparse_tensor::overheadTypeFunctionSuffix(Type tp) { in overheadTypeFunctionSuffix()
102 PrimaryType mlir::sparse_tensor::primaryTypeEncoding(Type elemTp) { in primaryTypeEncoding()
140 StringRef mlir::sparse_tensor::primaryTypeFunctionSuffix(Type elemTp) { in primaryTypeFunctionSuffix()
[all …]
H A DBufferizableOpInterfaceImpl.cpp23 using namespace mlir::sparse_tensor;
26 namespace sparse_tensor { namespace
31 sparse_tensor::ConvertOp> {
61 sparse_tensor::LoadOp> {
85 sparse_tensor::NewOp> {
101 void mlir::sparse_tensor::registerBufferizableOpInterfaceExternalModels( in registerBufferizableOpInterfaceExternalModels()
104 +[](MLIRContext *ctx, sparse_tensor::SparseTensorDialect *dialect) { in registerBufferizableOpInterfaceExternalModels()
105 sparse_tensor::ConvertOp::attachInterface<ConvertOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
106 sparse_tensor::LoadOp::attachInterface<LoadOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()
107 sparse_tensor::NewOp::attachInterface<NewOpInterface>(*ctx); in registerBufferizableOpInterfaceExternalModels()

123456