mindspore 2.6.0rc1__cp311-cp311-win_amd64.whl → 2.7.0__cp311-cp311-win_amd64.whl

mindspore/mint/__init__.py

@@ -33,7 +33,7 @@ from mindspore.ops.function.array_func import zeros_like_ext as zeros_like
 from mindspore.ops.function.array_func import unique_ext as unique
 from mindspore.ops.function.array_func import chunk_ext as chunk
 from mindspore.ops.functional_overload import empty
-from mindspore.ops.function.array_func import empty_like
+from mindspore.ops.functional_overload import empty_like
 from mindspore.ops.function.math_func import isclose
 from mindspore.ops.auto_generate import abs
 from mindspore.ops.auto_generate import clone
@@ -63,7 +63,7 @@ from mindspore.ops.auto_generate import stack_ext as stack
 # 7
 from mindspore.ops.function.array_func import unsqueeze
 # 8
-from mindspore.ops.auto_generate import transpose_ext as transpose
+from mindspore.ops.auto_generate import transpose_ext_view as transpose
 from mindspore.ops.auto_generate import batch_norm_elemt
 from mindspore.ops.auto_generate import batch_norm_gather_stats_with_counts
 from mindspore.ops.auto_generate import batch_norm_stats
@@ -291,7 +291,7 @@ from mindspore.ops.function.nn_func import softmax_ext
 # 118
 
 # 119
-
+from mindspore.ops.functional_overload import any
 # 120
 from mindspore.ops.auto_generate import isneginf_ext as isneginf
 # 121
@@ -364,7 +364,7 @@ from mindspore.ops.function.array_func import scatter_add_ext as scatter_add
 # 289
 from mindspore.ops.auto_generate import sign
 
-from mindspore.ops.auto_generate import select_ext as select
+from mindspore.ops.auto_generate import select_ext_view as select
 
 # 301
 from mindspore.ops.function.math_func import tan
@@ -478,7 +478,7 @@ from mindspore.ops.auto_generate import log2_ext as log2
 from mindspore.ops.function.math_func import isnan_ext as isnan
 
 # 916
-from mindspore.ops.auto_generate import index_add_ext as index_add
+from mindspore.ops.functional_overload import index_add
 
 # 1007
 from mindspore.ops.auto_generate import t_ext as t
@@ -490,46 +490,6 @@ from mindspore.ops.auto_generate.gen_ops_prim import equal_ext_op
 from mindspore.ops.function.array_func import unbind_ext as unbind
 
 
-def any(input, dim=None, keepdim=False):
-    r"""
-    Tests if any element in `input` evaluates to `True` along the given axes.
-
-    Args:
-        input (Tensor): The input tensor.
-        dim (Union[int, tuple(int), list(int), Tensor], optional): The dimensions to reduce. If ``None`` ,
-                all dimensions are reduced. Default ``None`` .
-        keepdim (bool, optional): Whether the output tensor has dim retained or not. Default ``False`` .
-
-    Returns:
-        Tensor
-
-    Supported Platforms:
-        ``Ascend`` ``GPU`` ``CPU``
-
-    Examples:
-        >>> import mindspore
-        >>> input = mindspore.tensor([[True, False], [True, True]])
-        >>>
-        >>> # case 1:  By default, mindspore.mint.any tests along all the axes.
-        >>> mindspore.mint.any(input)
-        Tensor(shape=[], dtype=Bool, value= True)
-        >>>
-        >>> # case 2: Reduces a dimension along dim 1, with keepdim False.
-        >>> mindspore.mint.any(input, dim=1)
-        Tensor(shape=[2], dtype=Bool, value= [ True,  True])
-        >>>
-        >>> # case 3: Reduces a dimension along dim (0, 1), with keepdim False.
-        >>> mindspore.mint.any(input, dim=(0,1))
-        Tensor(shape=[], dtype=Bool, value= True)
-        >>>
-        >>> # case 4: Reduces a dimension along dim [0, 1], with keepdim True.
-        >>> mindspore.mint.any(input, dim=[0,1], keepdim=True)
-        Tensor(shape=[1, 1], dtype=Bool, value=
-        [[ True]])
-    """
-    return ops.functional.any(input, dim, keepdim)
-
-
 def all(input, dim=None, keepdim=False):
     r"""
     all(input) -> Tensor
@@ -717,7 +677,7 @@ def cummax(input, dim):
         \end{array}
 
     .. note::
-        O2 mode is not supported in Ascend.
+        GE backend is not supported in Ascend.
 
     Args:
         input (Tensor): The input Tensor. Rank of `input` must be greater than 0.

mindspore/mint/distributed/__init__.py

@@ -24,6 +24,7 @@ Please see the `msrun start up
 for more details.
 """
 from __future__ import absolute_import
+from mindspore.mint.distributed.distributed import TCPStore
 from mindspore.mint.distributed.distributed import (
     init_process_group,
     destroy_process_group,
@@ -36,9 +37,11 @@ from mindspore.mint.distributed.distributed import (
     get_group_rank,
     all_reduce,
     all_gather_into_tensor,
+    all_gather_into_tensor_uneven,
     all_to_all,
     all_to_all_single,
     reduce_scatter_tensor,
+    reduce_scatter_tensor_uneven,
     isend,
     irecv,
     send,
@@ -72,9 +75,11 @@ __all__ = [
     "get_group_rank",
     "all_reduce",
     "all_gather_into_tensor",
+    "all_gather_into_tensor_uneven",
     "all_to_all",
     "all_to_all_single",
     "reduce_scatter_tensor",
+    "reduce_scatter_tensor_uneven",
     "isend",
     "irecv",
     "send",