warp-lang 1.3.0__py3-none-macosx_10_13_universal2.whl → 1.3.1__py3-none-macosx_10_13_universal2.whl

warp/builtins.py

@@ -35,7 +35,7 @@ def sametypes_create_value_func(default):
             return default
 
         if not sametypes(arg_types):
-            raise RuntimeError(f"Input types must be the same, found: {[type_repr(t) for t in arg_types]}")
+            raise RuntimeError(f"Input types must be the same, got {[type_repr(t) for t in arg_types.values()]}")
 
         arg_type_0 = next(iter(arg_types.values()))
         return arg_type_0
@@ -414,7 +414,7 @@ def scalar_sametypes_value_func(arg_types: Mapping[str, type], arg_values: Mappi
         return Scalar
 
     if not sametypes(arg_types):
-        raise RuntimeError(f"Input types must be exactly the same, {list(arg_types)}")
+        raise RuntimeError(f"Input types must be exactly the same, got {[type_repr(t) for t in arg_types.values()]}")
 
     return scalar_infer_type(arg_types)
 

warp/config.py

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

warp/context.py

@@ -4778,7 +4778,7 @@ def launch(
 
         # detect illegal inter-kernel read/write access patterns if verification flag is set
         if warp.config.verify_autograd_array_access:
-            runtime.tape.check_kernel_array_access(kernel, fwd_args)
+            runtime.tape._check_kernel_array_access(kernel, fwd_args)
 
 
 def synchronize():

warp/tape.py

@@ -197,7 +197,7 @@ class Tape:
         else:
             self.scopes.append((len(self.launches), None, None))
 
-    def check_kernel_array_access(self, kernel, args):
+    def _check_kernel_array_access(self, kernel, args):
         """Detect illegal inter-kernel write after read access patterns during launch capture"""
         adj = kernel.adj
         kernel_name = adj.fun_name
@@ -256,7 +256,7 @@ class Tape:
         self.scopes = []
         self.zero()
         if wp.config.verify_autograd_array_access:
-            self.reset_array_read_flags()
+            self._reset_array_read_flags()
 
     def zero(self):
         """
@@ -271,7 +271,7 @@ class Tape:
                 else:
                     g.zero_()
 
-    def reset_array_read_flags(self):
+    def _reset_array_read_flags(self):
         """
         Reset all recorded array read flags to False
         """

warp/tests/test_codegen.py

@@ -467,6 +467,28 @@ def test_error_collection_construct(test, device):
         wp.launch(kernel, dim=1)
 
 
+def test_error_unmatched_arguments(test, device):
+    def kernel_1_fn():
+        a = 1 * 1.0
+
+    def kernel_2_fn():
+        x = wp.dot(wp.vec2(1.0, 2.0), wp.vec2h(wp.float16(1.0), wp.float16(2.0)))
+
+    kernel = wp.Kernel(func=kernel_1_fn)
+    with test.assertRaisesRegex(
+        RuntimeError,
+        r"Input types must be the same, got \['int32', 'float32'\]",
+    ):
+        wp.launch(kernel, dim=1)
+
+    kernel = wp.Kernel(func=kernel_2_fn)
+    with test.assertRaisesRegex(
+        RuntimeError,
+        r"Input types must be exactly the same, got \[\"vector\(length=2, dtype=<class 'warp.types.float32'>\)\", \"vector\(length=2, dtype=<class 'warp.types.float16'>\)\"\]",
+    ):
+        wp.launch(kernel, dim=1)
+
+
 @wp.kernel
 def test_call_syntax():
     expected_pow = 16.0
@@ -618,6 +640,9 @@ add_function_test(TestCodeGen, func=test_error_global_var, name="test_error_glob
 add_function_test(
     TestCodeGen, func=test_error_collection_construct, name="test_error_collection_construct", devices=devices
 )
+add_function_test(
+    TestCodeGen, func=test_error_unmatched_arguments, name="test_error_unmatched_arguments", devices=devices
+)
 
 add_kernel_test(TestCodeGen, name="test_call_syntax", kernel=test_call_syntax, dim=1, devices=devices)
 

{warp_lang-1.3.0.dist-info → warp_lang-1.3.1.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: warp-lang
-Version: 1.3.0
+Version: 1.3.1
 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.0/warp_lang-1.3.0+cu11-py3-none-manylinux2014_aarch64.whl` |
-| Linux x86-64    | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.0/warp_lang-1.3.0+cu11-py3-none-manylinux2014_x86_64.whl`  |
-| Windows x86-64  | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.0/warp_lang-1.3.0+cu11-py3-none-win_amd64.whl`             |
+| Linux aarch64   | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.1/warp_lang-1.3.1+cu11-py3-none-manylinux2014_aarch64.whl` |
+| Linux x86-64    | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.1/warp_lang-1.3.1+cu11-py3-none-manylinux2014_x86_64.whl`  |
+| Windows x86-64  | `pip install https://github.com/NVIDIA/warp/releases/download/v1.3.1/warp_lang-1.3.1+cu11-py3-none-win_amd64.whl`             |
 
 The `--force-reinstall` option may need to be used to overwrite a previous installation.
 

{warp_lang-1.3.0.dist-info → warp_lang-1.3.1.dist-info}/RECORD

@@ -3,18 +3,18 @@ warp/__init__.pyi,sha256=Vxal9LM-q5ztScfRhrM4INpm5LJV6fN6B-H2NI5pkyU,21
 warp/autograd.py,sha256=t6MKb6YabHAOIoBgNiYVSQRyj7eAJnWZ2xZy1sF0FPQ,33588
 warp/build.py,sha256=N2-SDksxkffUkzV97dtLW2Wjqd5ERfgwpqzVoe8GNRw,3904
 warp/build_dll.py,sha256=p9eTXMhKbI0uTkpqeY5DxOmVhupfaykM-a-re6fXBXg,15896
-warp/builtins.py,sha256=w7KaAhXB_AV1-jbOC4_E5WH_RjUKf7gl577gGEItj6s,141752
+warp/builtins.py,sha256=si-PK9XhF32NMpCv86GYpvMvHYNAmtSi3-gwWICS2iA,141789
 warp/codegen.py,sha256=w6dzi9P5IMhF22IyLUsS-TH5kls-jjV_y4U8TLVvVC8,117159
-warp/config.py,sha256=osH0uv4yiAXiwR4xnqH1Ijd2GJpzpW-QE2Ou_4t5t-0,2808
+warp/config.py,sha256=WoFdUnKmxsTlb-ox6yy7YsWaEQvBVFlTyKrodfZ_1sk,2808
 warp/constants.py,sha256=ckfEqwbg109RJDW98SILe_yfOBEFolOATVjt0mU2py4,1321
-warp/context.py,sha256=_XYONCAZQRgJIHWwsx_02UjurSYCww9p6QrL5fs2zPI,223970
+warp/context.py,sha256=E7xoo0pWyEYFKT-Bi64nLwpE6gMWBZFsJ233LEbsJgA,223971
 warp/dlpack.py,sha256=Ol_wrve9k6ipSarDpNGlHoEH4KRFJucz09Aqvtgccs8,16777
 warp/fabric.py,sha256=qWOomvuhM2y3EzScsgaj_Dxvdt0wJPfFu4jUUG1JALg,11439
 warp/jax.py,sha256=QKJXfpz1jjaRD-Yp4C4wC4jUv8jHj0en3MW0VXKq0ZA,5280
 warp/jax_experimental.py,sha256=sQH-B4Lm_iqBfxdYypnIxvmaQ7-EuQTXNvDH0EH2weY,13400
 warp/sparse.py,sha256=APAp2M2FNpmOaQkI-ddxjluEcwJ-Ua2plNwq8IdP_Gw,64640
 warp/stubs.py,sha256=pOsOnFy_ab5Pwxc5ba76PL8xGqHlbjItsJd2Xr03OEw,50057
-warp/tape.py,sha256=gPqyPPxf_zYs6ujPQhYYHpK5AKLUMGkSPZH2yMLr0PY,48985
+warp/tape.py,sha256=w96cmAIWTEvmUCvFVoNp3qMO2CCI934CjwE-7buPpDw,48988
 warp/torch.py,sha256=jv3SYQHVKsiCzenlWUVmVs5mLYfsWcdTW446IqvEwVA,14505
 warp/types.py,sha256=7ofleQXHgslDPy57WnexClbo_LnH10Dy1u6UygN5QoY,176302
 warp/utils.py,sha256=wwyxRAujKkpdm-sB4pX5upAEgLv5RymOi6pQoNOUs-I,34663
@@ -262,7 +262,7 @@ warp/tests/test_bool.py,sha256=f-9kM--pYSg176KD9gRMOdnADfu8HP1ZnZEjvOYPjug,6472
 warp/tests/test_builtins_resolution.py,sha256=ixoOhKgOgbuH-xbqNLx1DB1Nur3PdAtR0qW-TutaHrI,63989
 warp/tests/test_bvh.py,sha256=Qa6g_uDo4NyAeJSqZlLDgxSnly4Vk7HF5cuZVB1_naE,5230
 warp/tests/test_closest_point_edge_edge.py,sha256=SHcEi6X0r7q130NwIHRp9WWp0wtLW_MF1vbwahYtiUc,6825
-warp/tests/test_codegen.py,sha256=FB7JJU7vYiN6jaNLq6Yi3ERHiAFQ7Bl_lWpPv4nRihI,15253
+warp/tests/test_codegen.py,sha256=GYB1Dv832OFMGJM3CJehWc1V9P41jRAN-TESntwIrAg,16104
 warp/tests/test_compile_consts.py,sha256=UgdpA95GnQANZ_oISmy-OBgO5srbLHT61pOn86jBY3Q,8141
 warp/tests/test_conditional.py,sha256=MmUO0RD1j4Ceew0uNmqmzZzQVTN9qPDz48bYW24hAhY,5495
 warp/tests/test_copy.py,sha256=993Xy5Y7VAFfUKhv74NOrbW4G4_hiXO7Jrpd-AVFjEM,7703
@@ -361,8 +361,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.0.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
-warp_lang-1.3.0.dist-info/METADATA,sha256=e6HWNBOvjDdSqP5vgW45uH7daZ9ZmCJgtjKaBIkKqYM,23154
-warp_lang-1.3.0.dist-info/WHEEL,sha256=yWkE5ZXPpQreDyVJwlOVgUZiiGtfawVd7c0lVluKA1c,113
-warp_lang-1.3.0.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
-warp_lang-1.3.0.dist-info/RECORD,,
+warp_lang-1.3.1.dist-info/LICENSE.md,sha256=Gq_TN0Xfat8Ftq2Rlvc31NgxdpLGVB-34kp-4eoeDV0,19265
+warp_lang-1.3.1.dist-info/METADATA,sha256=mhiGSJoUmA6R3siDsRZOcJj7Jvw4t1LCgNzfkFodglU,23154
+warp_lang-1.3.1.dist-info/WHEEL,sha256=yWkE5ZXPpQreDyVJwlOVgUZiiGtfawVd7c0lVluKA1c,113
+warp_lang-1.3.1.dist-info/top_level.txt,sha256=8pupHORyKoiN_BYWlTmv5OFBWdhqpppiBYQV5KxgEvg,5
+warp_lang-1.3.1.dist-info/RECORD,,