From 7e44f0736a24a36ea40d733bece38896e70e60c7 Mon Sep 17 00:00:00 2001
From: Kun Wu <kunww@google.com>
Date: Thu, 1 Jun 2023 18:00:21 +0000
Subject: [PATCH] [mlir][gpu][sparse] fix broken type in cusparseCreateCsr

Differential Revision: https://reviews.llvm.org/D151912
---
 mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
index 17be418..1c9497b 100644
--- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
+++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -322,7 +322,7 @@ mgpuCreateCsr(intptr_t rows, intptr_t cols, intptr_t nnz, void *rowPos,
   auto dTp = static_cast<cudaDataType_t>(dtp);
   CUSPARSE_REPORT_IF_ERROR(cusparseCreateCsr(&mat, rows, cols, nnz, rowPos,
                                              colIdxs, values, pTp, iTp,
-                                             CUSPARSE_INDEX_BASE_ZERO, dtp))
+                                             CUSPARSE_INDEX_BASE_ZERO, dTp))
   return reinterpret_cast<void *>(mat);
 }
 
-- 
2.7.4