Skip to content

Adds HaloExchange Wrapper#25

Open
szaman19 wants to merge 8 commits intoLBANN:mainfrom
szaman19:put-api
Open

Adds HaloExchange Wrapper#25
szaman19 wants to merge 8 commits intoLBANN:mainfrom
szaman19:put-api

Conversation

@szaman19
Copy link
Contributor

Higher-level API for simplified GNN layer communication

  • Conforms with torch-geometric.MessagePassing layer API
  • Exposes distributed put functionality
  • Adds a global graph to local CommunicationPattern converter

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