Skip to content

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

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

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