warp-lang 1.3.2__py3-none-manylinux2014_aarch64.whl → 1.3.3__py3-none-manylinux2014_aarch64.whl

warp/config.py

@@ -7,7 +7,7 @@
 
 from typing import Optional
 
-version: str = "1.3.2"
+version: str = "1.3.3"
 """Warp version string"""
 
 verify_fp: bool = False

warp/tests/test_array.py

@@ -2295,6 +2295,39 @@ def test_array_from_numpy(test, device):
     assert_np_equal(result.numpy(), expected.numpy())
 
 
+def test_array_aliasing_from_numpy(test, device):
+    device = wp.get_device(device)
+    assert device.is_cpu
+
+    a_np = np.ones(8, dtype=np.int32)
+    a_wp = wp.array(a_np, dtype=int, copy=False, device=device)
+    test.assertIs(a_wp._ref, a_np)  # check that some ref is kept to original array
+    test.assertEqual(a_wp.ptr, a_np.ctypes.data)
+
+    a_np_2 = a_wp.numpy()
+    test.assertTrue((a_np_2 == 1).all())
+
+    # updating source array should update aliased array
+    a_np.fill(2)
+    test.assertTrue((a_np_2 == 2).all())
+
+    # trying to alias from a different type should do a copy
+    # do it twice to check that the copy buffer is not being reused for different arrays
+
+    b_np = np.ones(8, dtype=np.int64)
+    c_np = np.zeros(8, dtype=np.int64)
+    b_wp = wp.array(b_np, dtype=int, copy=False, device=device)
+    c_wp = wp.array(c_np, dtype=int, copy=False, device=device)
+
+    test.assertNotEqual(b_wp.ptr, b_np.ctypes.data)
+    test.assertNotEqual(b_wp.ptr, c_wp.ptr)
+
+    b_np_2 = b_wp.numpy()
+    c_np_2 = c_wp.numpy()
+    test.assertTrue((b_np_2 == 1).all())
+    test.assertTrue((c_np_2 == 0).all())
+
+
 def test_array_from_cai(test, device):
     import torch
 
@@ -2447,6 +2480,7 @@ add_function_test(TestArray, "test_array_of_structs_grad", test_array_of_structs
 add_function_test(TestArray, "test_array_of_structs_from_numpy", test_array_of_structs_from_numpy, devices=devices)
 add_function_test(TestArray, "test_array_of_structs_roundtrip", test_array_of_structs_roundtrip, devices=devices)
 add_function_test(TestArray, "test_array_from_numpy", test_array_from_numpy, devices=devices)
+add_function_test(TestArray, "test_array_aliasing_from_numpy", test_array_aliasing_from_numpy, devices=["cpu"])
 
 add_function_test(TestArray, "test_direct_from_numpy", test_direct_from_numpy, devices=["cpu"])
 add_function_test(TestArray, "test_kernel_array_from_ptr", test_kernel_array_from_ptr, devices=devices)

warp/tests/test_volume.py

@@ -843,6 +843,33 @@ def test_volume_from_numpy(test, device):
     test.assertIsNone(sphere_vdb_array.deleter)
 
 
+def test_volume_from_numpy_3d(test, device):
+    # Volume.allocate_from_tiles() is only available with CUDA
+    mins = np.array([-3.0, -3.0, -3.0])
+    voxel_size = 0.2
+    maxs = np.array([3.0, 3.0, 3.0])
+    nums = np.ceil((maxs - mins) / (voxel_size)).astype(dtype=int)
+    centers = np.array([[-1.0, -1.0, -1.0], [0.0, 0.0, 0.0], [1.0, 1.0, 1.0]])
+    rad = 2.5
+    sphere_sdf_np = np.zeros(tuple(nums) + (3,))
+    for x in range(nums[0]):
+        for y in range(nums[1]):
+            for z in range(nums[2]):
+                for k in range(3):
+                    pos = mins + voxel_size * np.array([x, y, z])
+                    dis = np.linalg.norm(pos - centers[k])
+                    sphere_sdf_np[x, y, z, k] = dis - rad
+    sphere_vdb = wp.Volume.load_from_numpy(
+        sphere_sdf_np, mins, voxel_size, (rad + 3.0 * voxel_size,) * 3, device=device
+    )
+
+    test.assertNotEqual(sphere_vdb.id, 0)
+
+    sphere_vdb_array = sphere_vdb.array()
+    test.assertEqual(sphere_vdb_array.dtype, wp.uint8)
+    test.assertIsNone(sphere_vdb_array.deleter)
+
+
 def test_volume_aniso_transform(test, device):
     # XY-rotation + z scale
     transform = [
@@ -894,6 +921,9 @@ add_function_test(TestVolume, "test_volume_introspection", test_volume_introspec
 add_function_test(
     TestVolume, "test_volume_from_numpy", test_volume_from_numpy, devices=get_selected_cuda_test_devices()
 )
+add_function_test(
+    TestVolume, "test_volume_from_numpy_3d", test_volume_from_numpy_3d, devices=get_selected_cuda_test_devices()
+)
 add_function_test(
     TestVolume, "test_volume_aniso_transform", test_volume_aniso_transform, devices=get_selected_cuda_test_devices()
 )

warp/types.py

@@ -1601,6 +1601,9 @@ class array(Array):
         self._array_interface = None
         self.is_transposed = False
 
+        # reference to other array
+        self._ref = None
+
         # canonicalize dtype
         if dtype == int:
             dtype = int32
@@ -1652,9 +1655,6 @@ class array(Array):
                 if requires_grad:
                     self._alloc_grad()
 
-        # reference to other array
-        self._ref = None
-
     def _init_from_data(self, data, dtype, shape, device, copy, pinned):
         if not hasattr(data, "__len__"):
             raise RuntimeError(f"Data must be a sequence or array, got scalar {data}")
@@ -2991,7 +2991,7 @@ class Mesh:
 
         Args:
             points (:class:`warp.array`): Array of vertex positions of type :class:`warp.vec3`
-            indices (:class:`warp.array`): Array of triangle indices of type :class:`warp.int32`, should be a 1d array with shape (num_tris, 3)
+            indices (:class:`warp.array`): Array of triangle indices of type :class:`warp.int32`, should be a 1d array with shape (num_tris * 3)
             velocities (:class:`warp.array`): Array of vertex velocities of type :class:`warp.vec3` (optional)
             support_winding_number (bool): If true the mesh will build additional datastructures to support `wp.mesh_query_point_sign_winding_number()` queries
         """
@@ -3529,8 +3529,9 @@ class Volume:
         )
         if hasattr(bg_value, "__len__"):
             # vec3, assuming the numpy array is 4D
-            padded_array = np.array((target_shape[0], target_shape[1], target_shape[2], 3), dtype=np.single)
-            padded_array[:, :, :, :] = np.array(bg_value)
+            padded_array = np.full(
+                shape=(target_shape[0], target_shape[1], target_shape[2], 3), fill_value=bg_value, dtype=np.single
+            )
             padded_array[0 : ndarray.shape[0], 0 : ndarray.shape[1], 0 : ndarray.shape[2], :] = ndarray
         else:
             padded_amount = (

{warp_lang-1.3.2.dist-info → warp_lang-1.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: warp-lang
-Version: 1.3.2
+Version: 1.3.3
 Summary: A Python framework for high-performance simulation and graphics programming
 Author-email: NVIDIA Corporation <mmacklin@nvidia.com>
 License: NVIDIA Software License
@@ -77,9 +77,9 @@ the `pip install` command, e.g.
 
 | Platform        | Install Command                                                                                                               |
 | --------------- | ----------------------------------------------------------------------------------------------------------------------------- |
-| Linux aarch64   | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.2/warp_lang-1.3.2+cu11-py3-none-manylinux2014_aarch64.whl` |
-| Linux x86-64    | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.2/warp_lang-1.3.2+cu11-py3-none-manylinux2014_x86_64.whl`  |
-| Windows x86-64  | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.2/warp_lang-1.3.2+cu11-py3-none-win_amd64.whl`             |
+| Linux aarch64   | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.3/warp_lang-1.3.3+cu11-py3-none-manylinux2014_aarch64.whl` |
+| Linux x86-64    | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.3/warp_lang-1.3.3+cu11-py3-none-manylinux2014_x86_64.whl`  |
+| Windows x86-64  | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.3/warp_lang-1.3.3+cu11-py3-none-win_amd64.whl`             |
 
 The `--force-reinstall` option may need to be used to overwrite a previous installation.
 

{warp_lang-1.3.2.dist-info → warp_lang-1.3.3.dist-info}/RECORD

@@ -5,7 +5,7 @@ warp/build.py,sha256=N2-SDksxkffUkzV97dtLW2Wjqd5ERfgwpqzVoe8GNRw,3904
 warp/build_dll.py,sha256=p9eTXMhKbI0uTkpqeY5DxOmVhupfaykM-a-re6fXBXg,15896
 warp/builtins.py,sha256=2eqAYDotCCaZ2pdDLNWLSqTM_qnMb9O9O-OfkZpj4Yg,142275
 warp/codegen.py,sha256=bwWinPrKbCBm_2xLywCJDAgpWhEoo4i4zmeCTklCWJo,116383
-warp/config.py,sha256=MrmOHA0ZkWoxm3itvgK5YbxEqKZCnz_5lqe-MFUIT1I,2808
+warp/config.py,sha256=Wc2uujoeByAWRwCusQR0_YCesd2VyWnzf0MST70hwl0,2808
 warp/constants.py,sha256=ckfEqwbg109RJDW98SILe_yfOBEFolOATVjt0mU2py4,1321
 warp/context.py,sha256=wHh56A2qHp9CKWSpN8-P_sQNWBfOS0RMTXMMVqi1Jjs,226312
 warp/dlpack.py,sha256=Ol_wrve9k6ipSarDpNGlHoEH4KRFJucz09Aqvtgccs8,16777
@@ -16,7 +16,7 @@ warp/sparse.py,sha256=APAp2M2FNpmOaQkI-ddxjluEcwJ-Ua2plNwq8IdP_Gw,64640
 warp/stubs.py,sha256=0TKzfYAuQR9_gGk3TRzJhvHdjSUkAL2KlBXdD6Q73-U,51776
 warp/tape.py,sha256=w96cmAIWTEvmUCvFVoNp3qMO2CCI934CjwE-7buPpDw,48988
 warp/torch.py,sha256=jv3SYQHVKsiCzenlWUVmVs5mLYfsWcdTW446IqvEwVA,14505
-warp/types.py,sha256=3k3ZXjm0nL18Fkhpcu849D4chflB9tdMqNCr4F2e6jU,176639
+warp/types.py,sha256=L18PVTGCluXFX6smghknfhpbtMcI8TOPaptzD1MsiTY,176638
 warp/utils.py,sha256=wwyxRAujKkpdm-sB4pX5upAEgLv5RymOi6pQoNOUs-I,34663
 warp/bin/warp-clang.so,sha256=hcnCbX_yQWdlKKhPJ1-divB8qY-MRmE5nIW6C1Mkj3g,52883864
 warp/bin/warp.so,sha256=qCu2su90l7EYpbREv9xUexXn2yTZtNNu062CAT8nhNQ,135657384
@@ -254,7 +254,7 @@ warp/tests/disabled_kinematics.py,sha256=B8M_ih-JAobZA6-zPmtqSYGBCbKRyjtW2obR82i
 warp/tests/run_coverage_serial.py,sha256=UjcqNXQixQk13U3Lckm2OlmqPKKzUk_KaBPyHzxkdT4,1045
 warp/tests/test_adam.py,sha256=u5efgdobQgGdJilsCALB0TKZO5M5T8JDQ8pD3L0CApE,5361
 warp/tests/test_arithmetic.py,sha256=ISNFQYSSRIpl7vEJNxOR0AqfWegdewYZznJ588ZAwX4,43840
-warp/tests/test_array.py,sha256=I4ieXdo9ZSyz_hIW5ILZvzlCeXX7xCpxm68CYrUAAHE,87357
+warp/tests/test_array.py,sha256=DZhZYdrwrA1lG-rvSJ3FaR1iYU9auCQvxmChsCzqDKo,88589
 warp/tests/test_array_reduce.py,sha256=LH_nUdEPuc-FDRelqgO4c5wK5lU6SPpR4s9IcA5Wktw,4678
 warp/tests/test_async.py,sha256=YKFNJVT7R3gGXPuHAXBkZ8VIzBrfnqUsj-tXyHKUgbQ,26498
 warp/tests/test_atomic.py,sha256=jnzNwf6EsECVMfqf-wkgwzeqNa0WxzGLEILe8s-Zi9I,5343
@@ -341,7 +341,7 @@ warp/tests/test_vec.py,sha256=VRzPlnNwOX-AIoQpxNHzSirjTrTY8aPo7rFsPL-t0hc,40808
 warp/tests/test_vec_lite.py,sha256=c1yKwqWSSzCHUd2Ft1TSNpUpqrHAwF2rWBjQj5nDZQ4,2370
 warp/tests/test_vec_scalar_ops.py,sha256=j_1OW-uCcxayC3DsuFBeypW2lgotQZlGJjA5JAWOjew,88042
 warp/tests/test_verify_fp.py,sha256=YVBgLM0CTGMUnMwNYZOcj-BU2sz2zhItfY4RYutKd28,2337
-warp/tests/test_volume.py,sha256=tK4q4sc0vk4P-p2BLaAvM9BuaKZ0XXCaZNpGIib3rD8,34932
+warp/tests/test_volume.py,sha256=GpMZQA-iFS57AWrWhrJtZ0Z7e6B_YCVm_7yJhC2ILlU,36125
 warp/tests/test_volume_write.py,sha256=_QhIY9WotnS_NQD00g4JR16p3JC_vLegiD_PXCG_UIQ,11525
 warp/tests/unittest_serial.py,sha256=VXuwpkIo2tBgNnphZ4GDxFSHUVMupgGQWuDZFb64qr8,1134
 warp/tests/unittest_suites.py,sha256=XfVRlR4IYKI03gDl9npPk-s5gMTsAJnBrq_h5LrsXhM,13645
@@ -362,8 +362,8 @@ warp/thirdparty/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 warp/thirdparty/appdirs.py,sha256=2UXPK-AkDGoN5APeqAMMYkkfgf3Q_aUEZxcfkIxlQn0,24254
 warp/thirdparty/dlpack.py,sha256=OzMyVi5uQU01n0yEquDotgYvEAmfXT93QaIV4p4QqB4,4273
 warp/thirdparty/unittest_parallel.py,sha256=-FgbXrKiQ_RDj7YXYyhpRIPC3OmvJEnjvGkN99cqQA0,20888
-warp_lang-1.3.2.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
-warp_lang-1.3.2.dist-info/METADATA,sha256=lETJFLGWvbHc7_S95qG3xmDkoSxMgqvMUpxxRW6Y-s8,23145
-warp_lang-1.3.2.dist-info/WHEEL,sha256=x1fNTIFmrgdGi9cv8BF6yp8eLbJ9uVVTmpMQAxKGwn0,111
-warp_lang-1.3.2.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
-warp_lang-1.3.2.dist-info/RECORD,,
+warp_lang-1.3.3.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
+warp_lang-1.3.3.dist-info/METADATA,sha256=O17N1jUvTke6VTIPVPbiHDR32lG_2tHx5S-cK77kht4,23145
+warp_lang-1.3.3.dist-info/WHEEL,sha256=x1fNTIFmrgdGi9cv8BF6yp8eLbJ9uVVTmpMQAxKGwn0,111
+warp_lang-1.3.3.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
+warp_lang-1.3.3.dist-info/RECORD,,