Update sharding_optimizer.py (#53164)

fb5fab07 · 张春乔 · GitHub · 6a8d98e0 · fb5fab07
隐藏空白更改
内联并排

Showing with 0 addition and 3 deletion

python/paddle/distributed/fleet/meta_optimizers/sharding_optimizer.py ...e/distributed/fleet/meta_optimizers/sharding_optimizer.py +0 -3

未找到文件。
--- a/python/paddle/distributed/fleet/meta_optimizers/sharding_optimizer.py
+++ b/python/paddle/distributed/fleet/meta_optimizers/sharding_optimizer.py
@@ -755,9 +755,6 @@ class ShardingOptimizer(MetaOptimizerBase):
                sync=False,
            )

-        if core.is_compiled_with_custom_device('npu'):
-            return
-
        # GPU
        for pair in self.pipeline_pair:
            pair_key = pair[0] * 1000 + pair[1]