[Mlir-commits] [mlir] 8998bcf - [mlir][sparse][gpu] refine type of workspace size variables (#66438)

[llvmlistbot at llvm.org]

Author: Aart Bik
Date: 2023-09-14T15:49:52-07:00
New Revision: 8998bcfbce5f380aebb20f0d6704c1541c811fc1

URL: https://github.com/llvm/llvm-project/commit/8998bcfbce5f380aebb20f0d6704c1541c811fc1
DIFF: https://github.com/llvm/llvm-project/commit/8998bcfbce5f380aebb20f0d6704c1541c811fc1.diff

LOG: [mlir][sparse][gpu] refine type of workspace size variables (#66438)

Rationale:
Some compiler settings don't like the size_t vs uint64_t setup.

Added: 
    

Modified: 
    mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

Removed: 
    


################################################################################
diff  --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
index d19d473a5327627..3dbbc674128732e 100644
--- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
+++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -813,9 +813,9 @@ mgpuCuSparseLtSpMMBufferSize(void *bs, int32_t ma, int32_t mb, void *a, void *b,
   auto matA = reinterpret_cast<cusparseLtSpMatHandleAndData *>(a);
   auto matB = reinterpret_cast<cusparseLtDnMatHandleAndData *>(b);
   auto matC = reinterpret_cast<cusparseLtDnMatHandleAndData *>(c);
-  auto workspace_size = reinterpret_cast<int64_t *>(bs);
-  auto compressed_size = &(reinterpret_cast<int64_t *>(bs)[1]);
-  auto compressed_buffer_size = &(reinterpret_cast<int64_t *>(bs)[2]);
+  auto workspace_size = reinterpret_cast<size_t *>(bs);
+  auto compressed_size = &(reinterpret_cast<size_t *>(bs)[1]);
+  auto compressed_buffer_size = &(reinterpret_cast<size_t *>(bs)[2]);
   auto cTp = static_cast<cusparseComputeType>(ctp);
 
   cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma);