Skip to content

GitLab

Commit 136930ae authored by xiabo's avatar xiabo
Browse files

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

parent 7e0cdd29
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
csrc/custom_all_reduce.cuh
View file @ 136930ae
...@@ -1062,7 +1062,7 @@ class CustomAllreduce { ...@@ -1062,7 +1062,7 @@ class CustomAllreduce {
#define KL(ngpus, name) \ #define KL(ngpus, name) \
{ \ { \
void* kernelArgs[] = { \ void* kernelArgs[] = { \
&ptrs, &sg_, &self_sg_, &output, &rank_, &size \ &ptrs, &sg_, &self_sg_, &output, &rank_, &size, &dev_curr_hdp_reg, &world_size_ \
}; \ }; \
hipExtLaunchKernel( \ hipExtLaunchKernel( \
(void*)name<T, ngpus>, \ (void*)name<T, ngpus>, \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment