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

{taichi → gstaichi}/math/mathimpl.py

@@ -6,11 +6,11 @@ Math functions for glsl-like functions and other stuff.
 """
 import math
 
-from taichi.lang import impl, ops
-from taichi.lang.impl import static, zero
-from taichi.lang.kernel_impl import func
-from taichi.lang.matrix import Matrix
-from taichi.lang.ops import (
+from gstaichi.lang import impl, ops
+from gstaichi.lang.impl import static, zero
+from gstaichi.lang.kernel_impl import func
+from gstaichi.lang.matrix import Matrix
+from gstaichi.lang.ops import (
     acos,
     asin,
     atan2,
@@ -28,8 +28,8 @@ from taichi.lang.ops import (
     tan,
     tanh,
 )
-from taichi.types import matrix, template, vector
-from taichi.types.primitive_types import f64, u32, u64
+from gstaichi.types import matrix, template, vector
+from gstaichi.types.primitive_types import f64, u32, u64
 
 cfg = impl.default_cfg
 
@@ -108,21 +108,21 @@ def mix(x, y, a):
     `a` to weight between them. The return value is computed as
     `x * (1 - a) + a * y`.
 
-    The arguments can be scalars or :class:`~taichi.Matrix`,
+    The arguments can be scalars or :class:`~gstaichi.Matrix`,
     as long as the operation can be performed.
 
     This function is similar to the `mix` function in GLSL.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the start of the range in which to interpolate.
-        y (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        y (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the end of the range in which to interpolate.
-        a (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        a (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the weight to use to interpolate between x and y.
 
     Returns:
-        (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The linear
+        (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The linear
             interpolation of `x` and `y` by weight `a`.
 
     Example::
@@ -146,15 +146,15 @@ def clamp(x, xmin, xmax):
     """Constrain a value to lie between two further values, element-wise.
     The returned value is computed as `min(max(x, xmin), xmax)`.
 
-    The arguments can be scalars or :class:`~taichi.Matrix`,
+    The arguments can be scalars or :class:`~gstaichi.Matrix`,
     as long as they can be broadcasted to a common shape.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the value to constrain.
-        y (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        y (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the lower end of the range into which to constrain `x`.
-        a (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        a (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the upper end of the range into which to constrain `x`.
 
     Returns:
@@ -181,13 +181,13 @@ def step(edge, x):
     For element i of the return value, 0.0 is returned if x[i] < edge[i],
     and 1.0 is returned otherwise.
 
-    The two arguments can be scalars or :class:`~taichi.Matrix`,
+    The two arguments can be scalars or :class:`~gstaichi.Matrix`,
     as long as they can be broadcasted to a common shape.
 
     Args:
-        edge (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        edge (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the location of the edge of the step function.
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specify
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specify
             the value to be used to generate the step function.
 
     Returns:
@@ -209,7 +209,7 @@ def fract(x):
     It's equivalent to `x - ti.floor(x)`.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The
             input value.
 
     Returns:
@@ -229,17 +229,17 @@ def smoothstep(edge0, edge1, x):
     """Performs smooth Hermite interpolation between 0 and 1 when
     `edge0 < x < edge1`, element-wise.
 
-    The arguments can be scalars or :class:`~taichi.Matrix`,
+    The arguments can be scalars or :class:`~gstaichi.Matrix`,
     as long as they can be broadcasted to a common shape.
 
     This function is equivalent to the `smoothstep` in GLSL.
 
     Args:
-        edge0 (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specifies
+        edge0 (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specifies
             the value of the lower edge of the Hermite function.
-        edge1 (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specifies
+        edge1 (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specifies
             the value of the upper edge of the Hermite function.
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): Specifies
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): Specifies
             the source value for interpolation.
 
     Returns:
@@ -262,7 +262,7 @@ def sign(x):
     """Extract the sign of the parameter, element-wise.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The
             input value.
 
     Returns:
@@ -286,7 +286,7 @@ def normalize(v):
     It's equivalent to the `normalize` function is GLSL.
 
     Args:
-        x (:class:`~taichi.Matrix`): The vector to normalize.
+        x (:class:`~gstaichi.Matrix`): The vector to normalize.
 
     Returns:
         The normalized vector :math:`v/|v|`.
@@ -308,7 +308,7 @@ def log2(x):
     This is equivalent to the `log2` function is GLSL.
 
     Args:
-        x (:class:`~taichi.Matrix`): The input value.
+        x (:class:`~gstaichi.Matrix`): The input value.
 
     Returns:
         The base 2 logarithm of `x`.
@@ -335,8 +335,8 @@ def reflect(x, n):
     `n` should be normalized in order to achieve the desired result.
 
     Args:
-        x (:class:`~taichi.Matrix`): The incident vector.
-        n (:class:`~taichi.Matrix`): The normal vector.
+        x (:class:`~gstaichi.Matrix`): The incident vector.
+        n (:class:`~gstaichi.Matrix`): The normal vector.
 
     Returns:
         The reflected vector.
@@ -357,7 +357,7 @@ def degrees(x):
     """Convert `x` in radians to degrees, element-wise.
 
     Args:
-        x (:class:`~taichi.Matrix`): The input angle in radians.
+        x (:class:`~gstaichi.Matrix`): The input angle in radians.
 
     Returns:
         angle in degrees.
@@ -376,7 +376,7 @@ def radians(x):
     """Convert `x` in degrees to radians, element-wise.
 
     Args:
-        x (:class:`~taichi.Matrix`): The input angle in degrees.
+        x (:class:`~gstaichi.Matrix`): The input angle in degrees.
 
     Returns:
         angle in radians.
@@ -397,8 +397,8 @@ def distance(x, y):
     This function is equivalent to the `distance` function is GLSL.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The first input point.
-        y (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The second input point.
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The first input point.
+        y (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The second input point.
 
     Returns:
         The distance between the two points.
@@ -420,12 +420,12 @@ def refract(x, n, eta):
     This function is equivalent to the `refract` function in GLSL.
 
     Args:
-        x (:class:`~taichi.Matrix`): The incident vector.
-        n (:class:`~taichi.Matrix`): The normal vector.
+        x (:class:`~gstaichi.Matrix`): The incident vector.
+        n (:class:`~gstaichi.Matrix`): The normal vector.
         eta (float): The ratio of indices of refraction.
 
     Returns:
-        :class:`~taichi.Matrix`: The refraction direction vector.
+        :class:`~gstaichi.Matrix`: The refraction direction vector.
 
     Example::
 
@@ -447,8 +447,8 @@ def dot(x, y):
     """Calculate the dot product of two vectors.
 
     Args:
-        x (:class:`~taichi.Matrix`): The first input vector.
-        y (:class:`~taichi.Matrix`): The second input vector.
+        x (:class:`~gstaichi.Matrix`): The first input vector.
+        y (:class:`~gstaichi.Matrix`): The second input vector.
 
     Returns:
         The dot product of two vectors.
@@ -469,11 +469,11 @@ def cross(x, y):
 
     The two input vectors must have the same dimension :math:`d <= 3`.
 
-    This function calls the `cross` method of :class:`~taichi.Vector`.
+    This function calls the `cross` method of :class:`~gstaichi.Vector`.
 
     Args:
-        x (:class:`~taichi.Matrix`): The first input vector.
-        y (:class:`~taichi.Matrix`): The second input vector.
+        x (:class:`~gstaichi.Matrix`): The first input vector.
+        y (:class:`~gstaichi.Matrix`): The second input vector.
 
     Returns:
         The cross product of two vectors.
@@ -493,8 +493,8 @@ def mod(x, y):
     """Compute value of one parameter modulo another, element-wise.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The first input.
-        y (:mod:`~taichi.types.primitive_types`, :class:`~taichi.Matrix`): The second input.
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The first input.
+        y (:mod:`~gstaichi.types.primitive_types`, :class:`~gstaichi.Matrix`): The second input.
 
     Returns:
         the value of `x` modulo `y`. This is computed as `x - y * floor(x/y)`.
@@ -519,7 +519,7 @@ def translate(dx, dy, dz):
         dz (float): delta z.
 
     Returns:
-        :class:`~taichi.math.mat4`: translation matrix.
+        :class:`~gstaichi.math.mat4`: translation matrix.
 
     Example::
 
@@ -549,7 +549,7 @@ def scale(sx, sy, sz):
         sz (float): scale z.
 
     Returns:
-        :class:`~taichi.math.mat4`: scale matrix.
+        :class:`~gstaichi.math.mat4`: scale matrix.
 
     Example::
 
@@ -578,7 +578,7 @@ def rot_by_axis(axis, ang):
         ang (float): angle in radians unit
 
     Returns:
-        :class:`~taichi.math.mat4`: rotation matrix
+        :class:`~gstaichi.math.mat4`: rotation matrix
     """
     c = ops.cos(ang)
     s = ops.sin(ang)
@@ -620,7 +620,7 @@ def rot_yaw_pitch_roll(yaw, pitch, roll):
         roll  (float): roll angle in radians unit
 
     Returns:
-        :class:`~taichi.math.mat4`: rotation matrix
+        :class:`~gstaichi.math.mat4`: rotation matrix
     """
     ch = ops.cos(yaw)
     sh = ops.sin(yaw)
@@ -648,7 +648,7 @@ def rotation2d(ang):
         ang (float): Angle of rotation in radians.
 
     Returns:
-        :class:`~taichi.math.mat2`: 2x2 rotation matrix.
+        :class:`~gstaichi.math.mat2`: 2x2 rotation matrix.
 
     Example::
 
@@ -668,7 +668,7 @@ def rotation3d(ang_x, ang_y, ang_z):
         ang_y (float): angle in radians unit around Y axis
         ang_z (float): angle in radians unit around Z axis
     Returns:
-        :class:`~taichi.math.mat4`: rotation matrix
+        :class:`~gstaichi.math.mat4`: rotation matrix
     Example:
 
         >>> ti.math.rotation3d(0.52, -0.785, 1.046)
@@ -684,7 +684,7 @@ def rotation3d(ang_x, ang_y, ang_z):
 def eye(n: template()):
     """Returns the nxn identity matrix.
 
-    Alias for :func:`~taichi.Matrix.identity`.
+    Alias for :func:`~gstaichi.Matrix.identity`.
     """
     return Matrix.identity(float, n)
 
@@ -695,7 +695,7 @@ def length(x):
 
     This function is equivalent to the `length` function in GLSL.
     Args:
-        x (:class:`~taichi.Matrix`): The vector of which to calculate the length.
+        x (:class:`~gstaichi.Matrix`): The vector of which to calculate the length.
 
     Returns:
         The Euclidean norm of the vector.
@@ -711,7 +711,7 @@ def length(x):
 
 @func
 def determinant(m):
-    """Alias for :func:`taichi.Matrix.determinant`."""
+    """Alias for :func:`gstaichi.Matrix.determinant`."""
     return m.determinant()
 
 
@@ -722,7 +722,7 @@ def inverse(mat):  # pylint: disable=R1710
     This function is equivalent to the `inverse` function in GLSL.
 
     Args:
-        mat (:class:`taichi.Matrix`): The matrix of which to take the inverse. \
+        mat (:class:`gstaichi.Matrix`): The matrix of which to take the inverse. \
             Supports only 2x2, 3x3 and 4x4 matrices.
 
     Returns:
@@ -744,7 +744,7 @@ def isinf(x):
     """Determines whether the parameter is positive or negative infinity, element-wise.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`taichi.Matrix`): The input.
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`gstaichi.Matrix`): The input.
 
     Example:
 
@@ -770,7 +770,7 @@ def isnan(x):
     """Determines whether the parameter is a number, element-wise.
 
     Args:
-        x (:mod:`~taichi.types.primitive_types`, :class:`taichi.Matrix`): The input.
+        x (:mod:`~gstaichi.types.primitive_types`, :class:`gstaichi.Matrix`): The input.
 
     Example:
 
@@ -795,7 +795,7 @@ def isnan(x):
 def vdir(ang):
     """Returns the 2d unit vector with argument equals `ang`.
 
-    x (:mod:`~taichi.types.primitive_types`): The input angle in radians.
+    x (:mod:`~gstaichi.types.primitive_types`): The input angle in radians.
 
     Example:
 

gstaichi/profiler/__init__.py

@@ -0,0 +1,6 @@
+# type: ignore
+
+from gstaichi.profiler.kernel_metrics import *
+from gstaichi.profiler.kernel_profiler import *
+from gstaichi.profiler.memory_profiler import *
+from gstaichi.profiler.scoped_profiler import *

{taichi → gstaichi}/profiler/kernel_metrics.py

@@ -1,24 +1,24 @@
 # type: ignore
 
-from taichi._lib import core as _ti_core
+from gstaichi._lib import core as _ti_core
 
 
 class CuptiMetric:
-    """A class to add CUPTI metric for :class:`~taichi.profiler.kernel_profiler.KernelProfiler`.
+    """A class to add CUPTI metric for :class:`~gstaichi.profiler.kernel_profiler.KernelProfiler`.
 
     This class is designed to add user selected CUPTI metrics.
     Only available for the CUDA backend now, i.e. you need ``ti.init(kernel_profiler=True, arch=ti.cuda)``.
-    For usage of this class, see examples in func :func:`~taichi.profiler.set_kernel_profiler_metrics` and :func:`~taichi.profiler.collect_kernel_profiler_metrics`.
+    For usage of this class, see examples in func :func:`~gstaichi.profiler.set_kernel_profiler_metrics` and :func:`~gstaichi.profiler.collect_kernel_profiler_metrics`.
 
     Args:
-        name (str): name of metric that collected by CUPTI toolkit. used by :func:`~taichi.profiler.set_kernel_profiler_metrics` and :func:`~taichi.profiler.collect_kernel_profiler_metrics`.
-        header (str): column header of this metric, used by :func:`~taichi.profiler.print_kernel_profiler_info`.
-        val_format (str): format for print metric value (and unit of this value), used by :func:`~taichi.profiler.print_kernel_profiler_info`.
-        scale (float): scale of metric value, used by :func:`~taichi.profiler.print_kernel_profiler_info`.
+        name (str): name of metric that collected by CUPTI toolkit. used by :func:`~gstaichi.profiler.set_kernel_profiler_metrics` and :func:`~gstaichi.profiler.collect_kernel_profiler_metrics`.
+        header (str): column header of this metric, used by :func:`~gstaichi.profiler.print_kernel_profiler_info`.
+        val_format (str): format for print metric value (and unit of this value), used by :func:`~gstaichi.profiler.print_kernel_profiler_info`.
+        scale (float): scale of metric value, used by :func:`~gstaichi.profiler.print_kernel_profiler_info`.
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(kernel_profiler=True, arch=ti.cuda)
         >>> num_elements = 128*1024*1024
@@ -46,7 +46,7 @@ class CuptiMetric:
         >>> ti.profiler.print_kernel_profiler_info('trace')
 
     Note:
-        For details about using CUPTI in Taichi, please visit https://docs.taichi-lang.org/docs/profiler#advanced-mode.
+        For details about using CUPTI in GsTaichi, please visit https://docs.taichi-lang.org/docs/profiler#advanced-mode.
     """
 
     def __init__(self, name="", header="unnamed_header", val_format="     {:8.0f} ", scale=1.0):
@@ -244,7 +244,7 @@ def get_predefined_cupti_metrics(name=""):
         name (str): cupti metri name.
     """
     if name not in predefined_cupti_metrics:
-        _ti_core.warn("Valid Taichi predefined metrics list (str):")
+        _ti_core.warn("Valid GsTaichi predefined metrics list (str):")
         for key in predefined_cupti_metrics:
             _ti_core.warn(f"    '{key}'")
         return None
@@ -253,7 +253,7 @@ def get_predefined_cupti_metrics(name=""):
 
 # Default metrics list
 default_cupti_metrics = [dram_bytes_sum]
-"""The metrics list, each is an instance of the :class:`~taichi.profiler.CuptiMetric`.
+"""The metrics list, each is an instance of the :class:`~gstaichi.profiler.CuptiMetric`.
 Default to `dram_bytes_sum`.
 """
 

{taichi → gstaichi}/profiler/kernel_profiler.py

@@ -2,9 +2,9 @@
 
 from contextlib import contextmanager
 
-from taichi._lib import core as _ti_core
-from taichi.lang import impl
-from taichi.profiler.kernel_metrics import default_cupti_metrics
+from gstaichi._lib import core as _ti_core
+from gstaichi.lang import impl
+from gstaichi.profiler.kernel_metrics import default_cupti_metrics
 
 
 class StatisticalResult:
@@ -40,16 +40,16 @@ class StatisticalResult:
 
 
 class KernelProfiler:
-    """Kernel profiler of Taichi.
+    """Kernel profiler of GsTaichi.
 
     Kernel profiler acquires kernel profiling records from backend, counts records in Python scope,
-    and prints the results to the console by :func:`~taichi.profiler.kernel_profiler.KernelProfiler.print_info`.
+    and prints the results to the console by :func:`~gstaichi.profiler.kernel_profiler.KernelProfiler.print_info`.
 
     ``KernelProfiler`` now support detailed low-level performance metrics (such as memory bandwidth consumption) in its advanced mode.
     This mode is only available for the CUDA backend with CUPTI toolkit, i.e. you need ``ti.init(kernel_profiler=True, arch=ti.cuda)``.
 
     Note:
-        For details about using CUPTI in Taichi, please visit https://docs.taichi-lang.org/docs/profiler#advanced-mode.
+        For details about using CUPTI in GsTaichi, please visit https://docs.taichi-lang.org/docs/profiler#advanced-mode.
     """
 
     def __init__(self):
@@ -63,14 +63,14 @@ class KernelProfiler:
     # public methods
 
     def set_kernel_profiler_mode(self, mode=False):
-        """Turn on or off :class:`~taichi.profiler.kernel_profiler.KernelProfiler`."""
+        """Turn on or off :class:`~gstaichi.profiler.kernel_profiler.KernelProfiler`."""
         if type(mode) is bool:
             self._profiling_mode = mode
         else:
             raise TypeError(f"Arg `mode` must be of type boolean. Type {type(mode)} is not supported.")
 
     def get_kernel_profiler_mode(self):
-        """Get status of :class:`~taichi.profiler.kernel_profiler.KernelProfiler`."""
+        """Get status of :class:`~gstaichi.profiler.kernel_profiler.KernelProfiler`."""
         return self._profiling_mode
 
     def set_toolkit(self, toolkit_name="default"):
@@ -98,7 +98,7 @@ class KernelProfiler:
         return self._total_time_ms / 1000  # ms to s
 
     def clear_info(self):
-        """Clear all records both in front-end :class:`~taichi.profiler.kernel_profiler.KernelProfiler` and back-end instance ``KernelProfilerBase``.
+        """Clear all records both in front-end :class:`~gstaichi.profiler.kernel_profiler.KernelProfiler` and back-end instance ``KernelProfilerBase``.
 
         Note:
             The values of ``self._profiling_mode`` and ``self._metric_list`` will not be cleared.
@@ -114,7 +114,7 @@ class KernelProfiler:
         return None
 
     def query_info(self, name):
-        """For docstring of this function, see :func:`~taichi.profiler.query_kernel_profiler_info`."""
+        """For docstring of this function, see :func:`~gstaichi.profiler.query_kernel_profiler_info`."""
         if self._check_not_turned_on_with_warning_message():
             return None
         self._update_records()  # kernel records
@@ -123,7 +123,7 @@ class KernelProfiler:
         return impl.get_runtime().prog.query_kernel_profile_info(name)
 
     def set_metrics(self, metric_list=default_cupti_metrics):
-        """For docstring of this function, see :func:`~taichi.profiler.set_kernel_profiler_metrics`."""
+        """For docstring of this function, see :func:`~gstaichi.profiler.set_kernel_profiler_metrics`."""
         if self._check_not_turned_on_with_warning_message():
             return None
         self._metric_list = metric_list
@@ -137,7 +137,7 @@ class KernelProfiler:
     def collect_metrics_in_context(self, metric_list=default_cupti_metrics):
         """This function is not exposed to user now.
 
-        For usage of this function, see :func:`~taichi.profiler.collect_kernel_profiler_metrics`.
+        For usage of this function, see :func:`~gstaichi.profiler.collect_kernel_profiler_metrics`.
         """
         if self._check_not_turned_on_with_warning_message():
             return None
@@ -148,13 +148,13 @@ class KernelProfiler:
         return None
 
     # mode of print_info
-    COUNT = "count"  # print the statistical results (min,max,avg time) of Taichi kernels.
-    TRACE = "trace"  # print the records of launched Taichi kernels with specific profiling metrics (time, memory load/store and core utilization etc.)
+    COUNT = "count"  # print the statistical results (min,max,avg time) of GsTaichi kernels.
+    TRACE = "trace"  # print the records of launched GsTaichi kernels with specific profiling metrics (time, memory load/store and core utilization etc.)
 
     def print_info(self, mode=COUNT):
-        """Print the profiling results of Taichi kernels.
+        """Print the profiling results of GsTaichi kernels.
 
-        For usage of this function, see :func:`~taichi.profiler.print_kernel_profiler_info`.
+        For usage of this function, see :func:`~gstaichi.profiler.print_kernel_profiler_info`.
 
         Args:
             mode (str): the way to print profiling results.
@@ -183,7 +183,7 @@ class KernelProfiler:
         return False
 
     def _clear_frontend(self):
-        """Clear member variables in :class:`~taichi.profiler.kernel_profiler.KernelProfiler`.
+        """Clear member variables in :class:`~gstaichi.profiler.kernel_profiler.KernelProfiler`.
 
         Note:
             The values of ``self._profiling_mode`` and ``self._metric_list`` will not be cleared.
@@ -340,7 +340,7 @@ _ti_kernel_profiler = KernelProfiler()
 
 
 def get_default_kernel_profiler():
-    """We have only one :class:`~taichi.profiler.kernelprofiler.KernelProfiler` instance(i.e. ``_ti_kernel_profiler``) now.
+    """We have only one :class:`~gstaichi.profiler.kernelprofiler.KernelProfiler` instance(i.e. ``_ti_kernel_profiler``) now.
 
     For ``KernelProfiler`` using ``CuptiToolkit``, GPU devices can only work in a certain configuration.
     Profiling mode and metrics are configured by the host(CPU) via CUPTI APIs, and device(GPU) will use
@@ -353,7 +353,7 @@ def get_default_kernel_profiler():
 
 
 def print_kernel_profiler_info(mode="count"):
-    """Print the profiling results of Taichi kernels.
+    """Print the profiling results of GsTaichi kernels.
 
     To enable this profiler, set ``kernel_profiler=True`` in ``ti.init()``.
     ``'count'`` mode: print the statistics (min,max,avg time) of launched kernels,
@@ -365,7 +365,7 @@ def print_kernel_profiler_info(mode="count"):
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(ti.cpu, kernel_profiler=True)
         >>> var = ti.field(ti.f32, shape=1)
@@ -382,9 +382,6 @@ def print_kernel_profiler_info(mode="count"):
         >>> ti.profiler.print_kernel_profiler_info('trace')
 
     Note:
-        Currently the result of `KernelProfiler` could be incorrect on OpenGL
-        backend due to its lack of support for `ti.sync()`.
-
         For advanced mode of `KernelProfiler`, please visit https://docs.taichi-lang.org/docs/profiler#advanced-mode.
     """
     get_default_kernel_profiler().print_info(mode)
@@ -403,7 +400,7 @@ def query_kernel_profiler_info(name):
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(ti.cpu, kernel_profiler=True)
         >>> n = 1024*1024
@@ -427,9 +424,6 @@ def query_kernel_profiler_info(name):
     Note:
         [1] To get the correct result, query_kernel_profiler_info() must be used in conjunction with
         clear_kernel_profiler_info().
-
-        [2] Currently the result of `KernelProfiler` could be incorrect on OpenGL
-        backend due to its lack of support for `ti.sync()`.
     """
     return get_default_kernel_profiler().query_info(name)
 
@@ -461,7 +455,7 @@ def set_kernel_profiler_toolkit(toolkit_name="default"):
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(arch=ti.cuda, kernel_profiler=True)
         >>> x = ti.field(ti.f32, shape=1024*1024)
@@ -488,11 +482,11 @@ def set_kernel_profiler_metrics(metric_list=default_cupti_metrics):
     """Set metrics that will be collected by the CUPTI toolkit.
 
     Args:
-        metric_list (list): a list of :class:`~taichi.profiler.CuptiMetric()` instances, default value: :data:`~taichi.profiler.kernel_metrics.default_cupti_metrics`.
+        metric_list (list): a list of :class:`~gstaichi.profiler.CuptiMetric()` instances, default value: :data:`~gstaichi.profiler.kernel_metrics.default_cupti_metrics`.
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(kernel_profiler=True, arch=ti.cuda)
         >>> ti.profiler.set_kernel_profiler_toolkit('cupti')
@@ -507,9 +501,9 @@ def set_kernel_profiler_metrics(metric_list=default_cupti_metrics):
         >>>     for i in x:
         >>>         y[None] += x[i]
 
-        >>> # In the case of not parameter, Taichi will print its pre-defined metrics list
+        >>> # In the case of not parameter, GsTaichi will print its pre-defined metrics list
         >>> ti.profiler.get_predefined_cupti_metrics()
-        >>> # get Taichi pre-defined metrics
+        >>> # get GsTaichi pre-defined metrics
         >>> profiling_metrics = ti.profiler.get_predefined_cupti_metrics('shared_access')
 
         >>> global_op_atom = ti.profiler.CuptiMetric(
@@ -536,11 +530,11 @@ def collect_kernel_profiler_metrics(metric_list=default_cupti_metrics):
     """Set temporary metrics that will be collected by the CUPTI toolkit within this context.
 
     Args:
-        metric_list (list): a list of :class:`~taichi.profiler.CuptiMetric()` instances, default value: :data:`~taichi.profiler.kernel_metrics.default_cupti_metrics`.
+        metric_list (list): a list of :class:`~gstaichi.profiler.CuptiMetric()` instances, default value: :data:`~gstaichi.profiler.kernel_metrics.default_cupti_metrics`.
 
     Example::
 
-        >>> import taichi as ti
+        >>> import gstaichi as ti
 
         >>> ti.init(kernel_profiler=True, arch=ti.cuda)
         >>> ti.profiler.set_kernel_profiler_toolkit('cupti')
@@ -555,9 +549,9 @@ def collect_kernel_profiler_metrics(metric_list=default_cupti_metrics):
         >>>     for i in x:
         >>>         y[None] += x[i]
 
-        >>> # In the case of not parameter, Taichi will print its pre-defined metrics list
+        >>> # In the case of not parameter, GsTaichi will print its pre-defined metrics list
         >>> ti.profiler.get_predefined_cupti_metrics()
-        >>> # get Taichi pre-defined metrics
+        >>> # get GsTaichi pre-defined metrics
         >>> profiling_metrics = ti.profiler.get_predefined_cupti_metrics('device_utilization')
 
         >>> global_op_atom = ti.profiler.CuptiMetric(

{taichi → gstaichi}/profiler/memory_profiler.py

@@ -1,6 +1,6 @@
 # type: ignore
 
-from taichi.lang.impl import get_runtime
+from gstaichi.lang.impl import get_runtime
 
 
 def print_memory_profiler_info():

{taichi → gstaichi}/profiler/scoped_profiler.py

@@ -1,6 +1,6 @@
 # type: ignore
 
-from taichi._lib import core as _ti_core
+from gstaichi._lib import core as _ti_core
 
 
 def print_scoped_profiler_info():
@@ -12,7 +12,7 @@ def print_scoped_profiler_info():
 
     Example::
 
-            >>> import taichi as ti
+            >>> import gstaichi as ti
             >>> ti.init(arch=ti.cpu)
             >>> var = ti.field(ti.f32, shape=1)
             >>> @ti.kernel