xla

1f88b80f - Restrict the mark_sharding api to devicedata