Skip to content

Commit 7f5fb90

Browse files
committed
[mlir][sparse] Fixing GPU tests (followup to D150330)
The GPU tests weren't updated when rebasing D150330, so this patch fixes that. Reviewed By: anlunx Differential Revision: https://reviews.llvm.org/D150822
1 parent 691927c commit 7f5fb90

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

mlir/test/Dialect/SparseTensor/GPU/gpu_matvec_lib.mlir

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
// RUN: --sparsification="enable-gpu-libgen" | FileCheck %s
33

44
#SortedCOO = #sparse_tensor.encoding<{
5-
dimLevelType = [ "compressed-nu", "singleton" ]
5+
lvlTypes = [ "compressed-nu", "singleton" ]
66
}>
77

88
module {

mlir/test/Dialect/SparseTensor/sparse_tensor_reshape.mlir

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,7 @@
11
// RUN: mlir-opt %s --post-sparsification-rewrite="enable-runtime-library=false enable-convert=false" \
22
// RUN: --cse --canonicalize | FileCheck %s
33

4-
#SparseMatrix = #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ] }>
4+
#SparseMatrix = #sparse_tensor.encoding<{ lvlTypes = [ "compressed", "compressed" ] }>
55

66
// CHECK: func.func @sparse_reshape(
77
// CHECK-SAME: %[[S:.*]]:
@@ -35,11 +35,11 @@
3535
// CHECK: scf.yield %[[RET_1]]
3636
// CHECK: }
3737
// CHECK: %[[NT1:.*]] = sparse_tensor.load %[[RET]] hasInserts
38-
// CHECK: return %[[NT1]] : tensor<10x10xf64, #sparse_tensor.encoding<{ dimLevelType = [ "compressed", "compressed" ] }>>
38+
// CHECK: return %[[NT1]] : tensor<10x10xf64, #sparse_tensor.encoding<{ lvlTypes = [ "compressed", "compressed" ] }>>
3939
//
4040
func.func @sparse_reshape(%arg0: tensor<4x25xf64, #SparseMatrix>) -> tensor<10x10xf64, #SparseMatrix> {
4141
%shape = arith.constant dense <[ 10, 10 ]> : tensor<2xi32>
4242
%0 = tensor.reshape %arg0(%shape) :
4343
(tensor<4x25xf64, #SparseMatrix>, tensor<2xi32>) -> tensor<10x10xf64, #SparseMatrix>
4444
return %0 : tensor<10x10xf64, #SparseMatrix>
45-
}
45+
}

mlir/test/Integration/Dialect/SparseTensor/GPU/CUDA/sparse-matvec-lib.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -14,11 +14,11 @@
1414
// TODO: without RT lib (AoS COO):
1515

1616
#SortedCOO = #sparse_tensor.encoding<{
17-
dimLevelType = [ "compressed-nu", "singleton" ]
17+
lvlTypes = [ "compressed-nu", "singleton" ]
1818
}>
1919

2020
#CSR = #sparse_tensor.encoding<{
21-
dimLevelType = [ "dense", "compressed" ],
21+
lvlTypes = [ "dense", "compressed" ],
2222
posWidth = 32,
2323
crdWidth = 32
2424
}>

0 commit comments

Comments
 (0)