解决custom allreduce在K100AI上新模型报错问题

136930ae · xiabo · 7e0cdd29 · 136930ae
Commit 136930ae authored Mar 05, 2026 by xiabo
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

csrc/custom_all_reduce.cuh csrc/custom_all_reduce.cuh +1 -1

No files found.
--- a/csrc/custom_all_reduce.cuh
+++ b/csrc/custom_all_reduce.cuh
@@ -1062,7 +1062,7 @@ class CustomAllreduce {
 #define KL(ngpus, name)                                                       \
  {                                                                           \
    void* kernelArgs[] = {                                                    \
-      &ptrs, &sg_, &self_sg_, &output, &rank_, &size                          \
+      &ptrs, &sg_, &self_sg_, &output, &rank_, &size, &dev_curr_hdp_reg, &world_size_                         \
    };                                                                        \
    hipExtLaunchKernel(                                                       \
      (void*)name<T, ngpus>,                                                  \