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

Various ncclize improvements #6

Merged
merged 19 commits into from
Oct 5, 2021
Merged

Conversation

olsaarik
Copy link
Contributor

This includes a collection of work on improving ncclize, including:

  • Remove --old-format and --use-scratch
  • Add an experimental (off-by-default) greedy scratch sorting
  • Remove the ill advised MaxConcurrency channel policy (low number of threadblocks is important apparently)
  • Make MatchTopology channel policy split sends if necessary to balance bandwidth between channels
  • Remove old <copy/> tags that were never implemented in SCCL-RT in favor of new cpy operations.

Now sets of addresses that result in larger counts when contiguosly
allocated are greedily allocated first.
This fixes a problem where nop ops with no dependencies were created.
Any addresses that are both input and output are now copied with the new
instruction before the main algorithm starts.
The greedy scratch allocation algorithm will also insert copies to make
input/output addresses contiguous in scratch.
Also improve greedy scratch sorting to work better with this.
Also now postcopies are preferred, unless doing a precopy allows
maintaining contiguity.
This was breaking extra memory constraints.
Now contiguous sends are split up as necessary to match concurrency.
Also remove MaxConcurrency channel policy.
The old format has become entrenched.
New option is --no-scratch
Not added with a sledgehammer anymore, hooks into the same dependency
tracking all other operations use.
@olsaarik olsaarik force-pushed the feature/ncclize-improvements branch from 90da877 to 8406fe9 Compare September 30, 2021 18:28
@olsaarik olsaarik merged commit 695bbd7 into main Oct 5, 2021
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