gstaichi 2.2.0.dev1__cp310-cp310-win_amd64.whl → 2.2.1__cp310-cp310-win_amd64.whl

gstaichi/CHANGELOG.md

@@ -1,4 +1,4 @@
 Highlights:
 
 Full changelog:
-   - Merge branch 'hp/pin-rerunfailures' into hp/instrument-fe-ll-cache (by **Hugh Perkins**)
+   - [misc] Better error reporting (#167) (by **Hugh Perkins**)

gstaichi/_lib/core/gstaichi_python.pyi

@@ -512,7 +512,8 @@ class CompileResult:
     def compiled_kernel_data(self) -> CompiledKernelData:
         ...
 class CompiledKernelData:
-    pass
+    def _debug_dump_to_string(self) -> str:
+        ...
 class Config:
     pass
 class ConvType:

gstaichi/_version.py

@@ -1 +1 @@
-__version__ = '2.2.0.dev1'
+__version__ = '2.2.1'

gstaichi/lang/_fast_caching/args_hasher.py

@@ -43,6 +43,8 @@ def stringify_obj_type(path: tuple[str, ...], obj: object) -> str | None:
     String should somehow represent the type of obj. Doesnt have to be hashed, nor does it have
     to be the actual python type string, just a string that is representative of the type, and won't collide
     with different (allowed) types.
+
+    Note that fields are not included in fast cache.
     """
     arg_type = type(obj)
     if isinstance(obj, ScalarNdarray):
@@ -50,7 +52,10 @@ def stringify_obj_type(path: tuple[str, ...], obj: object) -> str | None:
     if isinstance(obj, VectorNdarray):
         return f"[ndv-{obj.n}-{obj.dtype}-{len(obj.shape)}]"
     if isinstance(obj, ScalarField):
-        return f"[f-{obj.snode._id}-{obj.dtype}-{obj.shape}]"
+        # disabled for now, because we need to think about how to handle field offset
+        # etc
+        # TODO: think about whether there is a way to include fields
+        return None
     if isinstance(obj, MatrixNdarray):
         return f"[ndm-{obj.m}-{obj.n}-{obj.dtype}-{len(obj.shape)}]"
     if "torch.Tensor" in str(arg_type):
@@ -58,7 +63,10 @@ def stringify_obj_type(path: tuple[str, ...], obj: object) -> str | None:
     if isinstance(obj, np.ndarray):
         return f"[np-{obj.dtype}-{obj.ndim}]"
     if isinstance(obj, MatrixField):
-        return f"[fm-{obj.m}-{obj.n}-{obj.snode._id}-{obj.dtype}-{obj.shape}]"
+        # disabled for now, because we need to think about how to handle field offset
+        # etc
+        # TODO: think about whether there is a way to include fields
+        return None
     if dataclasses.is_dataclass(obj):
         return dataclass_to_repr(path, obj)
     if is_data_oriented(obj):

gstaichi/lang/_ndarray.py

@@ -35,6 +35,20 @@ class Ndarray:
         self.arr = None
         self.layout = Layout.AOS
         self.grad: "TensorNdarray | None" = None
+        # we register with runtime, in order to enable reset to work later
+        impl.get_runtime().ndarrays.add(self)
+
+    def _reset(self):
+        """
+        Called by runtime, when we call ti.reset()
+        """
+        self.arr = None
+        self.grad = None
+        self.host_accessor = None
+        self.shape = None
+        self.element_type = None
+        self.dtype = None
+        self.layout = None
 
     def get_type(self):
         return NdarrayTypeMetadata(self.element_type, self.shape, self.grad is not None)

gstaichi/lang/ast/ast_transformers/call_transformer.py

@@ -177,7 +177,10 @@ class CallTransformer:
             if dataclasses.is_dataclass(val):
                 dataclass_type = val
                 for field in dataclasses.fields(dataclass_type):
-                    child_name = create_flat_name(arg.id, field.name)
+                    try:
+                        child_name = create_flat_name(arg.id, field.name)
+                    except Exception as e:
+                        raise RuntimeError(f"Exception whilst processing {field.name} in {type(dataclass_type)}") from e
                     load_ctx = ast.Load()
                     arg_node = ast.Name(
                         id=child_name,

gstaichi/lang/impl.py

@@ -1,6 +1,7 @@
 import numbers
+import weakref
 from types import FunctionType, MethodType
-from typing import Any, Iterable, Sequence
+from typing import TYPE_CHECKING, Any, Iterable, Sequence
 
 import numpy as np
 
@@ -56,6 +57,7 @@ from gstaichi.lang.util import (
     warning,
 )
 from gstaichi.types.enums import SNodeGradType
+from gstaichi.types.ndarray_type import NdarrayType
 from gstaichi.types.primitive_types import (
     all_types,
     f16,
@@ -68,6 +70,9 @@ from gstaichi.types.primitive_types import (
     u64,
 )
 
+if TYPE_CHECKING:
+    from gstaichi.lang._ndarray import Ndarray
+
 
 @gstaichi_scope
 def expr_init_shared_array(shape, element_type):
@@ -206,6 +211,10 @@ def subscript(ast_builder, value, *_indices, skip_reordered=False):
             SharedArray,
         ),
     ):
+        if isinstance(value, NdarrayType):
+            raise Exception(
+                "Cannot subscript NdarrayType. Did you access a global py dataclass inadvertently?", value, type(value)
+            )
         if len(_indices) == 1:
             _indices = _indices[0]
         return value.__getitem__(_indices)
@@ -347,8 +356,10 @@ class PyGsTaichi:
         self.fwd_mode_manager = None
         self.grad_replaced = False
         self.kernels: list[Kernel] = kernels or []
+        self.ndarrays: weakref.WeakSet[Ndarray] = weakref.WeakSet()
         self._signal_handler_registry = None
         self.unfinalized_fields_builder = {}
+        self.print_non_pure: bool = False
         self.short_circuit_operators: bool = False
         self.unrolling_limit: int = 0
         self.src_ll_cache: bool = True
@@ -538,9 +549,12 @@ def get_runtime() -> PyGsTaichi:
 
 def reset():
     global pygstaichi
+    old_ndarrays = pygstaichi.ndarrays
     old_kernels = pygstaichi.kernels
     pygstaichi.clear()
     pygstaichi = PyGsTaichi(old_kernels)
+    for nd in old_ndarrays:
+        nd._reset()
     for k in old_kernels:
         k.reset()
     _ti_core.reset_default_compile_config()

gstaichi/lang/kernel_impl.py

@@ -335,8 +335,15 @@ def _process_args(self: "Func | Kernel", is_func: bool, args: tuple[Any, ...], k
     if len_args > len(fused_args):
         arg_str = ", ".join(map(str, args))
         expected_str = ", ".join(f"{arg.name} : {arg.annotation}" for arg in self.arg_metas)
-        msg = f"Too many arguments. Expected ({expected_str}), got ({arg_str})."
-        raise GsTaichiSyntaxError(msg)
+        msg_l = []
+        msg_l.append(f"Too many arguments. Expected ({expected_str}), got ({arg_str}).")
+        for i in range(len_args):
+            if i < len(self.arg_metas):
+                msg_l.append(f" - {i} arg meta: {self.arg_metas[i].name} arg type: {type(args[i])}")
+            else:
+                msg_l.append(f" - {i} arg meta: <out of arg metas> arg type: {type(args[i])}")
+        msg_l.append(f"In function: {self.func}")
+        raise GsTaichiSyntaxError("\n".join(msg_l))
 
     for i, arg in enumerate(args):
         fused_args[i] = arg
@@ -351,14 +358,28 @@ def _process_args(self: "Func | Kernel", is_func: bool, args: tuple[Any, ...], k
         else:
             raise GsTaichiSyntaxError(f"Unexpected argument '{key}'.")
 
+    missing_parameters = []
     for i, arg in enumerate(fused_args):
         if arg is inspect.Parameter.empty:
             if self.arg_metas[i].annotation is inspect._empty:
-                raise GsTaichiSyntaxError(f"Parameter `{self.arg_metas[i].name}` missing.")
+                missing_parameters.append(f"Parameter `{self.arg_metas[i].name}` missing.")
             else:
-                raise GsTaichiSyntaxError(
+                missing_parameters.append(
                     f"Parameter `{self.arg_metas[i].name} : {self.arg_metas[i].annotation}` missing."
                 )
+    if len(missing_parameters) > 0:
+        msg_l = []
+        msg_l.append("Error: missing parameters.")
+        msg_l.extend(missing_parameters)
+        msg_l.append("")
+        msg_l.append("Debug info follows.")
+        msg_l.append("fused args:")
+        for i, arg in enumerate(fused_args):
+            msg_l.append(f"  {i} {arg}")
+        msg_l.append("arg metas:")
+        for i, arg in enumerate(self.arg_metas):
+            msg_l.append(f"  {i} {arg}")
+        raise GsTaichiSyntaxError("\n".join(msg_l))
 
     return tuple(fused_args)
 
@@ -615,6 +636,7 @@ class Kernel:
         self.visited_functions: set[FunctionSourceInfo] = set()
         self.kernel_function_info: FunctionSourceInfo | None = None
         self.compiled_kernel_data_by_key: dict[CompiledKernelKeyType, CompiledKernelData] = {}
+        self._last_compiled_kernel_data: CompiledKernelData | None = None  # for dev/debug
 
         self.src_ll_cache_observations: SrcLlCacheObservations = SrcLlCacheObservations()
         self.fe_ll_cache_observations: FeLlCacheObservations = FeLlCacheObservations()
@@ -720,6 +742,13 @@ class Kernel:
                 )
                 if self.compiled_kernel_data_by_key[key]:
                     self.src_ll_cache_observations.cache_loaded = True
+        elif self.gstaichi_callable and not self.gstaichi_callable.is_pure and self.runtime.print_non_pure:
+            # The bit in caps should not be modified without updating corresponding test
+            # freetext can be freely modified.
+            # As for why we are using `print` rather than eg logger.info, it is because
+            # this is only printed when ti.init(print_non_pure=..) is True. And it is
+            # confusing to set that to True, and see nothing printed.
+            print(f"[NOT_PURE] Debug information: not pure: {self.func.__name__}")
 
         kernel_name = f"{self.func.__name__}_c{self.kernel_counter}_{key[1]}"
         _logging.trace(f"Materializing kernel {kernel_name} in {self.autodiff_mode}...")
@@ -990,6 +1019,8 @@ class Kernel:
                 set_arg_sparse_matrix_builder(indices, v)
                 return 1
             if dataclasses.is_dataclass(needed_arg_type):
+                if provided_arg_type != needed_arg_type:
+                    raise GsTaichiRuntimeError("needed", needed_arg_type, "!= provided", provided_arg_type)
                 assert provided_arg_type == needed_arg_type
                 idx = 0
                 for j, field in enumerate(dataclasses.fields(needed_arg_type)):
@@ -1064,6 +1095,7 @@ class Kernel:
                         compiled_kernel_data,
                     )
                     self.src_ll_cache_observations.cache_stored = True
+            self._last_compiled_kernel_data = compiled_kernel_data
             prog.launch_kernel(compiled_kernel_data, launch_ctx)
         except Exception as e:
             e = handle_exception_from_cpp(e)
@@ -1102,7 +1134,10 @@ class Kernel:
         raise GsTaichiRuntimeTypeError(f"Invalid return type on index={index}")
 
     def ensure_compiled(self, *args: tuple[Any, ...]) -> tuple[Callable, int, AutodiffMode]:
-        instance_id, arg_features = self.mapper.lookup(args)
+        try:
+            instance_id, arg_features = self.mapper.lookup(args)
+        except Exception as e:
+            raise type(e)(f"exception while trying to ensure compiled {self.func}:\n{e}") from e
         key = (self.func, instance_id, self.autodiff_mode)
         self.materialize(key=key, args=args, arg_features=arg_features)
         return key
@@ -1111,10 +1146,6 @@ class Kernel:
     # Thus this part needs to be fast. (i.e. < 3us on a 4 GHz x64 CPU)
     @_shell_pop_print
     def __call__(self, *args, **kwargs) -> Any:
-        # reset the observations
-        self.src_ll_cache_observations = SrcLlCacheObservations()
-        self.fe_ll_cache_observations = FeLlCacheObservations()
-
         args = _process_args(self, is_func=False, args=args, kwargs=kwargs)
 
         # Transform the primal kernel to forward mode grad kernel

gstaichi/lang/misc.py

@@ -303,6 +303,7 @@ def init(
     _test_mode: bool = False,
     enable_fallback: bool = True,
     require_version: str | None = None,
+    print_non_pure: bool = False,
     src_ll_cache: bool = True,
     **kwargs,
 ):
@@ -316,6 +317,8 @@ def init(
         default_fp (Optional[type]): Default floating-point type.
         default_ip (Optional[type]): Default integral type.
         require_version: A version string.
+        print_non_pure: Print the names of kernels, at the time they are executed, which are not annotated with
+                        @ti.pure
         src_ll_cache: enable SRC-LL-CACHE, which will accelerate loading from cache, across all architectures,
                       for pure kernels (i.e. kernels declared as @ti.pure)
         **kwargs: GsTaichi provides highly customizable compilation through
@@ -420,6 +423,7 @@ def init(
         runtime.print_full_traceback = spec_cfg.print_full_traceback
         runtime.unrolling_limit = spec_cfg.unrolling_limit
         runtime.src_ll_cache = src_ll_cache
+        runtime.print_non_pure = print_non_pure
         _logging.set_logging_level(spec_cfg.log_level.lower())
 
     # select arch (backend):

gstaichi/types/annotations.py

@@ -37,7 +37,7 @@ class Template(Generic[T]):
         self.ndim = ndim
 
     def __getitem__(self, i: Any) -> T:
-        raise NotImplemented
+        raise NotImplementedError()
 
 
 template = Template

gstaichi/types/ndarray_type.py

@@ -142,11 +142,11 @@ class NdarrayType:
 
     def __getitem__(self, i: Any) -> Any:
         # needed for pyright
-        raise NotImplemented
+        raise NotImplementedError()
 
     def __setitem__(self, i: Any, v: Any) -> None:
         # needed for pyright
-        raise NotImplemented
+        raise NotImplementedError()
 
 
 ndarray = NdarrayType

{gstaichi-2.2.0.dev1.dist-info → gstaichi-2.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gstaichi
-Version: 2.2.0.dev1
+Version: 2.2.1
 Summary: The GsTaichi Programming Language
 Home-page: https://github.com/Genesis-Embedded-AI/gstaichi
 Author: GsTaichi developers

{gstaichi-2.2.0.dev1.dist-info → gstaichi-2.2.1.dist-info}/RECORD

@@ -1,16 +1,16 @@
-gstaichi/CHANGELOG.md,sha256=y25NOBfNkT3yfsskpuu2jwillvlDmPMn5TnobuusiDc,127
+gstaichi/CHANGELOG.md,sha256=ULJ7MWOXcgIjvhQoTRfKlVtnZQCc43Pm_mM3ngNlCAk,97
 gstaichi/__init__.py,sha256=JFvEDYJ8EcHhPdrWMYez9cogC3nE1DMxjPmanOs8eYI,1044
 gstaichi/_funcs.py,sha256=u0o3F7Yq6r2dxkimThby4rvceRjUGajaLyzkPtwMJgM,23943
 gstaichi/_kernels.py,sha256=3R9_Ht0q1BSku-Jslvf5eYQY2bD8vFS5ZvL-im_iLPs,14664
 gstaichi/_logging.py,sha256=-4yqQ5rePskPzIbDdZSahAf4RYG_R7RNQDHau1wbcuc,3813
-gstaichi/_version.py,sha256=GyN4w8COty5f9ekpwC9gXF6R6cq1IM-adYIlpX4s3bw,28
+gstaichi/_version.py,sha256=DKiOwQ7NZUMJN1bLMEU03j1rBARcvobgfZWNrSmLABs,23
 gstaichi/_version_check.py,sha256=oifkwL4GmcOgh9OR7V4LtYt4dTGzokVUFxnVJspaOE0,3614
 gstaichi/experimental.py,sha256=SwtoqhUcSxCSea-sT10G-GgZaTcSWUNPmTACK0MCHH4,359
 gstaichi/_lib/__init__.py,sha256=rsFUVL8_2_eyK2Z4MiP6NvLIEykG1fsK8l1WZNQa3ko,74
 gstaichi/_lib/utils.py,sha256=sEi0azEnb0jardg4pKMZfQAjoWgz6WOnwrdf2NSMOEA,7735
 gstaichi/_lib/core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gstaichi/_lib/core/gstaichi_python.cp310-win_amd64.pyd,sha256=iJFNzmt5FcZy37YuAlLeok4-lZnur-5J8i_S8SmZueQ,80754688
-gstaichi/_lib/core/gstaichi_python.pyi,sha256=9SrxKGGX2YeNvDNLuW-8yYlAq2vWB6h13yh2vA4yz0g,113525
+gstaichi/_lib/core/gstaichi_python.cp310-win_amd64.pyd,sha256=YuUpgRO8g9BBLhWRMywTTFFvaMgw99j_ydULMi8SWGI,80759808
+gstaichi/_lib/core/gstaichi_python.pyi,sha256=7_57A_7rXZCOxePEciiEk2JNz_T21YsymFxIoED7kJE,113573
 gstaichi/_lib/core/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 gstaichi/_lib/runtime/runtime_cuda.bc,sha256=ZCHDx_T5HdLQ_u8oAETF6amiw3xdLHbgn8naellBfUw,155520
 gstaichi/_lib/runtime/runtime_x64.bc,sha256=dmh_JFX-s9H6zD2DCT_JGqYcYy-vkylvsb_LqXRnIDY,140112
@@ -35,7 +35,7 @@ gstaichi/examples/minimal.py,sha256=za9d0kngi_JSHWyRS69iwIiNzRMTSIi79CRftuqat60,
 gstaichi/lang/__init__.py,sha256=CBBu-q1yxgcZnGlRi4xQUVq8pdBeVdBHET3957hVPZ4,1331
 gstaichi/lang/_dataclass_util.py,sha256=1N8bZYLzwTDZ5Kau8ejMH9EtpxD2M2vPuhIWxZtLM7c,1122
 gstaichi/lang/_kernel_impl_dataclass.py,sha256=NKpNage3Xye_cBIWdQFD4hNd9aghH800xLFbwBRst94,7678
-gstaichi/lang/_ndarray.py,sha256=MW-Tb6o8U6qUFz_06n3LL1Y-VRTuwVMLZfWkosEu9nk,11126
+gstaichi/lang/_ndarray.py,sha256=6YY5-4ff_NMMJnG5in0mLkVDpR08k8ifBZgOJqiy8kE,11554
 gstaichi/lang/_ndrange.py,sha256=ATLSTdjdMGXcz3jCn2vkaaDGUglK5I4M9mwk4gJCLx8,5921
 gstaichi/lang/_template_mapper.py,sha256=5FX37WzYQrIPqGRCCwzzsljUiaRUXVbxhp6oFHcMfqU,9941
 gstaichi/lang/_texture.py,sha256=fGIPeGFZz1DO-RAUuEAiwSJIRN8o-7wgZi1URR4UFOw,6784
@@ -45,14 +45,14 @@ gstaichi/lang/common_ops.py,sha256=bIGmalXVnxVc5mw-k704zFwr38bR9uZx_4ieNS3PYpQ,8
 gstaichi/lang/exception.py,sha256=f-EWm56Px6TpskhCgEReqoY8meg_-4nZD9vFOA5TVic,2165
 gstaichi/lang/expr.py,sha256=bSWJyNJ55Xnv7mADHxKRhwH0lB2cXnRi9EpCrHgElLY,6587
 gstaichi/lang/field.py,sha256=pppo98T6DsHPhOR0-g-oc5lmEoR-mZeh96pMLkUyDQI,13677
-gstaichi/lang/impl.py,sha256=QE_8wSk0V6oRNBacrjiz41UO6aJR467aA6-AQ8T8Akk,42926
+gstaichi/lang/impl.py,sha256=-EnIJ9nkNUc_hn8_BmHrCJ6yR_iNqBkBSQvUkEmXYiQ,43491
 gstaichi/lang/kernel_arguments.py,sha256=ZUtk3AFs7xReYjoogTZ2xTMFaRZGAkFpOi1Et_lTGfY,6486
-gstaichi/lang/kernel_impl.py,sha256=aXo2iTUB0-kaCLEp8-9B6rMGtLkOiLOz_Ytzlk_5tAE,61225
+gstaichi/lang/kernel_impl.py,sha256=ii9GMTWT8dDI61uu1X854OtsuKFRpB5bn7tp5ouqs6E,63052
 gstaichi/lang/matrix.py,sha256=7bLpkkN_M9q9vHEY4ejpsGfQ99oTL8SFJiHzHIU7VBk,63762
 gstaichi/lang/matrix_ops.py,sha256=v89ID3thZ8irSLU9Tyf1hmEdQTjGyMHWqvfzsvnFxGo,10399
 gstaichi/lang/matrix_ops_utils.py,sha256=rrrC5KjmCt1w0ZNtLWqo5ZOjZNR74iqGjuVfHvqwwxg,5435
 gstaichi/lang/mesh.py,sha256=ukqJYWSf3ANPoiV4Ue-DWMPZ0y21bGrhEUseVAIJK4w,27758
-gstaichi/lang/misc.py,sha256=GP_80q1MyIsoSdSJBQg139aaxOuZgb4w5pXcQbC0ZTA,24332
+gstaichi/lang/misc.py,sha256=HMWBdv5D0Car1vFbs2Z5JCR3n9DRo0AXMcLJr9m6vBU,24563
 gstaichi/lang/ops.py,sha256=zKwfBLQrdnyyoHflIUe9DY3BQ2wQusfRrDLPRyAThpk,45919
 gstaichi/lang/runtime_ops.py,sha256=TKlPOBMWGGg0jQZFlik4f5c1Do5cUlhEMk6r4VlOfvM,237
 gstaichi/lang/shell.py,sha256=k9YPpByujyHis2Vv-oHSJowQAhhS1Iic9Oa7JKhy584,1095
@@ -61,7 +61,7 @@ gstaichi/lang/source_builder.py,sha256=ytBdJbFzcccD2ekVvy-v8YrllNEcSM1RRMUe-FWEN
 gstaichi/lang/struct.py,sha256=xrSMsj2yg1Cu9hcFEWEzHPyjZ8b8pb6XzeVDj9hrf_o,30060
 gstaichi/lang/util.py,sha256=VtBtYaJ-yGaXIS5lA2CV0dtdk8OZ9U55dy4MTZshOKk,7741
 gstaichi/lang/_fast_caching/__init__.py,sha256=6k4yq6r4Lu1SI4IHt8ByDPhV8uGA8HHKCAcGqvCoUmE,97
-gstaichi/lang/_fast_caching/args_hasher.py,sha256=eyDs5bupEGftgZ2HBfHP7G0OKGOm4L9gbkdMyN7eVNA,4243
+gstaichi/lang/_fast_caching/args_hasher.py,sha256=-P6khr5xip81cKXLyCehgVMtgTa9iXk1tI8TzlcPTXo,4542
 gstaichi/lang/_fast_caching/config_hasher.py,sha256=YtRFtJ8vA9sDgvzHBaMqLV8OG4voVjDTNeVDzPbHAzk,986
 gstaichi/lang/_fast_caching/fast_caching_types.py,sha256=sjnqzi_8E1zTa6EMUTgJpCOoAaxVLviRS3tUsc5l-i0,661
 gstaichi/lang/_fast_caching/function_hasher.py,sha256=Q53ReD1Lt694AR2-yuErSWP0WqIOQe4694IS2zXBhTY,1840
@@ -75,7 +75,7 @@ gstaichi/lang/ast/checkers.py,sha256=Lckg_gHzJyXTWVLYdp2HKBJxVygXwc1E2C8D_XTsL2k
 gstaichi/lang/ast/symbol_resolver.py,sha256=BtTp-U36tLZwbZVHI7Cq5-6GmrnaWdqC6S7SiWD4t4Q,1829
 gstaichi/lang/ast/transform.py,sha256=z28ji4mXsZ8Q9LmvaOkHu3BuE1m_opTaCMuP56ptvxQ,275
 gstaichi/lang/ast/ast_transformers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-gstaichi/lang/ast/ast_transformers/call_transformer.py,sha256=TpCl8KzDMJO_FpcjAm2sU8gedEylo55lp1m6y2R4pF0,13836
+gstaichi/lang/ast/ast_transformers/call_transformer.py,sha256=B67v8zZuxFuoiVsEaLvPQ8acnTQFMrBw1xyT7tsOHWM,14032
 gstaichi/lang/ast/ast_transformers/function_def_transformer.py,sha256=aR5WAw7INDpgJm7OI4pOJWTo3zuyVvC_qs0MonadxWI,12554
 gstaichi/lang/simt/__init__.py,sha256=63Z2qOmwx3hq1PdnJTCmwuKGIT9c24lFX_HgCtXMEJY,129
 gstaichi/lang/simt/block.py,sha256=2L0u67bFYfTamprexNwAD_7-ICClVshiZKeFsflXS60,3733
@@ -102,54 +102,54 @@ gstaichi/tools/diagnose.py,sha256=OZCQk_E5ijCXgBjSkUk3dDwenlQWXd-SLFBAgT4drHk,35
 gstaichi/tools/np2ply.py,sha256=2hJKNZvpoHCh8PBKXWBjkBCUu_q1U6L4eAmKdRBXUbg,15635
 gstaichi/tools/vtk.py,sha256=ziOsTJI4AaDy1OxNxsU2pi98J3_QXqkEJwTwMRsYJoM,1070
 gstaichi/types/__init__.py,sha256=mb4cL9OIiq2QosRoQIAYblE7CX_I2cH09NihtwMbh_s,621
-gstaichi/types/annotations.py,sha256=MFFvZjRHDWd6NFmZz1anVRpECQdJHn7OZsRJ8zdIdTU,1218
+gstaichi/types/annotations.py,sha256=nxLVzLhCoX2dvV6wskQPsx7xbgNuo0M1_dKr2-Z_je4,1225
 gstaichi/types/compound_types.py,sha256=DnfLSXAlfvDct88QJfez-KgIn8zGWH1RANsHuInFfOQ,1953
 gstaichi/types/enums.py,sha256=U6b2fXrIhBkYtWzz5IFmCNUndx8M-nyCP8nEUgr0l04,2099
-gstaichi/types/ndarray_type.py,sha256=D4idUVUqFX5uiTjumUpNRtf4PTAbxrmlVBI537-2X6g,7409
+gstaichi/types/ndarray_type.py,sha256=qaJE7G_6jp8jifWPSsvt_yzTMT3CNmNEJlcEz9V5F3s,7423
 gstaichi/types/primitive_types.py,sha256=otvV6aBEtMANfjiPaDjl2oB6_7yOsrkgwDkMgg73i_c,4273
 gstaichi/types/quant.py,sha256=orvn3oskYlhDd5Hln-Bt-oe0EXsPNjfOznMOCBEf1ak,3061
 gstaichi/types/texture_type.py,sha256=wWhaDNwOQwdMSnxCzXsltr3bWLmreOC9DvEVUV0vIAI,2444
 gstaichi/types/utils.py,sha256=8X1XmYXhPp-ANzwKRQhQmXV47DW9u-uWoc0fxMrWRDQ,276
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsConfig.cmake,sha256=jFprL3mwRRYJ0eh52yZqmUAWD9cA7xA7hhstBQU_scY,230
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsTarget-release.cmake,sha256=X438aOEvZsAeo6Suk2LfStdal8R5m3UyfUPCg51DXGM,1477
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsTarget.cmake,sha256=mriRCzvolwIXT55WDaD2ouu6AQvVHXQbRdJPmj0ZtSg,4468
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffConfig.cmake,sha256=3S1hOLuDTsAwTH-F_Xn9iBdCjjOH6Kx3EjNhDXBVShw,280
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffTargets-release.cmake,sha256=6fnDD1oaU-DEP4JIq75wotq4OSPW63pe5jEcTVwEtyo,890
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffTargets.cmake,sha256=U1RmMclqSVR8vePS98vpT2RqItqnivv5tXbfM58i9V0,5068
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkConfig.cmake,sha256=HgRoCSWZQhGnNEs938xBFDSdxK6crL6VtbrQ9-DqcMY,280
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkTargets-release.cmake,sha256=cqDRYOLX02s-xo8EKyuGQwYX6uNa2Lxb363OP0tUB54,890
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkTargets.cmake,sha256=nlQJ33ZSrKM5w-xi7lYlopolM5lwcX4Yyqrytlbvj5c,5028
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintConfig.cmake,sha256=E0o9wMS7mOV58CacqGpW5EOyG70WrdBKexx1AFlCyiQ,280
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintTargets-release.cmake,sha256=3QC3rN6BhZ4n5-WrR9pmu4AfH1QKFIGCCXZjPA3A8c0,890
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintTargets.cmake,sha256=X2bmbqKDvgHDUaYEjssse8Rgk_ZsETuMMS3EaX2gmo0,5068
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optConfig.cmake,sha256=CYygoz_0kxDnsT7_NbByRCuviPMS0TwwK86Qwc6tnnw,275
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optTargets-release.cmake,sha256=vSAY69UsznqGYYqEUdL255U2B3LUZl9PSexaQwmPECw,883
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optTargets.cmake,sha256=5Erfqk7DZeEa2sveBBlRw8QwBrdBZ55GDdo14DdNxpI,5029
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceConfig.cmake,sha256=i2lpvjP36I_6iEh5mObgdPMz5UOHQ4wIIOcbP5NIjTk,290
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceTarget-release.cmake,sha256=D2c9eUqoh3H-ihr_1up2Ppo_2nSLoNyxmo9eWUV_pGE,904
-gstaichi-2.2.0.dev1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceTarget.cmake,sha256=_Miasn_g_L8dG2fTiigOC-rrDc3aKgHy3IfMi7DB2b4,5077
-gstaichi-2.2.0.dev1.data/data/bin/SPIRV-Tools-shared.dll,sha256=2VPDXmLsv2LNhjJpNy27u2SMft3C8I4jqmyGFs4MqrU,2042368
-gstaichi-2.2.0.dev1.data/data/include/GLFW/glfw3.h,sha256=QIB4KnbXSK4tfZ99LqTfJdn-ObkYf5vp70ee9uBPzFQ,240786
-gstaichi-2.2.0.dev1.data/data/include/GLFW/glfw3native.h,sha256=3hGJGACURjjhav5xcOb9F1fW76uPjfAzoXZlK3UUxAc,19397
-gstaichi-2.2.0.dev1.data/data/include/spirv-tools/instrument.hpp,sha256=92ifeZiADg0tWjsnYsVqQoH5KbIxxnz3AK-S9r0ln2A,11917
-gstaichi-2.2.0.dev1.data/data/include/spirv-tools/libspirv.h,sha256=3rvQUbgMs9h3jxnC4OLRyPN-lWvqSjHgy7tS3ZkOYvI,42688
-gstaichi-2.2.0.dev1.data/data/include/spirv-tools/libspirv.hpp,sha256=TMJs7spPKjYx0FlnorbOSany4R1k2bk8B4wOi_vttic,15012
-gstaichi-2.2.0.dev1.data/data/include/spirv-tools/linker.hpp,sha256=SKevy72q3RRFOdqQotIb08h9NNMfAnNMKmSF7B20d0A,3691
-gstaichi-2.2.0.dev1.data/data/include/spirv-tools/optimizer.hpp,sha256=5MPYUrk-hRM6tbFkCr2IUKsukO8eAPjA9_DE-ib4Lnk,50060
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-diff.lib,sha256=ttu4f_ooLGR4gMkcLA_ZB53LSRZwWKj61tmIuA9TwR8,4766344
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-link.lib,sha256=WSAsOqCFM-ekfGxZaFPBQBR5hQVgWtkN33OafoYMHJM,4074904
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-lint.lib,sha256=TLkS0JQvTsTJLGON4z-uwbakgPhPrNk7Ehbc1RxgQ2o,4723542
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-opt.lib,sha256=Gxc85Cq6LZenVuJS1RLWcBl33RQwSbI2wXhWIGOVVPw,153197732
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-reduce.lib,sha256=5bD2YSRt-WS80uOlKKEf_fVC0ji-BAue4pvUTmV9he0,14806994
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools-shared.lib,sha256=_x4gg9SeOEDkWOsTAKNGJBr1l4eR-cO5xGr4BCvloMg,16470
-gstaichi-2.2.0.dev1.data/data/lib/SPIRV-Tools.lib,sha256=P8JoWOSPgZ3vgDN11FV9RbLcTWsMSUvXdfRGQCic2xM,37167956
-gstaichi-2.2.0.dev1.data/data/lib/glfw3.lib,sha256=lQ2WbikvG_NulojycswUbxzzqfc66qhYkmcSAEIJIo4,687808
-gstaichi-2.2.0.dev1.data/data/lib/cmake/glfw3/glfw3Config.cmake,sha256=YZVL9F2A70B51dNW0W--Scv9ga3cbyZYjlsaDCs-zHk,118
-gstaichi-2.2.0.dev1.data/data/lib/cmake/glfw3/glfw3ConfigVersion.cmake,sha256=mttbBnVYyxo57F-pEE8g4-E62lcH8xYujvlmTQyD7ug,2827
-gstaichi-2.2.0.dev1.data/data/lib/cmake/glfw3/glfw3Targets-release.cmake,sha256=VlH9iN0TgA6MlsPimvnOgTG0gxzsaoGj-RmWEqd9QHA,806
-gstaichi-2.2.0.dev1.data/data/lib/cmake/glfw3/glfw3Targets.cmake,sha256=D3NWABsZAPn6vu3kAtXXUo3Lqql5NiGhSHVqtwg3UOw,4242
-gstaichi-2.2.0.dev1.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
-gstaichi-2.2.0.dev1.dist-info/METADATA,sha256=ifQe-hzU-msSbbgxIshKQBjh7bQIIgfsFfJOS2q3b7c,4851
-gstaichi-2.2.0.dev1.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
-gstaichi-2.2.0.dev1.dist-info/top_level.txt,sha256=BoKkPzj6Sfycl5Ge5sB9QRDVZvFB5R0sxbZn4YNPb80,9
-gstaichi-2.2.0.dev1.dist-info/RECORD,,
+gstaichi-2.2.1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsConfig.cmake,sha256=jFprL3mwRRYJ0eh52yZqmUAWD9cA7xA7hhstBQU_scY,230
+gstaichi-2.2.1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsTarget-release.cmake,sha256=X438aOEvZsAeo6Suk2LfStdal8R5m3UyfUPCg51DXGM,1477
+gstaichi-2.2.1.data/data/SPIRV-Tools/cmake/SPIRV-ToolsTarget.cmake,sha256=mriRCzvolwIXT55WDaD2ouu6AQvVHXQbRdJPmj0ZtSg,4468
+gstaichi-2.2.1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffConfig.cmake,sha256=3S1hOLuDTsAwTH-F_Xn9iBdCjjOH6Kx3EjNhDXBVShw,280
+gstaichi-2.2.1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffTargets-release.cmake,sha256=6fnDD1oaU-DEP4JIq75wotq4OSPW63pe5jEcTVwEtyo,890
+gstaichi-2.2.1.data/data/SPIRV-Tools-diff/cmake/SPIRV-Tools-diffTargets.cmake,sha256=U1RmMclqSVR8vePS98vpT2RqItqnivv5tXbfM58i9V0,5068
+gstaichi-2.2.1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkConfig.cmake,sha256=HgRoCSWZQhGnNEs938xBFDSdxK6crL6VtbrQ9-DqcMY,280
+gstaichi-2.2.1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkTargets-release.cmake,sha256=cqDRYOLX02s-xo8EKyuGQwYX6uNa2Lxb363OP0tUB54,890
+gstaichi-2.2.1.data/data/SPIRV-Tools-link/cmake/SPIRV-Tools-linkTargets.cmake,sha256=nlQJ33ZSrKM5w-xi7lYlopolM5lwcX4Yyqrytlbvj5c,5028
+gstaichi-2.2.1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintConfig.cmake,sha256=E0o9wMS7mOV58CacqGpW5EOyG70WrdBKexx1AFlCyiQ,280
+gstaichi-2.2.1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintTargets-release.cmake,sha256=3QC3rN6BhZ4n5-WrR9pmu4AfH1QKFIGCCXZjPA3A8c0,890
+gstaichi-2.2.1.data/data/SPIRV-Tools-lint/cmake/SPIRV-Tools-lintTargets.cmake,sha256=X2bmbqKDvgHDUaYEjssse8Rgk_ZsETuMMS3EaX2gmo0,5068
+gstaichi-2.2.1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optConfig.cmake,sha256=CYygoz_0kxDnsT7_NbByRCuviPMS0TwwK86Qwc6tnnw,275
+gstaichi-2.2.1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optTargets-release.cmake,sha256=vSAY69UsznqGYYqEUdL255U2B3LUZl9PSexaQwmPECw,883
+gstaichi-2.2.1.data/data/SPIRV-Tools-opt/cmake/SPIRV-Tools-optTargets.cmake,sha256=5Erfqk7DZeEa2sveBBlRw8QwBrdBZ55GDdo14DdNxpI,5029
+gstaichi-2.2.1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceConfig.cmake,sha256=i2lpvjP36I_6iEh5mObgdPMz5UOHQ4wIIOcbP5NIjTk,290
+gstaichi-2.2.1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceTarget-release.cmake,sha256=D2c9eUqoh3H-ihr_1up2Ppo_2nSLoNyxmo9eWUV_pGE,904
+gstaichi-2.2.1.data/data/SPIRV-Tools-reduce/cmake/SPIRV-Tools-reduceTarget.cmake,sha256=_Miasn_g_L8dG2fTiigOC-rrDc3aKgHy3IfMi7DB2b4,5077
+gstaichi-2.2.1.data/data/bin/SPIRV-Tools-shared.dll,sha256=YUPSqEbIfjAtdtwj2NofDTNrFHc4A7xjMrPGUy_LwAQ,2042368
+gstaichi-2.2.1.data/data/include/GLFW/glfw3.h,sha256=QIB4KnbXSK4tfZ99LqTfJdn-ObkYf5vp70ee9uBPzFQ,240786
+gstaichi-2.2.1.data/data/include/GLFW/glfw3native.h,sha256=3hGJGACURjjhav5xcOb9F1fW76uPjfAzoXZlK3UUxAc,19397
+gstaichi-2.2.1.data/data/include/spirv-tools/instrument.hpp,sha256=92ifeZiADg0tWjsnYsVqQoH5KbIxxnz3AK-S9r0ln2A,11917
+gstaichi-2.2.1.data/data/include/spirv-tools/libspirv.h,sha256=3rvQUbgMs9h3jxnC4OLRyPN-lWvqSjHgy7tS3ZkOYvI,42688
+gstaichi-2.2.1.data/data/include/spirv-tools/libspirv.hpp,sha256=TMJs7spPKjYx0FlnorbOSany4R1k2bk8B4wOi_vttic,15012
+gstaichi-2.2.1.data/data/include/spirv-tools/linker.hpp,sha256=SKevy72q3RRFOdqQotIb08h9NNMfAnNMKmSF7B20d0A,3691
+gstaichi-2.2.1.data/data/include/spirv-tools/optimizer.hpp,sha256=5MPYUrk-hRM6tbFkCr2IUKsukO8eAPjA9_DE-ib4Lnk,50060
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-diff.lib,sha256=XT8Su_3ePujEkK-xJhzuy2ML5xxukwvwUU7X9EGdFaQ,4766344
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-link.lib,sha256=jAeaueoYmkUZDaBwqOhq1kgzVVbCFPwAhxtKUMNFMpg,4074904
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-lint.lib,sha256=cGb9h63vHCIJgUYy3JGFnQ-6llPHFZJkXOz33wlZe2o,4723542
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-opt.lib,sha256=EPxV3Pc_FbBvz45Luk2uMJL5cziKThqPVj9xxQMpogg,153197732
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-reduce.lib,sha256=DgA3GVFOz79wRmWd_Q7cOsiQHsX2uAfdMX5jNeP39A0,14806994
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools-shared.lib,sha256=_x4gg9SeOEDkWOsTAKNGJBr1l4eR-cO5xGr4BCvloMg,16470
+gstaichi-2.2.1.data/data/lib/SPIRV-Tools.lib,sha256=RP8LpiwZYLQBMDIwvVTreRQa5E095HDIj_FGnBvU0a0,37167956
+gstaichi-2.2.1.data/data/lib/glfw3.lib,sha256=rvIxfJlop2P4LpkyCxO87M_SW6PkbPjlr4EQtLjOXvc,687808
+gstaichi-2.2.1.data/data/lib/cmake/glfw3/glfw3Config.cmake,sha256=YZVL9F2A70B51dNW0W--Scv9ga3cbyZYjlsaDCs-zHk,118
+gstaichi-2.2.1.data/data/lib/cmake/glfw3/glfw3ConfigVersion.cmake,sha256=mttbBnVYyxo57F-pEE8g4-E62lcH8xYujvlmTQyD7ug,2827
+gstaichi-2.2.1.data/data/lib/cmake/glfw3/glfw3Targets-release.cmake,sha256=VlH9iN0TgA6MlsPimvnOgTG0gxzsaoGj-RmWEqd9QHA,806
+gstaichi-2.2.1.data/data/lib/cmake/glfw3/glfw3Targets.cmake,sha256=D3NWABsZAPn6vu3kAtXXUo3Lqql5NiGhSHVqtwg3UOw,4242
+gstaichi-2.2.1.dist-info/licenses/LICENSE,sha256=HrhfyXIkWY2tGFK11kg7vPCqhgh5DcxleloqdhrpyMY,11558
+gstaichi-2.2.1.dist-info/METADATA,sha256=PVR9aTWhsSZXekES-Ra_YZiSkNjLDRUDWTEVvai2p1A,4846
+gstaichi-2.2.1.dist-info/WHEEL,sha256=KUuBC6lxAbHCKilKua8R9W_TM71_-9Sg5uEP3uDWcoU,101
+gstaichi-2.2.1.dist-info/top_level.txt,sha256=BoKkPzj6Sfycl5Ge5sB9QRDVZvFB5R0sxbZn4YNPb80,9
+gstaichi-2.2.1.dist-info/RECORD,,