Skip to content

Conversation

@JackCaoG
Copy link
Collaborator

90% of the time I just need a 1d mesh, it is a bit annoying to have to init device_ids etc..

@JackCaoG
Copy link
Collaborator Author

Also update comments in the spmd, I think all methods listed in docs/source/index.rst should follow the pytorch style comment.

Copy link
Collaborator

@alanwaketan alanwaketan left a comment

Choose a reason for hiding this comment

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

LGTM.

@JackCaoG JackCaoG merged commit ab41f78 into master Jun 26, 2024
@JackCaoG JackCaoG deleted the JackCaoG/1d_mesh branch June 26, 2024 19:52
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

2 participants