xla
Fix mark_sharding logic
#8578
Merged

Loading