Skip to content

Commit

Permalink
[jax:custom_partitioning] Make SdyShardingRule a user facing class.
Browse files Browse the repository at this point in the history 
Move the parsing of a sharding rule string to a free function
str_to_sdy_sharding_rule. Move the building of the MLIR sharding rule to a free
function sdy_sharding_rule_to_mlir.

PiperOrigin-RevId: 704818640
  • Loading branch information
@bixia1 @Google-ML-Automation
bixia1 authored and Google-ML-Automation committed Dec 10, 2024
1 parent 9c32fe8 commit d4899f7
Show file tree
Hide file tree
Showing 2 changed files with 437 additions and 310 deletions.
Loading

0 comments on commit d4899f7

Please sign in to comment.