microsoft/mscclpp
1
0
Fork 0
mirror of https://github.com/microsoft/mscclpp.git synced 2026-05-11 17:00:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

Adding Channel Type in ReduceSend Operation on DSL (#777)

Browse Source 
The reduce send operation in DSL essentially combines the reduce and put
operations. The put operation carry the information about the channel
type, whereas previously, we were using the channel type from the reduce
operation.
This commit is contained in:
Caio Rocha
2026-04-08 16:59:08 -07:00
committed by GitHub
parent ed565ceb33
commit 3e5c41c98a
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/mscclpp/language/internal/operations.py
View File

@@ -745,7 +745,7 @@ class ReduceOperation(BaseOperation):
remote_dst_buff=self.remote_dst_buff + other.dst_buff,
channel_ids=self.channel_ids,
put_channel_ids=self.put_channel_ids + other.channel_ids,
channel_type=self.channel_type,
channel_type=other.channel_type,
reduce_operation=self.reduce_operation,
tbg_info=self.tbg_info,
packet=self.packet,