fix MS 2.7.1 caused error (#2230)

Author: lvyufeng

README.md

@@ -141,7 +141,8 @@ bash scripts/build_and_reinstall.sh
 | 0.2.x           | >=2.1.0           | >=3.8, <=3.9             |
 | 0.3.x           | >=2.1.0, <=2.3.1  | >=3.8, <=3.9             |
 | 0.4.x           | >=2.2.x, <=2.5.0  | >=3.9, <=3.11            |
-| 0.5.x           | >=2.5.0, <=2.7.0. | >=3.10, <=3.11           |
+| 0.5.0           | >=2.5.0, <=2.7.0  | >=3.10, <=3.11           |
+| 0.6.x           | >=2.7.1.            | >=3.10, <=3.11           |
 
 ## Introduction
 

mindtorch/_tensor.py

@@ -118,6 +118,7 @@ def __init__(self, *args, **kwargs):
 Tensor.__init__ = __init__
 origin_setitem = Tensor.__setitem__
 origin_is_contiguous = Tensor.is_contiguous
+origin_to = Tensor.to
 Tensor._requires_grad = False
 
 def tensor(data, *, dtype=None, device=None, requires_grad=False):
@@ -248,6 +249,8 @@ def __getitem__(self, slices):
 
         if self.device.type == 'meta':
             out = ops.getitem_np(self, slices)
+        elif self.device.type == 'cpu':
+            return ops.getitem(self, slices)
         else:
             out = ops.tensor_getitem(self, slices)
 
@@ -2251,7 +2254,7 @@ def _move_to(self, device, non_blocking=False):
             # self.data_sync(True)
             if self.device.type == 'cpu':
                 self.data_ptr()
-            data = Tensor.move_to(self, device_str, blocking=not non_blocking)
+            data = origin_to(self, device_str, non_blocking=non_blocking)
 
             out = Tensor(data)
             out._device = device

setup.py

@@ -170,7 +170,7 @@ def run(self):
     },
 
     install_requires=[
-        'mindspore>=2.5.0',
+        'mindspore>=2.7.1',
         'tqdm',
         'requests',
         'accelerate>=1.6.0', # hf dependency