diff --git a/mindspore/context.py b/mindspore/context.py index 74acd7cd01..632078ebd4 100644 --- a/mindspore/context.py +++ b/mindspore/context.py @@ -403,6 +403,9 @@ def set_auto_parallel_context(**kwargs): Note: Attribute name is required for setting attributes. + If a program has tasks with different parallel modes, then before setting new parallel mode for + next task, interface mindspore.context.reset_auto_parallel_context() needs to be called to reset + the configuration. Args: device_num (int): Available device number, the value must be in [1, 4096]. Default: 1.