Skip to content
This repository has been archived by the owner on Oct 19, 2024. It is now read-only.

[DataLoader] Use PlacementSpecs in data loader #581

Merged
merged 2 commits into from
Jul 3, 2022
Merged

[DataLoader] Use PlacementSpecs in data loader #581

merged 2 commits into from
Jul 3, 2022

Conversation

merrymercy
Copy link
Member

@merrymercy merrymercy commented Jul 3, 2022 •
edited
Loading

  • Use the more generate PlacementSpecs instead of ShardingSpec in data loaders.
  • Add get_output_placement_specs in mesh executables.

@merrymercy merrymercy changed the title [MINOR] Add output placement specs [DataLoader] Use PlacementSpecs in data loader Jul 3, 2022
@merrymercy merrymercy merged commit 4a68171 into main Jul 3, 2022
@merrymercy merrymercy deleted the pr-debug branch July 3, 2022 11:44
@merrymercy merrymercy mentioned this pull request Jul 4, 2022
Closed
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
1 participant
@merrymercy