a0615d020a
This commit is part of the migration of towards the new STEA syntax/design. In particular, this commit includes the following changes:
* Renaming compiler-internal functions/methods:
* `SparseTensorEncodingAttr::{getDimLevelType => getLvlTypes}`
* `Merger::{getDimLevelType => getLvlType}` (for consistency)
* `sparse_tensor::{getDimLevelType => buildLevelType}` (to help reduce confusion vs actual getter methods)
* Renaming external facets to match:
* the STEA parser and printer
* the C and Python bindings
* PyTACO
However, the actual renaming of the `DimLevelType` itself (along with all the "dlt" names) will be handled in a separate commit.
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D150330
15 lines
727 B
MLIR
15 lines
727 B
MLIR
// RUN: mlir-opt %s --sparse-tensor-codegen --sparse-storage-specifier-to-llvm | FileCheck %s
|
|
|
|
#SparseVector = #sparse_tensor.encoding<{ lvlTypes = ["compressed"] }>
|
|
|
|
// CHECK-LABEL: func @sparse_nop(
|
|
// CHECK-SAME: %[[A0:.*0]]: memref<?xindex>,
|
|
// CHECK-SAME: %[[A1:.*1]]: memref<?xindex>,
|
|
// CHECK-SAME: %[[A2:.*2]]: memref<?xf64>,
|
|
// CHECK-SAME: %[[A3:.*3]]: !llvm.struct<(array<1 x i64>, array<3 x i64>)>)
|
|
// CHECK: return %[[A0]], %[[A1]], %[[A2]], %[[A3]] :
|
|
// CHECK-SAME: memref<?xindex>, memref<?xindex>, memref<?xf64>, !llvm.struct<(array<1 x i64>, array<3 x i64>)>
|
|
func.func @sparse_nop(%arg0: tensor<?xf64, #SparseVector>) -> tensor<?xf64, #SparseVector> {
|
|
return %arg0 : tensor<?xf64, #SparseVector>
|
|
}
|