|
|
|
|
@ -151,8 +151,8 @@ class Optimizer(Cell):
|
|
|
|
|
if context.get_auto_parallel_context("enable_parallel_optimizer"):
|
|
|
|
|
if _get_parallel_mode() == ParallelMode.DATA_PARALLEL:
|
|
|
|
|
self.use_parallel = True
|
|
|
|
|
elif _get_parallel_mode() == ParallelMode.STAND_ALONE:
|
|
|
|
|
raise RuntimeError("Parallel optimizer is not supported in stand alone mode.")
|
|
|
|
|
elif _get_parallel_mode() in (ParallelMode.STAND_ALONE, ParallelMode.HYBRID_PARALLEL):
|
|
|
|
|
raise RuntimeError("Parallel optimizer is not supported in {}.".format(_get_parallel_mode()))
|
|
|
|
|
else:
|
|
|
|
|
self.use_parallel = False
|
|
|
|
|
else:
|
|
|
|
|
|