gridpoints 1.0.1__tar.gz → 1.0.3__tar.gz

{gridpoints-1.0.1 → gridpoints-1.0.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gridpoints
-Version: 1.0.1
+Version: 1.0.3
 Summary: fast multidimensional sorting tool for point clouds using the monotonic lagrangian grid 
 Author: Armand de Cacqueray
 License: MIT

{gridpoints-1.0.1 → gridpoints-1.0.3}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "gridpoints"
-version = "1.0.1"
+version = "1.0.3"
 description = "fast multidimensional sorting tool for point clouds using the monotonic lagrangian grid "
 readme = "README.md"
 requires-python = ">=3.9"

{gridpoints-1.0.1 → gridpoints-1.0.3}/src/griddemo/__init__.py

@@ -7,4 +7,4 @@ __all__ = (
     print_zoom,
 )
 
-__version__ = "1.0.1"
+__version__ = "1.0.3"

{gridpoints-1.0.1 → gridpoints-1.0.3}/src/griddemo/quick_demo.py

@@ -61,7 +61,10 @@ def show_result(X, animate=False):
         print("Missing dependencies for the demo. pip install gridsort[demo]")
     from squarenet import SquareNet
     import numpy as np
-    X = np.asarray(X)
+    try:
+        X = np.asarray(X)
+    except:
+        print("Array still on GPU, move it back to gpu first befor showing e.g. show_result(X.cpu())")
     sn = SquareNet(gridshape = X.shape[:-1])
     sn.pointsmaped = X
     sn.plot(style = "mesh", animate = animate, save = animate)
@@ -83,7 +86,10 @@ def print_zoom(Ygrid, center, side=1):
         Number of cells shown on each side of the center.
     """
     import numpy as np
-    Ygrid = np.asarray(Ygrid)
+    try:
+        Ygrid = np.asarray(Ygrid)
+    except:
+        print("Array still on GPU, move it back to gpu first befor showing e.g. print_zoom(X.cpu())")
     ndim = Ygrid.ndim - 1
     if ndim not in (2, 3):
         raise ValueError("printzoom only for 2D/3D grids")

{gridpoints-1.0.1 → gridpoints-1.0.3}/src/gridpoints/__init__.py

@@ -8,4 +8,4 @@ __all__ = (
     "invert_permutation",
 )
 
-__version__ = "1.0.1"
+__version__ = "1.0.3"

{gridpoints-1.0.1 → gridpoints-1.0.3}/src/gridpoints/array_libs.py

@@ -43,9 +43,11 @@ def astype(X, dtype_str, reference_array=None):
 
 
 def inplace_sort(X, axis):
-    _, backend_name = get_backend(X)
+    xp, backend_name = get_backend(X)
     if backend_name == 'torch':
         X.copy_(X.sort(dim=axis)[0])
+    if backend_name == "cupy":
+        X[...] = xp.sort(xp.ascontiguousarray(X), axis=axis)
     else:
         X.sort(axis=axis)
 

{gridpoints-1.0.1 → gridpoints-1.0.3}/src/gridpoints.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gridpoints
-Version: 1.0.1
+Version: 1.0.3
 Summary: fast multidimensional sorting tool for point clouds using the monotonic lagrangian grid 
 Author: Armand de Cacqueray
 License: MIT