AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix Team reduction intra-node

Browse Source
This commit is contained in:
Yiltan Hassan Temucin
2025-02-06 16:23:45 -06:00
parent 248972b30b
commit 022b2c27e7
6 changed files with 12 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/reverse_offload/context_ro_device.cpp
+1 -1
View File
@@ -507,7 +507,7 @@ __device__ void build_queue_element(
queue_element->op = op;
queue_element->datatype = datatype;
}
if (type == RO_NET_TEAM_TO_ALL) {
if (type == RO_NET_TEAM_REDUCE) {
queue_element->op = op;
queue_element->datatype = datatype;
queue_element->team_comm = team_comm;