fix diffusers pipelines k class ut (#2087)

Author: lvyufeng

mindnlp/core/_tensor.py

@@ -684,6 +684,9 @@ def __contains__(self, item):
     Tensor.scatter_reduce = ops.scatter_reduce
     StubTensor.scatter_reduce = ops.scatter_reduce
 
+    Tensor.tril_ = ops.inplace_tril
+    StubTensor.tril_ = ops.inplace_tril
+
 def _rebuild_from_type_v2(func, new_type, args, state):
     ret = func(*args)
     return ret

mindnlp/core/nn/modules/module.py

@@ -2002,6 +2002,24 @@ def bfloat16(self: T) -> T:
         """
         return self._apply(lambda t: t.bfloat16() if t.is_floating_point() else t)
 
+    def to_empty(
+        self, *, device, recurse: bool = True
+    ):
+        r"""Move the parameters and buffers to the specified device without copying storage.
+
+        Args:
+            device (:class:`torch.device`): The desired device of the parameters
+                and buffers in this module.
+            recurse (bool): Whether parameters and buffers of submodules should
+                be recursively moved to the specified device.
+
+        Returns:
+            Module: self
+        """
+        return self._apply(
+            lambda t: core.empty_like(t, device=device), recurse=recurse
+        )
+
     def float(self: T) -> T:
         r"""Casts all floating point parameters and buffers to ``float`` datatype.
 

mindnlp/core/ops/array.py

@@ -10,6 +10,7 @@
 from ..configs import use_pyboost, ON_ORANGE_PI
 from .other import broadcast_tensors, finfo
 from ._inner import call_ms_func
+from mindnlp import core
 
 # adjoint
 
@@ -221,15 +222,11 @@ def scatter_add(input, dim, index, src):
         return mindspore.mint.scatter_add(input, dim, index, src)
     return ops.tensor_scatter_elements(input, index, src, dim, 'add')
 
-scatter_reduce_dict = {
-    'sum': 'add',
-    'amax': 'max',
-    'amin': 'min',
-    'mean': 'mean'
-}
-# scatter_reduce
 def scatter_reduce(input, dim, index, src, reduce, *, include_self=True):
-    return inplace_scatter_src_reduce_op(input.clone(), dim, index, src, scatter_reduce_dict[reduce])
+    if reduce == 'sum':
+        return scatter_add(input, dim, index, src)
+    else:
+        raise ValueError(f'do not support reduce: {reduce}')
 
 # scatter_nd_update
 def scatter_nd_update(input, indices, update):

mindnlp/core/ops/inplace.py

@@ -185,6 +185,10 @@ def inplace_bernoulli(self, p=0.5, *, generator=None):
     self.data = core.bernoulli(self, generator=generator, p=p)
     return self
 
+def inplace_tril(self, diagonal=0):
+    self.data = core.tril(self, diagonal)
+    return self
+
 __all__ = [
     'inplace_copy',
     'inplace_zero',
@@ -207,5 +211,6 @@ def inplace_bernoulli(self, p=0.5, *, generator=None):
     'inplace_neg',
     'inplace_exp',
     'inplace_sub',
-    'inplace_bernoulli'
+    'inplace_bernoulli',
+    'inplace_tril'
 ]