Fix overflowing risk (sgl-project#440)

iridiumine · web-flow · commit c7b243e4573a · 2026-04-13T09:15:39.000+08:00
* fix overflowing risk

* fix overflowing risk

---------

Co-authored-by: iridiumine &lt;iridiumine@users.noreply.github.com&gt;
diff --git a/csrc/lightning_indexer/op_host/tiling/lightning_indexer_tiling.cpp b/csrc/lightning_indexer/op_host/tiling/lightning_indexer_tiling.cpp
@@ -364,7 +364,11 @@ ge::graphStatus LIInfoParser::GetS2SizeForPageAttention()
         return ge::GRAPH_FAILED;
     }
     maxBlockNumPerBatch_ = opParamInfo_.blockTable.tensor->GetStorageShape().GetDim(1);
-    s2Size_ = maxBlockNumPerBatch_ * blockSize_;
+    const int64_t s2SizeTemp = static_cast<int64_t>(maxBlockNumPerBatch_) * static_cast<int64_t>(blockSize_);
+    if (s2SizeTemp > static_cast<int64_t>(std::numeric_limits<uint32_t>::max())) {
+        return ge::GRAPH_FAILED;
+    }
+    s2Size_ = static_cast<uint32_t>(s2SizeTemp);
     return ge::GRAPH_SUCCESS;
 }
 

Original file line number	Diff line number	Diff line change
`@@ -364,7 +364,11 @@ ge::graphStatus LIInfoParser::GetS2SizeForPageAttention()`
`364`	`364`	`return ge::GRAPH_FAILED;`
`365`	`365`	`}`
`366`	`366`	`maxBlockNumPerBatch_ = opParamInfo_.blockTable.tensor->GetStorageShape().GetDim(1);`
`367`		`- s2Size_ = maxBlockNumPerBatch_ * blockSize_;`
	`367`	`+ const int64_t s2SizeTemp = static_cast<int64_t>(maxBlockNumPerBatch_) * static_cast<int64_t>(blockSize_);`
	`368`	`+ if (s2SizeTemp > static_cast<int64_t>(std::numeric_limits<uint32_t>::max())) {`
	`369`	`+ return ge::GRAPH_FAILED;`
	`370`	`+ }`
	`371`	`+ s2Size_ = static_cast<uint32_t>(s2SizeTemp);`
`368`	`372`	`return ge::GRAPH_SUCCESS;`
`369`	`373`	`}`
`370`	`374`