Making blockId var int64 too

pcastonguay · pcastonguay · commit 700da8318076 · 2025-10-09T16:50:06.000-07:00
Signed-off-by: Patrice Castonguay &lt;55748270+pcastonguay@users.noreply.github.com&gt;
diff --git a/cpp/tensorrt_llm/executor/cache_transmission/cacheSplitConcat.cu b/cpp/tensorrt_llm/executor/cache_transmission/cacheSplitConcat.cu
@@ -584,7 +584,7 @@ __global__ void splitKVCacheForMLAKernel(T const** __restrict__ inputBlocks, T**
     using VecType = typename common::BytesToType<vecSizeByte>::type;
 #pragma unroll 1
 
-    for (int blockId = blockIdx.y; blockId < inputBlockNum; blockId += gridDim.y)
+    for (int64_t blockId = blockIdx.y; blockId < inputBlockNum; blockId += gridDim.y)
     {
 #pragma unroll 1
 

Original file line number	Diff line number	Diff line change
`@@ -584,7 +584,7 @@ __global__ void splitKVCacheForMLAKernel(T const __restrict__ inputBlocks, T`
`584`	`584`	`using VecType = typename common::BytesToType<vecSizeByte>::type;`
`585`	`585`	`#pragma unroll 1`
`586`	`586`
`587`		`- for (int blockId = blockIdx.y; blockId < inputBlockNum; blockId += gridDim.y)`
	`587`	`+ for (int64_t blockId = blockIdx.y; blockId < inputBlockNum; blockId += gridDim.y)`
`588`	`588`	`{`
`589`	`589`	`#pragma unroll 1`
`590`	`590`