[lang] Hide dtype and needs_grad from SNode (#4308)

taichi-dev · Feb 17, 2022 · be74dbd · be74dbd
1 parent 063d630
commit be74dbd
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 12 deletions.
diff --git a/python/taichi/lang/field.py b/python/taichi/lang/field.py
@@ -44,7 +44,7 @@ def dtype(self):
         Returns:
             DataType: Data type of each individual value.
         """
-        return self.snode.dtype
+        return self.snode._dtype
 
     @property
     def name(self):

diff --git a/python/taichi/lang/matrix.py b/python/taichi/lang/matrix.py
@@ -1124,7 +1124,7 @@ def calc_dynamic_index_stride(self):
         length = len(paths[0])
         if any(
                 len(path) != length or ti_core.is_custom_type(path[length -
-                                                                   1].dtype)
+                                                                   1]._dtype)
                 for path in paths):
             return
         for i in range(length):

diff --git a/python/taichi/lang/snode.py b/python/taichi/lang/snode.py
@@ -201,7 +201,7 @@ def _path_from_root(self):
         return res
 
     @property
-    def dtype(self):
+    def _dtype(self):
         """Gets the data type of `self`.
 
         Returns:
@@ -255,15 +255,6 @@ def snode(self):
         """
         return self
 
-    @property
-    def needs_grad(self):
-        """Checks whether `self` has a corresponding gradient :class:`~taichi.lang.SNode`.
-
-        Returns:
-            bool: Whether `self` has a corresponding gradient :class:`~taichi.lang.SNode`.
-        """
-        return self.ptr.has_grad()
-
     def _get_children(self):
         """Gets all children components of `self`.