[BE] follow autoformating and linter (#151507)

Pull Request resolved: https://github.com/pytorch/pytorch/pull/151507
Approved by: https://github.com/Skylion007
This commit is contained in:
Xilun Wu 2025-04-23 23:26:17 -07:00 committed by PyTorch MergeBot
parent 5b368fa0b7
commit 5e320eea66

View File

@ -560,9 +560,11 @@ def propagate_shape_and_sharding(
shard_dim_map[in_dim.input_dim] = dim
input_tgt_placements = [
Replicate()
if isinstance(p, Shard) and not shardable_dims[p.dim][mesh_dim]
else p
(
Replicate()
if isinstance(p, Shard) and not shardable_dims[p.dim][mesh_dim]
else p
)
for mesh_dim, p in enumerate(input_src_placements)
]
output_placements = [