1// RUN: mlir-opt %s --loop-invariant-code-motion | FileCheck %s 2 3#CSR = #sparse_tensor.encoding<{ 4 map = (i, j) -> ( 5 i : dense, 6 j : compressed 7 ) 8}> 9 10// Make sure that pure instructions are hoisted outside the loop. 11// 12// CHECK: sparse_tensor.values 13// CHECK: sparse_tensor.positions 14// CHECK: sparse_tensor.coordinate 15// CHECK: sparse_tensor.iterate 16func.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 return 27} 28