Skip to content
GitLab
Unverified Commit 1a8fb887 authored by Boian Petkantchin's avatar Boian Petkantchin Committed by GitHub
Browse files

[mlir][mesh] Add resharding spmdization on a 1D device mesh (#76179) 

The current implementation supports only sharding of tensor axes that
have size divisible by the mesh axis size.
parent 7122f55c
Expand all Hide whitespace changes
Inline Side-by-side
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment