diff --git a/torch_xla/distributed/spmd/xla_sharding.py b/torch_xla/distributed/spmd/xla_sharding.py index c010fd4c352..c75fe938e83 100644 --- a/torch_xla/distributed/spmd/xla_sharding.py +++ b/torch_xla/distributed/spmd/xla_sharding.py @@ -31,6 +31,7 @@ Specifically, it is a tuple of one or more device mesh axes that describes how to shard the input tensor. For example, the first dimension of the tensor is sharded across the axis/axes described in the first element of this tuple and so on. +TEST """