Skip to content

Commit

Permalink
add comments
Browse files Browse the repository at this point in the history
  • Loading branch information
Binyang2014 committed Jun 12, 2024
1 parent 04b622e commit 5060748
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion tests/test_language.py
Original file line number Diff line number Diff line change
Expand Up @@ -241,7 +241,8 @@ def test_instruction_fusion_multi_deps_mscclpp():
collective = AllReduce(3, 1, True)
prgm = MSCCLPPProgram("allreduce", topology, collective, 1)
# last reduce_packet depends on put_packets(write after read)
# and first reduce_packet(rank 2 put data to rank 1, data put depends on first reduce_packet)
# and first reduce_packet(rank 2 put data to rank 1, data put depends on first reduce_packet).
# In this case, we don't need to fuse the instructions.
with prgm:
c0 = chunk(0, Buffer.input, 0)
c0.put_packet(1, "scratch", 0, sendtb=0)
Expand Down

0 comments on commit 5060748

Please sign in to comment.