fix transformers ut for 4.54.1 (#2117)

Author: lvyufeng

.gitignore

@@ -169,4 +169,5 @@ flagged/
 
 tests/diffusers/
 tests/transformers/
+tests/huggingface_transformers/
 .gradio/

mindnlp/core/_tensor.py

@@ -815,6 +815,15 @@ def record_stream(self, stream):
     Tensor.scatter = ops.scatter
     StubTensor.scatter = ops.scatter
 
+    Tensor.mul = ops.mul
+    StubTensor.mul = ops.mul
+
+    Tensor.index_select = ops.index_select
+    StubTensor.index_select = ops.index_select
+
+    Tensor.gather = ops.gather
+    StubTensor.gather = ops.gather
+
 def _rebuild_from_type_v2(func, new_type, args, state):
     ret = func(*args)
     return ret

mindnlp/core/npu/__init__.py

@@ -58,3 +58,9 @@ def __enter__(self):
 
     def __exit__(self, type: Any, value: Any, traceback: Any):
         return False
+
+def mem_get_info(index):
+    return (1024, 1024)
+
+def current_device():
+    return core.device('npu', 0)

mindnlp/core/ops/inplace.py

@@ -209,7 +209,7 @@ def inplace_random(self, from_=0, to=None, *, generator=None):
             generator = default_generator
         seed, offset = generator._step(  # pylint: disable=protected-access
             generator_step_)
-        return inplace_random_op(input, from_, to, seed, offset)
+        return inplace_random_op(self, from_, to, seed, offset)
     else:
         if isinstance(self.dtype, typing.Float):
             self.uniform_(from_, to, generator=generator)

mindnlp/core/types.py

@@ -38,7 +38,11 @@ def __init__(self, type=None, index=None):
                 _id = type.index
             elif isinstance(type, int):
                 _id = type
-                _target = DEVICE_MAP[mindspore.get_current_device().device_target]
+                try:
+                    device_target = mindspore.get_current_device().device_target
+                except:
+                    device_target = mindspore.get_context('device_target')
+                _target = DEVICE_MAP[device_target]
             else:
                 print(type)
                 raise TypeError("core.device(): `type` must be type of 'str' or 'core.device'.")