Lines Matching refs:CONV

2 // RUN: mlir-opt %s --sparse-tensor-conversion --cse | FileCheck %s --check-prefix=CHECK-CONV
17 // CHECK-CONV-LABEL: func.func @sparse_expand(
18 // CHECK-CONV-DAG:  %[[C0:.*]] = arith.constant 0 : index
19 // CHECK-CONV-DAG:  %[[C1:.*]] = arith.constant 1 : index
20 // CHECK-CONV-DAG:  %[[C10:.*]] = arith.constant 10 : index
21 // CHECK-CONV-DAG:  call @newSparseTensor
22 // CHECK-CONV-DAG:  call @newSparseTensor
23 // CHECK-CONV:      scf.while : () -> () {
24 // CHECK-CONV:        call @getNextF64
25 // CHECK-CONV:        scf.condition(%13)
26 // CHECK-CONV:      } do {
27 // CHECK-CONV:        %[[X:.*]] = memref.load %{{.*}}[%[[C0]]] : memref<?xindex>
28 // CHECK-CONV:        %[[D:.*]] = arith.divui %[[X]], %[[C10]] : index
29 // CHECK-CONV:        memref.store %[[D]], %{{.*}}[%[[C0]]] : memref<?xindex>
30 // CHECK-CONV:        %[[R:.*]] = arith.remui %[[X]], %[[C10]] : index
31 // CHECK-CONV:        memref.store %[[R]], %{{.*}}[%[[C1]]] : memref<?xindex>
32 // CHECK-CONV:        call @addEltF64
33 // CHECK-CONV:        scf.yield
34 // CHECK-CONV:      }
35 // CHECK-CONV:      %[[N:.*]] = call @newSparseTensor
36 // CHECK-CONV:      call @delSparseTensorCOOF64
37 // CHECK-CONV:      call @delSparseTensorCOOF64
38 // CHECK-CONV:      return %[[N]] : !llvm.ptr<i8>
56 // CHECK-CONV-LABEL: func.func @sparse_collapse(
57 // CHECK-CONV-DAG:  %[[C0:.*]] = arith.constant 0 : index
58 // CHECK-CONV-DAG:  %[[C1:.*]] = arith.constant 1 : index
59 // CHECK-CONV-DAG:  %[[C10:.*]] = arith.constant 10 : index
60 // CHECK-CONV-DAG:  call @newSparseTensor
61 // CHECK-CONV-DAG:  call @newSparseTensor
62 // CHECK-CONV:      scf.while : () -> () {
63 // CHECK-CONV:        call @getNextF64
64 // CHECK-CONV:        scf.condition(%13)
65 // CHECK-CONV:      } do {
66 // CHECK-CONV:        %[[X:.*]] = memref.load %{{.*}}[%[[C0]]] : memref<?xindex>
67 // CHECK-CONV:        %[[M:.*]] = arith.muli %[[X]], %[[C10]] : index
68 // CHECK-CONV:        %[[Y:.*]] = memref.load %{{.*}}[%[[C1]]] : memref<?xindex>
69 // CHECK-CONV:        %[[A:.*]] = arith.addi %[[M]], %[[Y]] : index
70 // CHECK-CONV:        memref.store %[[A]], %{{.*}}[%[[C0]]] : memref<?xindex>
71 // CHECK-CONV:        call @addEltF64
72 // CHECK-CONV:        scf.yield
73 // CHECK-CONV:      }
74 // CHECK-CONV:      %[[N:.*]] = call @newSparseTensor
75 // CHECK-CONV:      call @delSparseTensorCOOF64
76 // CHECK-CONV:      call @delSparseTensorCOOF64
77 // CHECK-CONV:      return %[[N]] : !llvm.ptr<i8>