Browse Source

!954 Add document for set_auto_parallel_context() that reset_auto_parallel_context() must be called when programs have multiple parallel modes

Merge pull request !954 from yihuaijie/master
tags/v0.3.0-alpha
mindspore-ci-bot Gitee 5 years ago
parent
commit
2c85295e18
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/context.py

+ 3
- 0
mindspore/context.py View File

@@ -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.


Loading…
Cancel
Save