warp-lang 1.2.0__py3-none-manylinux2014_aarch64.whl → 1.2.1__py3-none-manylinux2014_aarch64.whl

warp/config.py

@@ -7,7 +7,7 @@
 
 from typing import Optional
 
-version: str = "1.2.0"
+version: str = "1.2.1"
 
 verify_fp: bool = False  # verify inputs and outputs are finite after each launch
 verify_cuda: bool = False  # if true will check CUDA errors after each kernel launch / memory operation

warp/context.py

@@ -1488,8 +1488,9 @@ class Module:
 
                 # functions source
                 for function in module.functions.values():
-                    # include all overloads
-                    for sig, func in function.user_overloads.items():
+                    # include all concrete and generic overloads
+                    overloads = itertools.chain(function.user_overloads.items(), function.user_templates.items())
+                    for sig, func in overloads:
                         # signature
                         ch.update(bytes(sig, "utf-8"))
 
@@ -3632,6 +3633,9 @@ def is_mempool_access_supported(target_device: Devicelike, peer_device: Deviceli
 
     init()
 
+    target_device = runtime.get_device(target_device)
+    peer_device = runtime.get_device(peer_device)
+
     return target_device.is_mempool_supported and is_peer_access_supported(target_device, peer_device)
 
 

warp/types.py

@@ -1650,9 +1650,10 @@ class array(Array):
             try:
                 # Performance note: try first, ask questions later
                 device = warp.context.runtime.get_device(device)
-            except:
-                warp.context.init()
-                raise
+            except Exception:
+                # Fallback to using the public API for retrieving the device,
+                # which takes take of initializing Warp if needed.
+                device = warp.context.get_device(device)
 
             if device.is_cuda:
                 desc = data.__cuda_array_interface__
@@ -1779,9 +1780,10 @@ class array(Array):
         try:
             # Performance note: try first, ask questions later
             device = warp.context.runtime.get_device(device)
-        except:
-            warp.context.init()
-            raise
+        except Exception:
+            # Fallback to using the public API for retrieving the device,
+            # which takes take of initializing Warp if needed.
+            device = warp.context.get_device(device)
 
         if device.is_cpu and not copy and not pinned:
             # reference numpy memory directly
@@ -1805,9 +1807,10 @@ class array(Array):
         try:
             # Performance note: try first, ask questions later
             device = warp.context.runtime.get_device(device)
-        except:
-            warp.context.init()
-            raise
+        except Exception:
+            # Fallback to using the public API for retrieving the device,
+            # which takes take of initializing Warp if needed.
+            device = warp.context.get_device(device)
 
         check_array_shape(shape)
         ndim = len(shape)
@@ -1852,9 +1855,10 @@ class array(Array):
         try:
             # Performance note: try first, ask questions later
             device = warp.context.runtime.get_device(device)
-        except:
-            warp.context.init()
-            raise
+        except Exception:
+            # Fallback to using the public API for retrieving the device,
+            # which takes take of initializing Warp if needed.
+            device = warp.context.get_device(device)
 
         check_array_shape(shape)
         ndim = len(shape)

{warp_lang-1.2.0.dist-info → warp_lang-1.2.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: warp-lang
-Version: 1.2.0
+Version: 1.2.1
 Summary: A Python framework for high-performance simulation and graphics programming
 Author-email: NVIDIA <mmacklin@nvidia.com>
 License: NVIDIA Software License
@@ -108,7 +108,10 @@ print(lengths)
 
 ## Running Examples
 
-The `examples` directory contains a number of scripts that show how to implement different simulation methods using the Warp API. Most examples will generate USD files containing time-sampled animations (stored in the current working directory). Before running examples, users should ensure that the ``usd-core``, ``matplotlib``, and ``pyglet`` packages are installed using:
+The [warp/examples](./warp/examples/) directory contains a number of scripts categorized under different subdirectories
+that show how to implement different simulation methods using the Warp API.
+Most examples will generate USD files containing time-sampled animations (stored in the current working directory).
+Before running examples, users should ensure that the ``usd-core``, ``matplotlib``, and ``pyglet`` packages are installed using:
 
     pip install usd-core matplotlib pyglet
 

{warp_lang-1.2.0.dist-info → warp_lang-1.2.1.dist-info}/RECORD

@@ -4,9 +4,9 @@ warp/build.py,sha256=LmRSU00pdlN1P31XS1_FhXgu_k7GzKkAzPd0LacZClo,3690
 warp/build_dll.py,sha256=p9eTXMhKbI0uTkpqeY5DxOmVhupfaykM-a-re6fXBXg,15896
 warp/builtins.py,sha256=AlHPF8WTvTOF2DGo8nQJ_OmZ5nzkoTi_qPLDlc6fPg0,122322
 warp/codegen.py,sha256=MJC5a63p-eXelMivb0Tyg6hU_ZCuqx8nMDSE7Om0BV0,103205
-warp/config.py,sha256=Spi7LsenwhcJubEYYMAmlzYEoVdGQ1qPNJjeqEICsv8,1879
+warp/config.py,sha256=uimmJvTOD_xfYuBo4rExTn4izyvbzXJqsURttRkqwXA,1879
 warp/constants.py,sha256=ckfEqwbg109RJDW98SILe_yfOBEFolOATVjt0mU2py4,1321
-warp/context.py,sha256=J0w1vzGBh1r0JXz-A0bf1rytV5E5WtZEv2gPGxt4yzg,213400
+warp/context.py,sha256=_vyVfFBpo-teFiy_n26F0If5GU5cVfU7wWksbkyYckM,213618
 warp/dlpack.py,sha256=z242iSS70YF-5IP-GVE83R7HbHB5aRMmiYiQlAbyNis,16568
 warp/fabric.py,sha256=__Dbd9TmF5bqxUh411dpiAGxrlsfDAwZ8benjXN3Jco,11316
 warp/jax.py,sha256=QKJXfpz1jjaRD-Yp4C4wC4jUv8jHj0en3MW0VXKq0ZA,5280
@@ -15,10 +15,10 @@ warp/sparse.py,sha256=57Ar8OknSZ5P0apLLm_hnhxWlj0bh6U-mbu2qfYDcis,42356
 warp/stubs.py,sha256=wS3_3Joiutw9cecEzae5GyYHIOUH6wl8CdLeozueqks,57977
 warp/tape.py,sha256=hH3Qb9CiKHmq5OUIHRTZjZdYDXR2noHFNiB9HrIl5ZU,47864
 warp/torch.py,sha256=5hRCNL5jJzrh3hAcn1DdYfTvdOA5fvBy0VQnHviZpkE,11760
-warp/types.py,sha256=1OlIRSbwHFDbOPcnCxOi5lQcGCY6kn2Iwp-5kuntML8,169680
+warp/types.py,sha256=JViQDLNFdeaGyqklXWGMyHYrgGK1_6UKOBPbJSqkEZM,170284
 warp/utils.py,sha256=UoTCl9kGlPJBlsgvhM726VsnMkUsQSamxLQ0vgR1h5s,34148
-warp/bin/warp-clang.so,sha256=hcnCbX_yQWdlKKhPJ1-divB8qY-MRmE5nIW6C1Mkj3g,52883864
-warp/bin/warp.so,sha256=q2tvdkJ262_iBQ1L3ahQat33gGvG02eeWROrXFazg3A,124687832
+warp/bin/warp-clang.so,sha256=2a-D7s5Aku1pR96k6gfDbTsPnTU1nGenly-rbBBxoqY,52851096
+warp/bin/warp.so,sha256=MqGizmXxBAO6rO48PiH53Uq61aUzFMUnZGrH95FBBN0,124687832
 warp/examples/__init__.py,sha256=ul3vRFtQ17_zIZIWJaCnCY28KZTGOUHQA4wA_IW_qaw,606
 warp/examples/browse.py,sha256=6v7F0vpZz5ZtYGkg8hIpxhbIZ2ECv46cIKAmKVtV1S4,848
 warp/examples/assets/bear.usd,sha256=1SK7s-zpF3Ypzcc8Wt861o174Pe-xMuJpvosACXVNUk,226238
@@ -352,8 +352,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=5z6oi9tsk2fOxhP4fx8hXRGveDTLM9fcESfzMWxJLFs,20998
-warp_lang-1.2.0.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
-warp_lang-1.2.0.dist-info/METADATA,sha256=JEx5lbyT9BHlLuQzlBUxBXLSE79bcDchLWtHJi6c15I,20549
-warp_lang-1.2.0.dist-info/WHEEL,sha256=Jnjojrf4EGkJZw4ToGTqSAdpEYbK8-OHT7fzFXvzFsg,111
-warp_lang-1.2.0.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
-warp_lang-1.2.0.dist-info/RECORD,,
+warp_lang-1.2.1.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
+warp_lang-1.2.1.dist-info/METADATA,sha256=INLOWhLJDd8U5LCSIPBYnF791H3AH37SxdPKSpr3eMk,20615
+warp_lang-1.2.1.dist-info/WHEEL,sha256=Jnjojrf4EGkJZw4ToGTqSAdpEYbK8-OHT7fzFXvzFsg,111
+warp_lang-1.2.1.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
+warp_lang-1.2.1.dist-info/RECORD,,