28 lines · plain
1// RUN: mlir-opt %s --loop-invariant-code-motion | FileCheck %s2 3#CSR = #sparse_tensor.encoding<{4 map = (i, j) -> (5 i : dense,6 j : compressed7 )8}>9 10// Make sure that pure instructions are hoisted outside the loop.11//12// CHECK: sparse_tensor.values13// CHECK: sparse_tensor.positions14// CHECK: sparse_tensor.coordinate15// CHECK: sparse_tensor.iterate16func.func @sparse_iterate(%sp : tensor<?x?xf64, #CSR>) {17 %l1 = sparse_tensor.extract_iteration_space %sp lvls = 0 : tensor<?x?xf64, #CSR>18 -> !sparse_tensor.iter_space<#CSR, lvls = 0>19 sparse_tensor.iterate %it1 in %l1 at (%crd) : !sparse_tensor.iter_space<#CSR, lvls = 0> {20 %0 = sparse_tensor.values %sp : tensor<?x?xf64, #CSR> to memref<?xf64>21 %1 = sparse_tensor.positions %sp { level = 1 : index } : tensor<?x?xf64, #CSR> to memref<?xindex>22 %2 = sparse_tensor.coordinates %sp { level = 1 : index } : tensor<?x?xf64, #CSR> to memref<?xindex>23 "test.op"(%0, %1, %2) : (memref<?xf64>, memref<?xindex>, memref<?xindex>) -> ()24 }25 26 return27}28