7b9fb1c228
Updates: 1. Verification of block sparsity. 2. Verification of singleton level type can only follow compressed or loose_compressed levels. And all level types after singleton should be singleton. 3. Added getBlockSize function. 4. Added an invalid encoding test for an incorrect lvlToDim map that user provides.
273 lines
9.6 KiB
MLIR
273 lines
9.6 KiB
MLIR
// RUN: mlir-opt %s -split-input-file -verify-diagnostics
|
|
|
|
// expected-error@+1 {{expected '(' in dimension-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = []}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected '->'}}
|
|
#a = #sparse_tensor.encoding<{map = ()}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected ')' in dimension-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = (d0 -> d0)}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected '(' in dimension-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = d0 -> d0}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected '(' in level-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> d0}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected ':'}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0)}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected valid level format (e.g. dense, compressed or singleton)}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0:)}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected valid level format (e.g. dense, compressed or singleton)}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : (compressed))}>
|
|
func.func private @scalar(%arg0: tensor<f64, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+2 {{dimension-rank mismatch between encoding and tensor shape: 2 != 1}}
|
|
#a = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 : dense, d1 : compressed)}>
|
|
func.func private @tensor_dimlevel_size_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{use of undeclared identifier}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : dense, d1 : compressed)}>
|
|
func.func private @tensor_sizes_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{failed to infer lvlToDim from dimToLvl}}
|
|
#a = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 : dense)}>
|
|
func.func private @tensor_sizes_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected bare identifier}}
|
|
#a = #sparse_tensor.encoding<{map = (1)}>
|
|
func.func private @tensor_type_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{unexpected key: nap}}
|
|
#a = #sparse_tensor.encoding<{nap = (d0) -> (d0 : dense)}>
|
|
func.func private @tensor_type_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected '(' in dimension-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = -> (d0 : dense)}>
|
|
func.func private @tensor_type_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{unknown level format: strange}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : strange)}>
|
|
func.func private @tensor_value_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected valid level format (e.g. dense, compressed or singleton)}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : "wrong")}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected valid level property (e.g. nonordered, nonunique or high)}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : compressed("wrong"))}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{expected ')' in level-specifier list}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : compressed[high])}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{unknown level property: wrong}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : compressed(wrong))}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{use of undeclared identifier}}
|
|
#a = #sparse_tensor.encoding<{map = (d0) -> (d0 : compressed, dense)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<8xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{failed to infer lvlToDim from dimToLvl}}
|
|
#a = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 : dense, d0 : compressed)}>
|
|
func.func private @tensor_no_permutation(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{unexpected character}}
|
|
#a = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 : compressed; d1 : dense)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected attribute value}}
|
|
#a = #sparse_tensor.encoding<{map = (d0: d1) -> (d0 : compressed, d1 : dense)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected ':'}}
|
|
#a = #sparse_tensor.encoding<{map = (d0, d1) -> (d0 = compressed, d1 = dense)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected attribute value}}
|
|
#a = #sparse_tensor.encoding<{map = (d0 : compressed, d1 : compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{use of undeclared identifier}}
|
|
#a = #sparse_tensor.encoding<{map = (d0 = compressed, d1 = compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{use of undeclared identifier}}
|
|
#a = #sparse_tensor.encoding<{map = (d0 = l0, d1 = l1) {l0, l1} -> (l0 = d0 : dense, l1 = d1 : compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected '='}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (l0 : d0 = dense, l1 : d1 = compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{use of undeclared identifier 'd0'}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (d0 : l0 = dense, d1 : l1 = compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{use of undeclared identifier 'd0'}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (d0 : dense, d1 : compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{expected '='}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (l0 : dense, l1 : compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{use of undeclared identifier}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (l0 = dense, l1 = compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
// expected-error@+1 {{use of undeclared identifier 'd0'}}
|
|
#a = #sparse_tensor.encoding<{map = {l0, l1} (d0 = l0, d1 = l1) -> (d0 = l0 : dense, d1 = l1 : compressed)}>
|
|
func.func private @tensor_dimtolvl_mismatch(%arg0: tensor<16x32xi32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#a = #sparse_tensor.encoding<{posWidth = "x"}> // expected-error {{expected an integral position bitwidth}}
|
|
func.func private @tensor_no_int_ptr(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#a = #sparse_tensor.encoding<{posWidth = 42}> // expected-error {{unexpected position bitwidth: 42}}
|
|
func.func private @tensor_invalid_int_ptr(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#a = #sparse_tensor.encoding<{crdWidth = "not really"}> // expected-error {{expected an integral index bitwidth}}
|
|
func.func private @tensor_no_int_index(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#a = #sparse_tensor.encoding<{crdWidth = 128}> // expected-error {{unexpected coordinate bitwidth: 128}}
|
|
func.func private @tensor_invalid_int_index(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#a = #sparse_tensor.encoding<{key = 1}> // expected-error {{unexpected key: key}}
|
|
func.func private @tensor_invalid_key(%arg0: tensor<16x32xf32, #a>) -> ()
|
|
|
|
// -----
|
|
|
|
#CSR_SLICE = #sparse_tensor.encoding<{
|
|
map = (d0 : #sparse_tensor<slice(-1, ?, 1)>, d1 : #sparse_tensor<slice(?, 4, 2)>) -> (d0 : dense, d1 : compressed)// expected-error{{expect positive value or ? for slice offset/size/stride}}
|
|
}>
|
|
func.func private @sparse_slice(tensor<?x?xf64, #CSR_SLICE>)
|
|
|
|
// -----
|
|
|
|
// expected-error@+2 {{Level-rank mismatch between forward-declarations and specifiers. Declared 3 level-variables; but got 2 level-specifiers.}}
|
|
#TooManyLvlDecl = #sparse_tensor.encoding<{
|
|
map = {l0, l1, l2} (d0, d1) -> (l0 = d0 : dense, l1 = d1 : compressed)
|
|
}>
|
|
func.func private @too_many_lvl_decl(%arg0: tensor<?x?xf64, #TooManyLvlDecl>) {
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-error@+2 {{use of undeclared identifier 'l1'}}
|
|
#TooFewLvlDecl = #sparse_tensor.encoding<{
|
|
map = {l0} (d0, d1) -> (l0 = d0 : dense, l1 = d1 : compressed)
|
|
}>
|
|
func.func private @too_few_lvl_decl(%arg0: tensor<?x?xf64, #TooFewLvlDecl>) {
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-error@+2 {{Level-variable ordering mismatch. The variable 'l0' was forward-declared as the 1st level; but is bound by the 0th specification.}}
|
|
#WrongOrderLvlDecl = #sparse_tensor.encoding<{
|
|
map = {l1, l0} (d0, d1) -> (l0 = d0 : dense, l1 = d1 : compressed)
|
|
}>
|
|
func.func private @wrong_order_lvl_decl(%arg0: tensor<?x?xf64, #WrongOrderLvlDecl>) {
|
|
return
|
|
}
|
|
|
|
// -----
|
|
|
|
// expected-error@+1 {{expected lvlToDim to be an inverse of dimToLvl}}
|
|
#BSR_explicit = #sparse_tensor.encoding<{
|
|
map =
|
|
{il, jl, ii, jj}
|
|
( i = il * 3 + ii,
|
|
j = jl * 2 + jj
|
|
) ->
|
|
( il = i floordiv 2 : dense,
|
|
jl = j floordiv 3 : compressed,
|
|
ii = i mod 2 : dense,
|
|
jj = j mod 3 : dense
|
|
)
|
|
}>
|
|
func.func private @BSR_explicit(%arg0: tensor<?x?xf64, #BSR_explicit>) {
|
|
return
|
|
}
|