mindspore 2.0.0a0__cp38-cp38-win_amd64.whl → 2.0.0rc1__cp38-cp38-win_amd64.whl

mindspore/ops/operations/debug_ops.py

@@ -1,4 +1,4 @@
-# Copyright 2020-2021 Huawei Technologies Co., Ltd
+# Copyright 2020-2023 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.
@@ -15,27 +15,27 @@
 """debug_ops"""
 from types import FunctionType, MethodType
 
-from mindspore import context
 from mindspore import log as logger
 from mindspore._c_expression import security
 from mindspore._c_expression import Tensor as Tensor_
-from mindspore._checkparam import Validator as validator
-from mindspore._checkparam import Rel
+from mindspore import _checkparam as validator
 from mindspore.common import dtype as mstype
 from mindspore.common.parameter import Parameter
+from mindspore.common.tensor import Tensor
 from mindspore.ops.primitive import prim_attr_register, Primitive, PrimitiveWithInfer
 
 
-def _check_mode(class_name):
-    """Check for PyNative mode."""
-    mode = context.get_context('mode')
-    if mode == context.PYNATIVE_MODE:
-        raise RuntimeError(f"For '{class_name}', the operator does not support PyNative mode.")
+SUMMARY_TENSOR_CACHE = []
+
+
+def _cache_summary_data(op_name, define_name, tensor):
+    """Cache summary tensor data."""
+    global SUMMARY_TENSOR_CACHE
+    SUMMARY_TENSOR_CACHE.append([op_name, define_name, tensor])
 
 
 def _check_summary_param(name, value, class_name):
     """Checks the name and value is valid for summary."""
-    _check_mode(class_name)
     n_type = name['dtype']
     n_value = name['value']
     validator.check_value_type('name', n_type, [type(mstype.string)], class_name)
@@ -58,7 +58,7 @@ class ScalarSummary(Primitive):
     This operator will put a scalar to a summary file with protocol buffer format. It must be used with SummaryRecord
     or SummaryCollector, which specify the directory of the summary file. The summary file can
     be loaded and shown by MindInsight, see `MindInsight documents <https://www.mindspore.cn/
-    mindinsight/docs/en/r2.0.0-alpha/index.html>`_ for details.
+    mindinsight/docs/en/r2.0/index.html>`_ for details.
 
     Inputs:
         - **name** (str) - The name of the input variable, it must not be an empty string.
@@ -106,7 +106,7 @@ class ScalarSummary(Primitive):
         self.add_prim_attr("side_effect_io", True)
 
     def __call__(self, *args):
-        raise RuntimeError(f"PyNative not support Operator '{self.__class__.__name__}'")
+        _cache_summary_data(self.name, args[0], args[1])
 
 
 class ImageSummary(PrimitiveWithInfer):
@@ -114,7 +114,7 @@ class ImageSummary(PrimitiveWithInfer):
     This operator will put an image tensor to a summary file with protocol buffer format. It must be used with
     SummaryRecord or SummaryCollector, which specify the directory of the summary file. The summary file can
     be loaded and shown by MindInsight, see `MindInsight documents <https://www.mindspore.cn/
-    mindinsight/docs/en/r2.0.0-alpha/index.html>`_ for details.
+    mindinsight/docs/en/r2.0/index.html>`_ for details.
 
     Inputs:
         - **name** (str) - The name of the input variable, it must not be an empty string.
@@ -167,7 +167,7 @@ class ImageSummary(PrimitiveWithInfer):
         return SUMMARY_RETURN_VALUE
 
     def __call__(self, *args):
-        raise RuntimeError(f"PyNative not support Operator '{self.__class__.__name__}'")
+        _cache_summary_data(self.name, args[0], args[1])
 
 
 class TensorSummary(Primitive):
@@ -175,7 +175,7 @@ class TensorSummary(Primitive):
     This operator will put a tensor to a summary file with protocol buffer format. It must be used with SummaryRecord
     or SummaryCollector, which specify the directory of the summary file. The summary file can
     be loaded and shown by MindInsight, see `MindInsight documents <https://www.mindspore.cn/
-    mindinsight/docs/en/r2.0.0-alpha/index.html>`_ for details.
+    mindinsight/docs/en/r2.0/index.html>`_ for details.
 
     Inputs:
         - **name** (str) - The name of the input variable.
@@ -223,7 +223,7 @@ class TensorSummary(Primitive):
         self.add_prim_attr("side_effect_io", True)
 
     def __call__(self, *args):
-        raise RuntimeError(f"PyNative not support Operator '{self.__class__.__name__}'")
+        _cache_summary_data(self.name, args[0], args[1])
 
 
 class HistogramSummary(PrimitiveWithInfer):
@@ -231,7 +231,7 @@ class HistogramSummary(PrimitiveWithInfer):
     This operator will calculate the histogram of a tensor and put it to a summary file with protocol buffer format.
     It must be used with SummaryRecord or SummaryCollector, which specify the directory of the summary file.
     The summary file can be loaded and shown by MindInsight, see `MindInsight documents <https://www.mindspore.cn/
-    mindinsight/docs/en/r2.0.0-alpha/index.html>`_ for details.
+    mindinsight/docs/en/r2.0/index.html>`_ for details.
 
     Inputs:
         - **name** (str) - The name of the input variable.
@@ -289,7 +289,7 @@ class HistogramSummary(PrimitiveWithInfer):
         return SUMMARY_RETURN_VALUE
 
     def __call__(self, *args):
-        raise RuntimeError(f"PyNative not support Operator '{self.__class__.__name__}'")
+        _cache_summary_data(self.name, args[0], args[1])
 
 
 class InsertGradientOf(PrimitiveWithInfer):
@@ -454,7 +454,7 @@ class Print(Primitive):
     """
     Print the inputs to stdout.
 
-    Refer to :func:`mindspore.ops.print` for more detail.
+    Refer to :func:`mindspore.ops.print_` for more detail.
 
     Supported Platforms:
         ``Ascend`` ``GPU`` ``CPU``
@@ -475,11 +475,11 @@ class Print(Primitive):
         >>> result = net(x, y)
         Print Tensor x and Tensor y:
         Tensor(shape=[2, 1], dtype=Int32, value=
-        [[1]
+        [[1],
          [1]])
         Tensor(shape=[2, 2], dtype=Int32, value=
-        [[1 1]
-         [1 1]])
+        [[1, 1],
+         [1, 1]])
     """
 
     @prim_attr_register
@@ -494,7 +494,7 @@ class Print(Primitive):
         for arg in args:
             if isinstance(arg, Parameter):
                 print(Tensor_.__repr__(arg))
-            elif isinstance(arg, Tensor_):
+            elif isinstance(arg, (Tensor, Tensor_)):
                 print(arg.__repr__())
             else:
                 print(arg)
@@ -552,7 +552,7 @@ class Assert(PrimitiveWithInfer):
 
     def infer_shape(self, condition, inputs):
         condition_len = len(condition)
-        validator.check_int(condition_len, 1, Rel.LE, "condition's rank", self.name)
+        validator.check_int(condition_len, 1, validator.LE, "condition's rank", self.name)
         if condition_len == 1:
             validator.check_equal_int(condition[0], 1, "condition[0]", self.name)
         return [1]