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

[llvmlistbot at llvm.org]

llvmbot wrote:


<!--LLVM PR SUMMARY COMMENT-->

@llvm/pr-subscribers-mlir
            
<details>
<summary>Changes</summary>
Rationale:
Some compiler settings don't like the size_t vs uint64_t setup.
--
Full diff: https://github.com/llvm/llvm-project/pull/66438.diff

1 Files Affected:

- (modified) mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp (+3-3) 


<pre>
diff --git a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
index 7bf6804902479a8..8a1417d7750e586 100644
--- a/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
+++ b/mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp
@@ -792,9 +792,9 @@ mgpuCuSparseLtSpMMBufferSize(void *bs, int32_t ma, int32_t mb, void *a, void *b,
   auto matA = reinterpret_cast&lt;cusparseLtSpMatHandleAndData *&gt;(a);
   auto matB = reinterpret_cast&lt;cusparseLtDnMatHandleAndData *&gt;(b);
   auto matC = reinterpret_cast&lt;cusparseLtDnMatHandleAndData *&gt;(c);
-  auto workspace_size = reinterpret_cast&lt;int64_t *&gt;(bs);
-  auto compressed_size = &amp;(reinterpret_cast&lt;int64_t *&gt;(bs)[1]);
-  auto compressed_buffer_size = &amp;(reinterpret_cast&lt;int64_t *&gt;(bs)[2]);
+  auto workspace_size = reinterpret_cast&lt;size_t *&gt;(bs);
+  auto compressed_size = &amp;(reinterpret_cast&lt;size_t *&gt;(bs)[1]);
+  auto compressed_buffer_size = &amp;(reinterpret_cast&lt;size_t *&gt;(bs)[2]);
   auto cTp = static_cast&lt;cusparseComputeType&gt;(ctp);
 
   cusparseOperation_t modeA = static_cast&lt;cusparseOperation_t&gt;(ma);
</pre>
</details>


https://github.com/llvm/llvm-project/pull/66438