xla

Implement mark_sharding as a custom op to support dynamo spmd activation sharding

#5712

Merged