From 2b99848799fb66ef390ecb2623d81e192a5f9e02 Mon Sep 17 00:00:00 2001
From: lxn_WSL <1933650668@qq.com>
Date: Sun, 5 Jul 2026 23:54:24 +0800
Subject: [PATCH] Fix AllReduce3 auto-tune launch parameters

---
 python/mscclpp_benchmark/mscclpp_op.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/python/mscclpp_benchmark/mscclpp_op.py b/python/mscclpp_benchmark/mscclpp_op.py
index 60ecf0a12..0a105808d 100644
--- a/python/mscclpp_benchmark/mscclpp_op.py
+++ b/python/mscclpp_benchmark/mscclpp_op.py
@@ -208,7 +208,7 @@ def __init__(
         self.set_params(nblocks, block_size)
 
     def __call__(self, stream):
-        self.kernel.launch_kernel(self.params, 24, 1024, 0, stream)
+        self.kernel.launch_kernel(self.params, self.nblocks, self.block_size, 0, stream)
         return self.memory
 
     def set_params(self, nblocks, block_size):