fix diffuers pipelines c class ut (#2084)

Author: lvyufeng

mindnlp/core/_tensor.py

@@ -126,7 +126,7 @@ def __reduce_ex__(self, protocol):
     StubTensor.__reduce_ex__ = __reduce_ex__
 
     def to_(self, *args, **kwargs):
-        dtype_to = None
+        dtype_to = kwargs.get("dtype", None)
         if len(args) == 1:
             if isinstance(args[0], Type):
                 dtype_to = args[0]
@@ -675,6 +675,10 @@ def __contains__(self, item):
     Tensor.sub_ = ops.inplace_sub
     StubTensor.sub_ = ops.inplace_sub
 
+    Tensor.roll = ops.roll
+    StubTensor.roll = ops.roll
+
+
 def _rebuild_from_type_v2(func, new_type, args, state):
     ret = func(*args)
     return ret

mindnlp/core/fft/__init__.py

@@ -44,4 +44,10 @@ def ifftn(input, s=None, dim=None, norm=None, *, out=None):
 def ifftshift(input, dim=None):
     return ops.ifftshift(input, dim)
 
+def fft2(input, s=None, dim=(-2, -1), norm=None):
+    return ops.fft2(input, s, dim, norm)
+
+def ifft2(input, s=None, dim=(-2, -1), norm=None):
+    return ops.ifft2(input, s, dim, norm)
+
 __all__ = ['fft', 'fftn', 'irfft', 'rfft']

mindnlp/core/ops/array.py

@@ -273,7 +273,8 @@ def squeeze(input, *dim, **kwargs):
 
 # stack
 has_stack = hasattr(mindspore.mint, 'stack')
-def stack(tensors, dim=0, *, out=None):
+def stack(tensors, dim=0, *, out=None, **kwargs):
+    dim = kwargs.pop('axis', dim)
     if use_pyboost() and has_stack:
         return call_ms_func(mindspore.mint.stack, tensors, dim, out=out)
     return call_ms_func(ops.stack, tensors, dim, out=out)

mindnlp/core/ops/creation.py

@@ -139,8 +139,13 @@ def linspace(start, end, steps, *, dtype=None, **kwargs):
     return ops.linspace(start, end, steps).to(dtype)
 
 # logspace
-def logspace(start, end, steps, base=10.0, *, dtype=None):
-    return ops.logspace(start, end, steps, base, dtype=dtype)
+has_logspace = hasattr(mindspore.mint, 'logspace')
+def logspace(start, end, steps, base=10.0, *, dtype=None, **kwargs):
+    if dtype is None:
+        dtype = get_default_dtype()
+    if use_pyboost() and has_logspace:
+        return mindspore.mint.logspace(start, end, steps, base, dtype=dtype)
+    return ops.logspace(float(start), float(end), steps, int(base), dtype=dtype)
 
 # eye
 has_eye = hasattr(mindspore.mint, 'eye')