Skip to content

Commit 30c1866

Browse files
committed
[mlir][sparse][gpu] enable SpGEMM on GPU for libgen path
Direct IR supports pack, but libgen parth did not until this was added in https://reviews.llvm.org/D158012 Reviewed By: Peiming Differential Revision: https://reviews.llvm.org/D158020
1 parent cb5bc75 commit 30c1866

File tree

2 files changed

+12
-2
lines changed

2 files changed

+12
-2
lines changed

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

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,13 +1,23 @@
11
//
22
// NOTE: this test requires gpu-sm80
33
//
4+
// with RT lib:
5+
//
6+
// RUN: mlir-opt %s \
7+
// RUN: --sparse-compiler="enable-runtime-library=true enable-gpu-libgen gpu-triple=nvptx64-nvidia-cuda gpu-chip=sm_80 gpu-features=+ptx71" \
8+
// RUN: | mlir-cpu-runner \
9+
// RUN: --shared-libs=%mlir_cuda_runtime \
10+
// RUN: --shared-libs=%mlir_c_runner_utils \
11+
// RUN: --e main --entry-point-result=void \
12+
// RUN: | FileCheck %s
13+
//
414
// without RT lib:
515
//
616
// RUN: mlir-opt %s \
717
// RUN: --sparse-compiler="enable-runtime-library=false enable-gpu-libgen gpu-triple=nvptx64-nvidia-cuda gpu-chip=sm_80 gpu-features=+ptx71" \
818
// RUN: | mlir-cpu-runner \
919
// RUN: --shared-libs=%mlir_cuda_runtime \
10-
// RUN: --shared-libs=%mlir_runner_utils \
20+
// RUN: --shared-libs=%mlir_c_runner_utils \
1121
// RUN: --e main --entry-point-result=void \
1222
// RUN: | FileCheck %s
1323

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@
2323
// RUN: %{compile} enable-runtime-library=false gpu-data-transfer-strategy=pinned-dma" | %{run}
2424
// Tracker #64316
2525
// RUNNOT: %{compile} enable-runtime-library=false gpu-data-transfer-strategy=zero-copy" | %{run}
26-
//
26+
//
2727

2828
!Filename = !llvm.ptr<i8>
2929

0 commit comments

Comments
 (0)