fix typo

zcin · zcin · commit 7c4ff34c4e25 · 2025-10-16T19:59:46.000-07:00
Signed-off-by: Cindy Zhang &lt;cindyzyx9@gmail.com&gt;
diff --git a/csrc/renorm.cu b/csrc/renorm.cu
@@ -35,7 +35,7 @@ void top_p_renorm_probs(TensorView probs, TensorView renorm_probs,
   cudaError_t status = sampling::TopPRenormProb<float>(
       static_cast<float*>(probs->data), static_cast<float*>(renorm_probs->data),
       has_top_p_arr ? static_cast<float*>(maybe_top_p_arr.value()->data) : nullptr, batch_size,
-      top_p_val, vocab_size, probs_strides[0], stream);
+      top_p_val, vocab_size, probs->strides[0], stream);
   TVM_FFI_ICHECK(status == cudaSuccess)
       << "TopPRenormProb failed with error code " << cudaGetErrorString(status);
 }

Original file line number	Diff line number	Diff line change
`@@ -35,7 +35,7 @@ void top_p_renorm_probs(TensorView probs, TensorView renorm_probs,`
`35`	`35`	`cudaError_t status = sampling::TopPRenormProb<float>(`
`36`	`36`	`static_cast<float>(probs->data), static_cast<float>(renorm_probs->data),`
`37`	`37`	`has_top_p_arr ? static_cast<float*>(maybe_top_p_arr.value()->data) : nullptr, batch_size,`
`38`		`- top_p_val, vocab_size, probs_strides[0], stream);`
	`38`	`+ top_p_val, vocab_size, probs->strides[0], stream);`
`39`	`39`	`TVM_FFI_ICHECK(status == cudaSuccess)`
`40`	`40`	`<< "TopPRenormProb failed with error code " << cudaGetErrorString(status);`
`41`	`41`	`}`