warp-lang 1.0.1__py3-none-manylinux2014_aarch64.whl → 1.1.0__py3-none-manylinux2014_aarch64.whl

warp/tests/aux_test_class_kernel.py

@@ -1,26 +1,26 @@
-# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-"""Dummy class used in test_reload.py"""
-
-import warp as wp
-
-
-class ClassKernelTest:
-    def __init__(self, device):
-        # 3x3 frames in the rest pose:
-        self.identities = wp.zeros(shape=10, dtype=wp.mat33, device=device)
-        wp.launch(kernel=self.gen_identities_kernel, dim=10, inputs=[self.identities], device=device)
-
-    @wp.func
-    def return_identity(e: int):
-        return wp.mat33(1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0)
-
-    @wp.kernel
-    def gen_identities_kernel(s: wp.array(dtype=wp.mat33)):
-        tid = wp.tid()
-        s[tid] = ClassKernelTest.return_identity(tid)
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+"""Dummy class used in test_reload.py"""
+
+import warp as wp
+
+
+class ClassKernelTest:
+    def __init__(self, device):
+        # 3x3 frames in the rest pose:
+        self.identities = wp.zeros(shape=10, dtype=wp.mat33, device=device)
+        wp.launch(kernel=self.gen_identities_kernel, dim=10, inputs=[self.identities], device=device)
+
+    @wp.func
+    def return_identity(e: int):
+        return wp.mat33(1.0, 0.0, 0.0, 0.0, 1.0, 0.0, 0.0, 0.0, 1.0)
+
+    @wp.kernel
+    def gen_identities_kernel(s: wp.array(dtype=wp.mat33)):
+        tid = wp.tid()
+        s[tid] = ClassKernelTest.return_identity(tid)

warp/tests/aux_test_compile_consts_dummy.py

@@ -1,10 +1,10 @@
-# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-import warp as wp
-
-MINUS_ONE = wp.constant(-1)
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+import warp as wp
+
+MINUS_ONE = wp.constant(-1)

warp/tests/aux_test_conditional_unequal_types_kernels.py

@@ -1,21 +1,21 @@
-# Copyright (c) 2023 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-"""This file defines a kernel that fails on codegen.py"""
-
-import warp as wp
-
-
-@wp.kernel
-def unequal_types_kernel():
-    x = wp.int32(10)
-    y = 10
-    z = True
-
-    # Throws a TypeError
-    if x == y == z:
-        pass
+# Copyright (c) 2023 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+"""This file defines a kernel that fails on codegen.py"""
+
+import warp as wp
+
+
+@wp.kernel
+def unequal_types_kernel():
+    x = wp.int32(10)
+    y = 10
+    z = True
+
+    # Throws a TypeError
+    if x == y == z:
+        pass

warp/tests/aux_test_dependent.py

@@ -1,22 +1,22 @@
-# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-"""This file is used to test reloading module references."""
-
-import warp as wp
-import warp.tests.aux_test_reference as ref
-
-wp.init()
-
-
-@wp.kernel
-def kern(expect: float):
-    wp.expect_eq(ref.magic(), expect)
-
-
-def run(expect, device):
-    wp.launch(kern, dim=1, inputs=[expect], device=device)
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+"""This file is used to test reloading module references."""
+
+import warp as wp
+import warp.tests.aux_test_reference as ref
+
+wp.init()
+
+
+@wp.kernel
+def kern(expect: float):
+    wp.expect_eq(ref.magic(), expect)
+
+
+def run(expect, device):
+    wp.launch(kern, dim=1, inputs=[expect], device=device)

warp/tests/aux_test_grad_customs.py

@@ -1,23 +1,23 @@
-# Copyright (c) 2024 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-"""This file is used to test importing a user-defined function with a custom gradient"""
-
-import warp as wp
-
-wp.init()
-
-
-@wp.func
-def aux_custom_fn(x: float, y: float):
-    return x * 3.0 + y / 3.0, y**2.5
-
-
-@wp.func_grad(aux_custom_fn)
-def aux_custom_fn_grad(x: float, y: float, adj_ret0: float, adj_ret1: float):
-    wp.adjoint[x] += x * adj_ret0 * 42.0 + y * adj_ret1 * 10.0
-    wp.adjoint[y] += y * adj_ret1 * 3.0
+# Copyright (c) 2024 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+"""This file is used to test importing a user-defined function with a custom gradient"""
+
+import warp as wp
+
+wp.init()
+
+
+@wp.func
+def aux_custom_fn(x: float, y: float):
+    return x * 3.0 + y / 3.0, y**2.5
+
+
+@wp.func_grad(aux_custom_fn)
+def aux_custom_fn_grad(x: float, y: float, adj_ret0: float, adj_ret1: float):
+    wp.adjoint[x] += x * adj_ret0 * 42.0 + y * adj_ret1 * 10.0
+    wp.adjoint[y] += y * adj_ret1 * 3.0

warp/tests/aux_test_reference.py

@@ -1,11 +1,11 @@
-# This file is used to test reloading module references.
-
-import warp as wp
-import warp.tests.aux_test_reference_reference as refref
-
-wp.init()
-
-
-@wp.func
-def magic():
-    return 2.0 * refref.more_magic()
+# This file is used to test reloading module references.
+
+import warp as wp
+import warp.tests.aux_test_reference_reference as refref
+
+wp.init()
+
+
+@wp.func
+def magic():
+    return 2.0 * refref.more_magic()

warp/tests/aux_test_reference_reference.py

@@ -1,10 +1,10 @@
-# This file is used to test reloading module references.
-
-import warp as wp
-
-wp.init()
-
-
-@wp.func
-def more_magic():
-    return 2.0
+# This file is used to test reloading module references.
+
+import warp as wp
+
+wp.init()
+
+
+@wp.func
+def more_magic():
+    return 2.0

warp/tests/aux_test_square.py

@@ -1,17 +1,17 @@
-import warp as wp
-
-wp.init()
-
-
-@wp.func
-def multiply(x: float):
-    return x * x
-
-
-@wp.kernel
-def kern(expect: float):
-    wp.expect_eq(multiply(4.0), expect)
-
-
-def run(expect, device):
-    wp.launch(kern, dim=1, inputs=[expect], device=device)
+import warp as wp
+
+wp.init()
+
+
+@wp.func
+def multiply(x: float):
+    return x * x
+
+
+@wp.kernel
+def kern(expect: float):
+    wp.expect_eq(multiply(4.0), expect)
+
+
+def run(expect, device):
+    wp.launch(kern, dim=1, inputs=[expect], device=device)

warp/tests/aux_test_unresolved_func.py

@@ -1,14 +1,14 @@
-# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-import warp as wp
-
-
-@wp.kernel
-def unresolved_func_kernel():
-    # this should trigger an exception due to unresolved function
-    x = wp.missing_func(42)
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+import warp as wp
+
+
+@wp.kernel
+def unresolved_func_kernel():
+    # this should trigger an exception due to unresolved function
+    x = wp.missing_func(42)

warp/tests/aux_test_unresolved_symbol.py

@@ -1,14 +1,14 @@
-# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
-# NVIDIA CORPORATION and its licensors retain all intellectual property
-# and proprietary rights in and to this software, related documentation
-# and any modifications thereto.  Any use, reproduction, disclosure or
-# distribution of this software and related documentation without an express
-# license agreement from NVIDIA CORPORATION is strictly prohibited.
-
-import warp as wp
-
-
-@wp.kernel
-def unresolved_symbol_kernel():
-    # this should trigger an exception due to unresolved symbol
-    x = missing_symbol
+# Copyright (c) 2022 NVIDIA CORPORATION.  All rights reserved.
+# NVIDIA CORPORATION and its licensors retain all intellectual property
+# and proprietary rights in and to this software, related documentation
+# and any modifications thereto.  Any use, reproduction, disclosure or
+# distribution of this software and related documentation without an express
+# license agreement from NVIDIA CORPORATION is strictly prohibited.
+
+import warp as wp
+
+
+@wp.kernel
+def unresolved_symbol_kernel():
+    # this should trigger an exception due to unresolved symbol
+    x = missing_symbol  # noqa: F821