[Lang] Add @python_scope decorator for selected MatrixNdarray/VectorN…

…darray methods
taichi-dev · Aug 23, 2022 · 0e3567f · 0e3567f
1 parent 1b4006f
commit 0e3567f
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/taichi/lang/matrix.py b/python/taichi/lang/matrix.py
@@ -2053,16 +2053,19 @@ def from_numpy(self, arr):
         """
         self._ndarray_matrix_from_numpy(arr, self.layout, as_vector=1)
 
+    @python_scope
     def __deepcopy__(self, memo=None):
         ret_arr = VectorNdarray(self.n, self.dtype, self.shape, self.layout)
         ret_arr.copy_from(self)
         return ret_arr
 
+    @python_scope
     def _fill_by_kernel(self, val):
         from taichi._kernels import \
             fill_ndarray_matrix  # pylint: disable=C0415
         fill_ndarray_matrix(self, val)
 
+    @python_scope
     def __repr__(self):
         return f'<{self.n} {self.layout} ti.Vector.ndarray>'