From 12af12e405d2ee93a4b81f8b999e59edfdb3955b Mon Sep 17 00:00:00 2001 From: Yi Huaijie Date: Wed, 6 May 2020 21:48:23 +0800 Subject: [PATCH] Add document for setting automatic parallel mode that if a program has multiple tasks with different parallel modes should call reset_auto_parallel_context() to reset config. --- mindspore/context.py | 3 +++ 1 file changed, 3 insertions(+) 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.