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

3c05d645 · xiabo · 2fc9d1a2 · 3c05d645
Commit 3c05d645 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
@@ -651,7 +651,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>,                                                  \