Skip to content

Commit 8998bcf

Browse files
authored
[mlir][sparse][gpu] refine type of workspace size variables (#66438)
Rationale: Some compiler settings don't like the size_t vs uint64_t setup.
1 parent dc4bf78 commit 8998bcf

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -813,9 +813,9 @@ mgpuCuSparseLtSpMMBufferSize(void *bs, int32_t ma, int32_t mb, void *a, void *b,
813813
auto matA = reinterpret_cast<cusparseLtSpMatHandleAndData *>(a);
814814
auto matB = reinterpret_cast<cusparseLtDnMatHandleAndData *>(b);
815815
auto matC = reinterpret_cast<cusparseLtDnMatHandleAndData *>(c);
816-
auto workspace_size = reinterpret_cast<int64_t *>(bs);
817-
auto compressed_size = &(reinterpret_cast<int64_t *>(bs)[1]);
818-
auto compressed_buffer_size = &(reinterpret_cast<int64_t *>(bs)[2]);
816+
auto workspace_size = reinterpret_cast<size_t *>(bs);
817+
auto compressed_size = &(reinterpret_cast<size_t *>(bs)[1]);
818+
auto compressed_buffer_size = &(reinterpret_cast<size_t *>(bs)[2]);
819819
auto cTp = static_cast<cusparseComputeType>(ctp);
820820

821821
cusparseOperation_t modeA = static_cast<cusparseOperation_t>(ma);

0 commit comments

Comments
 (0)