gstaichi 0.1.20.dev0__cp310-cp310-macosx_15_0_arm64.whl → 0.1.25.dev0__cp310-cp310-macosx_15_0_arm64.whl

{taichi → gstaichi}/lang/kernel_arguments.py

@@ -2,17 +2,17 @@
 
 import inspect
 
-import taichi.lang
-from taichi._lib import core as _ti_core
-from taichi.lang import impl, ops
-from taichi.lang._texture import RWTextureAccessor, TextureSampler
-from taichi.lang.any_array import AnyArray
-from taichi.lang.expr import Expr
-from taichi.lang.matrix import MatrixType
-from taichi.lang.struct import StructType
-from taichi.lang.util import cook_dtype
-from taichi.types.compound_types import CompoundType
-from taichi.types.primitive_types import RefType, u64
+import gstaichi.lang
+from gstaichi._lib import core as _ti_core
+from gstaichi.lang import impl, ops
+from gstaichi.lang._texture import RWTextureAccessor, TextureSampler
+from gstaichi.lang.any_array import AnyArray
+from gstaichi.lang.expr import Expr
+from gstaichi.lang.matrix import MatrixType
+from gstaichi.lang.struct import StructType
+from gstaichi.lang.util import cook_dtype
+from gstaichi.types.compound_types import CompoundType
+from gstaichi.types.primitive_types import RefType, u64
 
 
 class KernelArgument:
@@ -32,9 +32,9 @@ class SparseMatrixEntry:
     def _augassign(self, value, op):
         call_func = f"insert_triplet_{self.dtype}"
         if op == "Add":
-            taichi.lang.impl.call_internal(call_func, self.ptr, self.i, self.j, ops.cast(value, self.dtype))
+            gstaichi.lang.impl.call_internal(call_func, self.ptr, self.i, self.j, ops.cast(value, self.dtype))
         elif op == "Sub":
-            taichi.lang.impl.call_internal(call_func, self.ptr, self.i, self.j, -ops.cast(value, self.dtype))
+            gstaichi.lang.impl.call_internal(call_func, self.ptr, self.i, self.j, -ops.cast(value, self.dtype))
         else:
             assert False, "Only operations '+=' and '-=' are supported on sparse matrices."
 
@@ -93,7 +93,7 @@ def decl_matrix_arg(matrixtype, name, arg_depth):
     arg_load = Expr(
         _ti_core.make_arg_load_expr(arg_id, arg_type, create_load=False, arg_depth=arg_depth, dbg_info=argload_di)
     )
-    return matrixtype.from_taichi_object(arg_load)
+    return matrixtype.from_gstaichi_object(arg_load)
 
 
 def decl_struct_arg(structtype, name, arg_depth):
@@ -103,7 +103,7 @@ def decl_struct_arg(structtype, name, arg_depth):
     arg_load = Expr(
         _ti_core.make_arg_load_expr(arg_id, arg_type, create_load=False, arg_depth=arg_depth, dbg_info=argload_di)
     )
-    return structtype.from_taichi_object(arg_load)
+    return structtype.from_gstaichi_object(arg_load)
 
 
 def push_argpack_arg(name):
@@ -112,7 +112,7 @@ def push_argpack_arg(name):
 
 def decl_argpack_arg(argpacktype, member_dict):
     impl.get_runtime().compiling_callable.pop_argpack_stack()
-    return argpacktype.from_taichi_object(member_dict)
+    return argpacktype.from_gstaichi_object(member_dict)
 
 
 def decl_sparse_matrix(dtype, name):