gstaichi 0.1.23.dev0__cp310-cp310-macosx_15_0_arm64.whl → 1.0.1__cp310-cp310-macosx_15_0_arm64.whl

gstaichi/CHANGELOG.md

@@ -0,0 +1,6 @@
+Highlights:
+   - **Build system**
+      - Fix miscellaneous pypi publish issues (#129) (by **Hugh Perkins**)
+
+Full changelog:
+   - [Build] Fix miscellaneous pypi publish issues (#129) (by **Hugh Perkins**)

gstaichi/__init__.py

@@ -0,0 +1,40 @@
+# type: ignore
+
+from gstaichi import (
+    ad,
+    algorithms,
+    experimental,
+    linalg,
+    math,
+    sparse,
+    tools,
+    types,
+)
+from gstaichi._funcs import *
+from gstaichi._lib import core as _ti_core
+from gstaichi._lib.utils import warn_restricted_version
+from gstaichi._logging import *
+from gstaichi._snode import *
+from gstaichi.lang import *  # pylint: disable=W0622 # TODO(archibate): It's `gstaichi.lang.core` overriding `gstaichi.core`
+from gstaichi.types.annotations import *
+
+# Provide a shortcut to types since they're commonly used.
+from gstaichi.types.primitive_types import *
+
+
+def __getattr__(attr):
+    if attr == "cfg":
+        return None if lang.impl.get_runtime()._prog is None else lang.impl.current_cfg()
+    raise AttributeError(f"module '{__name__}' has no attribute '{attr}'")
+
+
+__version__ = (
+    _ti_core.get_version_major(),
+    _ti_core.get_version_minor(),
+    _ti_core.get_version_patch(),
+)
+
+del _ti_core
+
+warn_restricted_version()
+del warn_restricted_version

{taichi → gstaichi}/_funcs.py

@@ -2,12 +2,12 @@
 
 import math
 
-from taichi.lang import impl, ops
-from taichi.lang.impl import get_runtime, grouped, static
-from taichi.lang.kernel_impl import func
-from taichi.lang.matrix import Matrix, Vector
-from taichi.types import f32, f64
-from taichi.types.annotations import template
+from gstaichi.lang import impl, ops
+from gstaichi.lang.impl import get_runtime, grouped, static
+from gstaichi.lang.kernel_impl import func
+from gstaichi.lang.matrix import Matrix, Vector
+from gstaichi.types import f32, f64
+from gstaichi.types.annotations import template
 
 
 @func
@@ -28,7 +28,7 @@ def _randn(dt):
 def randn(dt=None):
     """Generate a random float sampled from univariate standard normal
     (Gaussian) distribution of mean 0 and variance 1, using the
-    Box-Muller transformation. Must be called in Taichi scope.
+    Box-Muller transformation. Must be called in GsTaichi scope.
 
     Args:
         dt (DataType): Data type of the required random number. Default to `None`.
@@ -698,7 +698,7 @@ def solve(A, b, dt=None):
 
 
 @func
-def field_fill_taichi_scope(F: template(), val: template()):
+def field_fill_gstaichi_scope(F: template(), val: template()):
     for I in grouped(F):
         F[I] = val
 

{taichi → gstaichi}/_kernels.py

@@ -1,21 +1,21 @@
 # type: ignore
 
-from taichi._funcs import field_fill_taichi_scope
-from taichi._lib.utils import get_os_name
-from taichi.lang import ops
-from taichi.lang._ndrange import ndrange
-from taichi.lang.expr import Expr
-from taichi.lang.field import ScalarField
-from taichi.lang.impl import grouped, static, static_assert
-from taichi.lang.kernel_impl import func, kernel
-from taichi.lang.misc import loop_config
-from taichi.lang.simt import block, warp
-from taichi.lang.snode import deactivate
-from taichi.math import vec3
-from taichi.types import ndarray_type, texture_type, vector
-from taichi.types.annotations import template
-from taichi.types.enums import Format
-from taichi.types.primitive_types import f16, f32, f64, i32, u8
+from gstaichi._funcs import field_fill_gstaichi_scope
+from gstaichi._lib.utils import get_os_name
+from gstaichi.lang import ops
+from gstaichi.lang._ndrange import ndrange
+from gstaichi.lang.expr import Expr
+from gstaichi.lang.field import ScalarField
+from gstaichi.lang.impl import grouped, static, static_assert
+from gstaichi.lang.kernel_impl import func, kernel
+from gstaichi.lang.misc import loop_config
+from gstaichi.lang.simt import block, warp
+from gstaichi.lang.snode import deactivate
+from gstaichi.math import vec3
+from gstaichi.types import ndarray_type, texture_type, vector
+from gstaichi.types.annotations import template
+from gstaichi.types.enums import Format
+from gstaichi.types.primitive_types import f16, f32, f64, i32, u8
 
 
 # A set of helper (meta)functions
@@ -92,7 +92,7 @@ def vector_to_fast_image(img: template(), out: ndarray_type.ndarray()):
             r, g, b = color[:3]
 
         idx = j * img.shape[0] + i
-        # We use i32 for |out| since OpenGL and Metal doesn't support u8 types
+        # We use i32 for |out| since Metal doesn't support u8 types
         if static(get_os_name() != "osx"):
             out[idx] = (r << 16) + (g << 8) + b
         else:
@@ -235,7 +235,7 @@ def arr_vulkan_layout_to_arr_normal_layout(vk_arr: ndarray_type.ndarray(), norma
         normal_arr[i, j] = vk_arr[(h - 1 - j) * w + i]
 
 
-# extract ndarray of raw vulkan memory layout into a taichi-field data
+# extract ndarray of raw vulkan memory layout into a gstaichi-field data
 # structure with normal memory layout.
 @kernel
 def arr_vulkan_layout_to_field_normal_layout(vk_arr: ndarray_type.ndarray(), normal_field: template()):
@@ -259,7 +259,7 @@ def clear_gradients(_vars: template()):
 
 @kernel
 def field_fill_python_scope(F: template(), val: template()):
-    field_fill_taichi_scope(F, val)
+    field_fill_gstaichi_scope(F, val)
 
 
 @kernel

gstaichi/_lib/__init__.py

@@ -0,0 +1,3 @@
+# type: ignore
+
+from gstaichi._lib.utils import ti_python_core as core