[misc] Add needs_grad property for SNode (#2500)

* add needs_grad property * Auto Format Co-authored-by: Taichi Gardener <[email protected]>
taichi-dev · Jul 7, 2021 · dedd976 · dedd976
1 parent 0496b3a
commit dedd976
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/python/taichi/lang/snode.py b/python/taichi/lang/snode.py
@@ -148,6 +148,10 @@ def __call__(self):  # TODO: remove this after v0.7.0
     def snode(self):
         return self
 
+    @property
+    def needs_grad(self):
+        return self.ptr.has_grad()
+
     def get_children(self):
         children = []
         for i in range(self.ptr.get_num_ch()):