未验证 提交 25820216 编写于 作者: R Ruibiao Chen 提交者: GitHub

Throw error for NVCC lazy in 1F1B pipeline (#56725)

上级 0a15b0db
...@@ -242,6 +242,14 @@ class Engine: ...@@ -242,6 +242,14 @@ class Engine:
elif self._strategy.pipeline.enable: elif self._strategy.pipeline.enable:
self._acc_steps = self._strategy.pipeline.accumulate_steps self._acc_steps = self._strategy.pipeline.accumulate_steps
if (
self._strategy.pipeline.enable
and self._strategy.pipeline.schedule_mode == "1F1B"
):
assert (
os.getenv("CUDA_MODULE_LOADING") != "LAZY"
), "EXP_CUDA_MODULE_LOADING_LAZY not supported in 1F1B pipeline."
self.history = None self.history = None
paddle.framework.set_flags({'FLAGS_new_executor_sequential_run': 1}) paddle.framework.set_flags({'FLAGS_new_executor_sequential_run': 1})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册