mindspore 2.4.10__cp311-cp311-win_amd64.whl → 2.5.0__cp311-cp311-win_amd64.whl

mindspore/common/_stub_tensor.py

@@ -21,6 +21,7 @@ from mindspore.common.dtype import type_size_in_bytes
 from mindspore._c_expression import TensorNode, SequenceNode, NoneTypeNode, AnyTypeNode
 from mindspore._c_expression import Tensor as Tensor_
 from mindspore.common.api import _convert_python_data
+from mindspore.common._tensor_cpp_method import tensor_cpp_methods
 
 
 def _stub_member(var, init):
@@ -143,14 +144,33 @@ class StubTensor:
         """
         return self.ndim
 
+
+    def is_contiguous(self):
+        if self.stub:
+            return self.stub.get_value().is_contiguous()
+        return self.tensor.is_contiguous()
+
+    def set_cast_dtype(self):
+        if self.stub:
+            return self.stub.get_value().set_cast_dtype()
+        return self.tensor.set_cast_dtype()
+
+    def storage_offset(self):
+        if self.stub:
+            return self.stub.get_value().storage_offset()
+        return self.tensor.storage_offset()
+
+    def _need_contiguous(self):
+        if self.stub:
+            return self.stub.get_value()._need_contiguous()  # pylint: disable=protected-access
+        return self.tensor._need_contiguous()  # pylint: disable=protected-access
+
     asnumpy = _stub_method(Tensor.asnumpy)
     is_persistent_data = _stub_method(Tensor.is_persistent_data)
     asnumpy_of_slice_persistent_data = _stub_method(Tensor.asnumpy_of_slice_persistent_data)
     slice_num_of_persistent_data = _stub_method(Tensor.slice_num_of_persistent_data)
     slice_shape_of_persistent_data = _stub_method(Tensor.slice_shape_of_persistent_data)
     flush_from_cache = _stub_method(Tensor.flush_from_cache)
-    contiguous = _stub_method(Tensor.contiguous)
-    is_contiguous = _stub_method(Tensor.is_contiguous)
     register_hook = _stub_method(Tensor.register_hook)
 
     def stub_sync(self):
@@ -175,6 +195,7 @@ class StubTensor:
         self.stub = None
         self.tensor = Tensor(value, internal=True)
 
+no_stub_sync_cpp_api = ["set_cast_dtype", "storage_offset", "is_contiguous", "_need_contiguous"]
 
 def _init_stub_tensor_api():
     """adapt to python tensor and cpp tensor api"""
@@ -182,10 +203,10 @@ def _init_stub_tensor_api():
     cpp_tensor_func = dir(Tensor_)
     for attr in need_init_func:
         func = inspect.getattr_static(Tensor, attr)
-        if attr in cpp_tensor_func:
+        if attr in cpp_tensor_func and attr not in tensor_cpp_methods:
             # for cpp tensor api, we always need to sync for real tensor first
             setattr(StubTensor, attr, _stub_method(func))
-        else:
+        elif attr not in no_stub_sync_cpp_api:
             setattr(StubTensor, attr, func)
 
 

mindspore/common/_tensor_cpp_method.py

@@ -0,0 +1,17 @@
+# Copyright 2024 Huawei Technologies Co., Ltd
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+# ============================================================================
+"""Add tensor cpp methods for stub tensor"""
+
+tensor_cpp_methods = ['_to', 'abs', '__abs__', 'absolute', 'acos', 'arccos', 'acosh', 'arccosh', 'add', '__add__', 'add_', '__iadd__', 'addbmm', 'addmm', 'addmv', 'all', 'allclose', 'any', 'argmax', 'argmin', 'argsort', 'asin', 'arcsin', 'asinh', 'arcsinh', 'atan', 'arctan', 'atan2', 'arctan2', 'atanh', 'arctanh', 'bincount', 'bitwise_not', 'ceil', 'chunk', 'clamp', 'clip', 'clone', 'cos', 'cosh', 'cumsum', 'div', 'divide', 'div_', '__itruediv__', 'dot', 'eq', 'erf', 'erfc', 'exp', 'exp_', 'expand_as', 'expm1', 'fill_', 'flatten', 'floor', 'fmod', 'frac', 'gather', 'gcd', 'greater', 'gt', 'greater_equal', 'hardshrink', 'histc', 'index_select', 'inverse', 'isclose', 'isfinite', 'isinf', 'isneginf', 'lerp', 'less', 'lt', 'less_equal', 'le', 'log', 'log10', 'log1p', 'log2', 'logical_and', 'logical_not', 'logical_or', 'masked_fill', 'masked_fill_', 'masked_select', 'matmul', 'max', 'maximum', 'mean', 'median', 'min', 'minimum', 'mm', 'mul', 'mul_', '__imul__', 'nan_to_num', 'nansum', 'narrow', 'neg', 'negative', 'new_ones', 'new_zeros', 'not_equal', 'ne', 'outer', 'pow', '__pow__', 'prod', 'reciprocal', 'remainder', 'repeat_interleave', 'reshape', 'round', 'rsqrt', 'scatter', 'scatter_', 'scatter_add', 'select', 'sigmoid', 'sin', 'sinc', 'sinh', 'sort', 'split', 'sqrt', 'square', 'std', 'sub', '__sub__', 'sub_', '__isub__', 'subtract', 'sum', 't', 'tan', 'tanh', 'tile', 'topk', 'transpose', 'tril', 'triu', 'true_divide', 'trunc', 'type_as', 'unbind', 'unique', 'unsqueeze', 'var', 'view_as', 'where']