mindspore.parallel
This interface is ONLY used in Auto-parallel procedure.
- mindspore.parallel.get_algo_parameters(attr_key)[source]
Get algo parameter config attributes.
Note
Return value according to the attribute value.
- Parameters
attr_key (str) – The key of the attribute.
- Raises
ValueError – If context keyword is not recognized.
- mindspore.parallel.set_algo_parameters(**kwargs)[source]
Set algo parameter config.
Note
Attribute name is needed.
- Parameters
tensor_slice_align_enable (bool) – Whether checking tensor slice shape for MatMul. Default: False
tensor_slice_align_size (int) – The minimum tensor slice shape of MatMul, the value must be in [1, 1024]. Default: 16
fully_use_devices (bool) – Whether ONLY generating strategies that fully use all available devices. Default: True
elementwise_op_strategy_follow (bool) – Whether the elementwise operator have the same strategies as its subsequent operators. Default: False
- Raises
ValueError – If context keyword is not recognized.