This is an archive of the discontinued LLVM Phabricator instance.

Differential D141006

[mlir][sparse] add test that combines sparse codegen and lowering to llvm struct
ClosedPublic

Authored by aartbik on Jan 4 2023, 11:34 AM.

Details

Reviewers
bixia
nicolasvasilache
Commits
rG1c7ffe0c38eb: [mlir][sparse] add test that combines sparse codegen and lowering to llvm struct

Diff Detail

Event Timeline

aartbik created this revision.Jan 4 2023, 11:34 AM
Herald added a project: Restricted Project. · View Herald TranscriptJan 4 2023, 11:34 AM
Herald added subscribers: hanchung, jsetoain, Moerafaat and 21 others. · View Herald Transcript
aartbik requested review of this revision.Jan 4 2023, 11:34 AM
Herald added a reviewer: nicolasvasilache. · View Herald TranscriptJan 4 2023, 11:34 AM
Herald added a project: Restricted Project. · View Herald Transcript
Herald added subscribers: stephenneuendorffer, nicolasvasilache. · View Herald Transcript
bixia accepted this revision.Jan 4 2023, 11:39 AM
This revision is now accepted and ready to land.Jan 4 2023, 11:39 AM
Harbormaster completed remote builds in B205746: Diff 486351.Jan 4 2023, 12:05 PM
Closed by commit rG1c7ffe0c38eb: [mlir][sparse] add test that combines sparse codegen and lowering to llvm struct (authored by aartbik). · Explain WhyJan 4 2023, 12:12 PM
This revision was automatically updated to reflect the committed changes.
aartbik added a commit: rG1c7ffe0c38eb: [mlir][sparse] add test that combines sparse codegen and lowering to llvm struct.

Revision Contents

PathSize
mlir/
test/
Dialect/
SparseTensor/
14 lines

Diff 486364

mlir/test/Dialect/SparseTensor/codegen_to_llvm.mlir

  • This file was added.
// RUN: mlir-opt %s --sparse-tensor-codegen --sparse-storage-specifier-to-llvm | FileCheck %s
#SparseVector = #sparse_tensor.encoding<{ dimLevelType = ["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>
}