Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merge mscclpp-lang to mscclpp project #442

Open
wants to merge 23 commits into
base: main
Choose a base branch
from
Open

Conversation

Binyang2014
Copy link
Contributor

@Binyang2014 Binyang2014 commented Jan 7, 2025

First step to merge msccl-tools into mscclpp repo. In this step will move all msccl related code, pass the current tests and do some necessary refactor.

Add mscclpp.language module
Add _InstructionOptimizer and DagOptimizer class to optimize the dag
Add DagLower to lower dag to intermediate representation
Remove msccl related code

@Binyang2014 Binyang2014 changed the title Merge msccl-tools Merge mscclpp-lang to mscclpp project Jan 8, 2025
@Binyang2014 Binyang2014 marked this pull request as ready for review January 8, 2025 02:20
@Binyang2014 Binyang2014 requested a review from Copilot January 8, 2025 02:20
Copy link

@Copilot Copilot AI left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot reviewed 10 out of 25 changed files in this pull request and generated 2 comments.

Files not reviewed (15)
  • python/mscclpp/language/init.py: Evaluated as low risk
  • python/mscclpp/language/dag/init.py: Evaluated as low risk
  • .azure-pipelines/nccl-api-test.yaml: Evaluated as low risk
  • python/mscclpp/language/dag/lower.py: Evaluated as low risk
  • python/mscclpp/language/types.py: Evaluated as low risk
  • python/mscclpp/language/dag/instruction_dag.py: Evaluated as low risk
  • python/mscclpp/language/utils.py: Evaluated as low risk
  • python/examples/allreduce_allpairs_get.py: Evaluated as low risk
  • python/examples/allreduce_allpairs_packet.py: Evaluated as low risk
  • python/examples/allreduce_allpairs.py: Evaluated as low risk
  • python/mscclpp/language/collectives.py: Evaluated as low risk
  • python/mscclpp/language/rank.py: Evaluated as low risk
  • python/mscclpp/language/buffer.py: Evaluated as low risk
  • python/examples/allreduce_nvls.py: Evaluated as low risk
  • .github/workflows/mscclpp-lang.yml: Evaluated as low risk
Comments suppressed due to low confidence (3)

python/examples/send_recv_packet.py:28

  • The parameters temp_buffer and temp_buffer_index are not used in the send_recv function. This might be an oversight or an incomplete implementation.
c.put_packet(

python/examples/send_recv_packet.py:11

  • [nitpick] The function name send_recv is ambiguous as it is the same as in the other example. Consider renaming it to send_recv_packet to avoid confusion.
def send_recv(instances):

python/examples/send_recv_packet.py:0

  • Ensure that the new behavior introduced by put_packet and copy_packet is covered by tests.
def send_recv(instances):

chunks = [self, chunk]
return ReduceChunk(dst, chunks)
else:
assert True, "Trying to reduce with chunk of None"
Copy link
Preview

Copilot AI Jan 8, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The assertion should be corrected to assert chunk is not None, "Trying to reduce with chunk of None" to properly check for None.

Suggested change
assert True, "Trying to reduce with chunk of None"
assert chunk is not None, "Trying to reduce with chunk of None"

Copilot is powered by AI, so mistakes are possible. Review output carefully before use.

Positive Feedback
Negative Feedback

Provide additional feedback

Please help us improve GitHub Copilot by sharing more details about this comment.

Please select one or more of the options
elif type(chunk) is Chunk:
chunks = self.chunks + [chunk]
else:
assert True, "Trying to reduce with chunk of None"
Copy link
Preview

Copilot AI Jan 8, 2025

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

The assertion should be corrected to assert chunk is not None, "Trying to reduce with chunk of None" to properly check for None.

Suggested change
assert True, "Trying to reduce with chunk of None"
assert chunk is not None, "Trying to reduce with chunk of None"

Copilot is powered by AI, so mistakes are possible. Review output carefully before use.

Positive Feedback
Negative Feedback

Provide additional feedback

Please help us improve GitHub Copilot by sharing more details about this comment.

Please select one or more of the options
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant