gstaichi 0.1.23.dev0__cp310-cp310-win_amd64.whl → 0.1.25.dev0__cp310-cp310-win_amd64.whl

{taichi → gstaichi}/lang/field.py

@@ -1,11 +1,11 @@
 # type: ignore
 
-import taichi.lang
-from taichi._lib import core as _ti_core
-from taichi._logging import warn
-from taichi.lang import impl
-from taichi.lang.exception import TaichiSyntaxError
-from taichi.lang.util import (
+import gstaichi.lang
+from gstaichi._lib import core as _ti_core
+from gstaichi._logging import warn
+from gstaichi.lang import impl
+from gstaichi.lang.exception import GsTaichiSyntaxError
+from gstaichi.lang.util import (
     in_python_scope,
     python_scope,
     to_numpy_type,
@@ -15,7 +15,7 @@ from taichi.lang.util import (
 
 
 class Field:
-    """Taichi field class.
+    """GsTaichi field class.
 
     A field is constructed by a list of field members.
     For example, a scalar field has 1 field member, while a 3x3 matrix field has 9 field members.
@@ -48,7 +48,7 @@ class Field:
         Returns:
             SNode: Representative SNode (SNode of first field member).
         """
-        return taichi.lang.snode.SNode(self.vars[0].ptr.snode())
+        return gstaichi.lang.snode.SNode(self.vars[0].ptr.snode())
 
     @property
     def shape(self):
@@ -100,7 +100,7 @@ class Field:
         """Gets SNode of representative field member for loop range info.
 
         Returns:
-            taichi_python.SNode: SNode of representative (first) field member.
+            gstaichi_python.SNode: SNode of representative (first) field member.
         """
         return self.vars[0].ptr.snode()
 
@@ -214,7 +214,7 @@ class Field:
             raise TypeError("Cannot copy from a non-field object")
         if self.shape != other.shape:
             raise ValueError(f"ti.field shape {self.shape} does not match" f" the source field shape {other.shape}")
-        from taichi._kernels import tensor_to_tensor  # pylint: disable=C0415
+        from gstaichi._kernels import tensor_to_tensor  # pylint: disable=C0415
 
         tensor_to_tensor(self, other)
 
@@ -241,7 +241,7 @@ class Field:
         raise NotImplementedError()
 
     def __str__(self):
-        if taichi.lang.impl.inside_kernel():
+        if gstaichi.lang.impl.inside_kernel():
             return self.__repr__()  # make pybind11 happy, see Matrix.__str__
         if self._snode.ptr is None:
             return "<Field: Definition of this field is incomplete>"
@@ -261,18 +261,18 @@ class Field:
     def _initialize_host_accessors(self):
         if self.host_accessors:
             return
-        taichi.lang.impl.get_runtime().materialize()
+        gstaichi.lang.impl.get_runtime().materialize()
         self.host_accessors = [SNodeHostAccessor(e.ptr.snode()) for e in self.vars]
 
     def _host_access(self, key):
         return [SNodeHostAccess(e, key) for e in self.host_accessors]
 
     def __iter__(self):
-        raise NotImplementedError("Struct for is only available in Taichi scope.")
+        raise NotImplementedError("Struct for is only available in GsTaichi scope.")
 
 
 class ScalarField(Field):
-    """Taichi scalar field with SNode implementation.
+    """GsTaichi scalar field with SNode implementation.
 
     Args:
         var (Expr): Field member.
@@ -284,13 +284,15 @@ class ScalarField(Field):
     def fill(self, val):
         """Fills this scalar field with a specified value."""
         if in_python_scope():
-            from taichi._kernels import fill_field  # pylint: disable=C0415
+            from gstaichi._kernels import fill_field  # pylint: disable=C0415
 
             fill_field(self, val)
         else:
-            from taichi._funcs import field_fill_taichi_scope  # pylint: disable=C0415
+            from gstaichi._funcs import (  # pylint: disable=C0415
+                field_fill_gstaichi_scope,  # pylint: disable=C0415
+            )
 
-            field_fill_taichi_scope(self, val)
+            field_fill_gstaichi_scope(self, val)
 
     @python_scope
     def to_numpy(self, dtype=None):
@@ -304,10 +306,10 @@ class ScalarField(Field):
         import numpy as np  # pylint: disable=C0415
 
         arr = np.zeros(shape=self.shape, dtype=dtype)
-        from taichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
+        from gstaichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
 
         tensor_to_ext_arr(self, arr)
-        taichi.lang.runtime_ops.sync()
+        gstaichi.lang.runtime_ops.sync()
         return arr
 
     @python_scope
@@ -317,10 +319,10 @@ class ScalarField(Field):
 
         # pylint: disable=E1101
         arr = torch.zeros(size=self.shape, dtype=to_pytorch_type(self.dtype), device=device)
-        from taichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
+        from gstaichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
 
         tensor_to_ext_arr(self, arr)
-        taichi.lang.runtime_ops.sync()
+        gstaichi.lang.runtime_ops.sync()
         return arr
 
     @python_scope
@@ -331,10 +333,10 @@ class ScalarField(Field):
         # pylint: disable=E1101
         # paddle.empty() doesn't support argument `place``
         arr = paddle.to_tensor(paddle.zeros(self.shape, to_paddle_type(self.dtype)), place=place)
-        from taichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
+        from gstaichi._kernels import tensor_to_ext_arr  # pylint: disable=C0415
 
         tensor_to_ext_arr(self, arr)
-        taichi.lang.runtime_ops.sync()
+        gstaichi.lang.runtime_ops.sync()
         return arr
 
     @python_scope
@@ -344,10 +346,10 @@ class ScalarField(Field):
         for i, _ in enumerate(self.shape):
             if self.shape[i] != arr.shape[i]:
                 raise ValueError(f"ti.field shape {self.shape} does not match" f" the numpy array shape {arr.shape}")
-        from taichi._kernels import ext_arr_to_tensor  # pylint: disable=C0415
+        from gstaichi._kernels import ext_arr_to_tensor  # pylint: disable=C0415
 
         ext_arr_to_tensor(arr, self)
-        taichi.lang.runtime_ops.sync()
+        gstaichi.lang.runtime_ops.sync()
 
     @python_scope
     def from_numpy(self, arr):
@@ -430,7 +432,7 @@ class SNodeHostAccess:
 
 
 class BitpackedFields:
-    """Taichi bitpacked fields, where fields with quantized types are packed together.
+    """GsTaichi bitpacked fields, where fields with quantized types are packed together.
 
     Args:
         max_num_bits (int): Maximum number of bits all fields inside can occupy in total. Only 32 or 64 is allowed.
@@ -458,7 +460,7 @@ class BitpackedFields:
         if shared_exponent:
             self.bit_struct_type_builder.end_placing_shared_exponent()
             if count <= 1:
-                raise TaichiSyntaxError("At least 2 fields need to be placed when shared_exponent=True")
+                raise GsTaichiSyntaxError("At least 2 fields need to be placed when shared_exponent=True")
 
 
 __all__ = ["BitpackedFields", "Field", "ScalarField"]