1// RUN: mlir-opt %s -split-input-file | mlir-opt | FileCheck %s
2
3// CHECK-LABEL: func private @sparse_1d_tensor(
4// CHECK-SAME: tensor<32xf64, #sparse_tensor.encoding<{ dimLevelType = [ "compressed" ], pointerBitWidth = 0, indexBitWidth = 0 }>>)
5func.func private @sparse_1d_tensor(tensor<32xf64, #sparse_tensor.encoding<{dimLevelType = ["compressed"]}>>)
6
7// -----
8
9#CSR = #sparse_tensor.encoding<{
10  dimLevelType = [ "dense", "compressed" ],
11  dimOrdering = affine_map<(i,j) -> (i,j)>,
12  pointerBitWidth = 64,
13  indexBitWidth = 64
14}>
15
16// CHECK-LABEL: func private @sparse_2d_tensor(
17// CHECK-SAME: tensor<?x?xf32, #sparse_tensor.encoding<{ dimLevelType = [ "dense", "compressed" ], dimOrdering = affine_map<(d0, d1) -> (d0, d1)>, pointerBitWidth = 64, indexBitWidth = 64 }>>)
18func.func private @sparse_2d_tensor(tensor<?x?xf32, #CSR>)
19