From 7c1fd12329dca79812be6f0d9c7d98457a0f36c8 Mon Sep 17 00:00:00 2001
From: Skeleton003 <799284168@qq.com>
Date: Thu, 13 Jun 2024 11:33:48 +0000
Subject: [PATCH] add check in cuda file

---
 src/array/cuda/coo2csr.cu | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/array/cuda/coo2csr.cu b/src/array/cuda/coo2csr.cu
index 1105adef6c82..237a35a26b3e 100644
--- a/src/array/cuda/coo2csr.cu
+++ b/src/array/cuda/coo2csr.cu
@@ -40,6 +40,7 @@ CSRMatrix COOToCSR<kDGLCUDA, int32_t>(COOMatrix coo) {
   }
 
   const int64_t nnz = coo.row->shape[0];
+  CHECK_NO_OVERFLOW(coo.row->dtype, nnz);
   // TODO(minjie): Many of our current implementation assumes that CSR must have
   //   a data array. This is a temporary workaround. Remove this after:
   //   - The old immutable graph implementation is deprecated.