mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
[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:
parent
5b368fa0b7
commit
5e320eea66
|
|
@ -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 = [
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user