warp-lang 1.0.0b5__py3-none-manylinux2014_x86_64.whl → 1.0.0b6__py3-none-manylinux2014_x86_64.whl

warp/tests/test_lvalue.py

@@ -1,10 +1,16 @@
-import warp as wp
-from warp.tests.test_base import *
+# 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.
 
 import unittest
 
+import warp as wp
+from warp.tests.unittest_utils import *
+
 wp.init()
-wp.config.mode = "debug"
 
 
 @wp.kernel
@@ -88,6 +94,54 @@ def test_lookup(test, device):
             raise AssertionError(f"Unexpected result, got: {f} expected: {1}")
 
 
+@wp.func
+def lookup3(foos: wp.array(dtype=wp.float32), index: int):
+    return foos[index]
+
+
+@wp.kernel
+def grad_kernel(foos: wp.array(dtype=wp.float32), bars: wp.array(dtype=wp.float32)):
+    i = wp.tid()
+
+    x = lookup3(foos, i)
+    bars[i] = x * wp.float32(i) + 1.0
+
+
+def test_grad(test, device):
+    num = 10
+    data = np.linspace(20, 20 + num, num, endpoint=False, dtype=np.float32)
+    input = wp.array(data, device=device, requires_grad=True)
+    output = wp.zeros(num, dtype=wp.float32, device=device)
+
+    ones = wp.array(np.ones(len(output)), dtype=wp.float32, device=device)
+
+    tape = wp.Tape()
+    with tape:
+        wp.launch(
+            kernel=grad_kernel,
+            dim=(num,),
+            inputs=[input],
+            outputs=[output],
+            device=device,
+        )
+
+    tape.backward(grads={output: ones})
+
+    wp.synchronize()
+
+    # test forward results
+    for i, f in enumerate(output.list()):
+        expected = data[i] * i + 1
+        if f != expected:
+            raise AssertionError(f"Unexpected result, got: {f} expected: {expected}")
+
+    # test backward results
+    for i, f in enumerate(tape.gradients[input].list()):
+        expected = i
+        if f != expected:
+            raise AssertionError(f"Unexpected result, got: {f} expected: {expected}")
+
+
 @wp.func
 def lookup2(foos: wp.array(dtype=wp.uint32), index: int):
     if index % 2 == 0:
@@ -102,7 +156,7 @@ def lookup2(foos: wp.array(dtype=wp.uint32), index: int):
 def lookup2_kernel(foos: wp.array(dtype=wp.uint32)):
     i = wp.tid()
 
-    x = lookup(foos, i)
+    x = lookup2(foos, i)
     foos[i] = x + wp.uint32(1)
 
 
@@ -401,44 +455,39 @@ def test_swizzle(test, device):
             raise AssertionError(f"Unexpected result, got: {f} expected: {expected}")
 
 
-def test_swizzle_error_invalid_attribute(test, device):
-    v = wp.vec3(1, 2, 3)
-    with test.assertRaisesRegex(
-        AttributeError,
-        r"'vec3f' object has no attribute 'foo'$",
-    ):
-        v.foo
-
-    try:
-        v.bar = 123
-    except AttributeError:
-        test.fail()
+devices = get_test_devices()
 
 
-def register(parent):
-    devices = get_test_devices()
+class TestLValue(unittest.TestCase):
+    def test_swizzle_error_invalid_attribute(self):
+        v = wp.vec3(1, 2, 3)
+        with self.assertRaisesRegex(
+            AttributeError,
+            r"'vec3f' object has no attribute 'foo'$",
+        ):
+            v.foo
 
-    class TestLValue(parent):
-        pass
+        try:
+            v.bar = 123
+        except AttributeError:
+            self.fail()
 
-    add_function_test(TestLValue, "test_rmw_array", test_rmw_array, devices=devices)
-    add_function_test(TestLValue, "test_rmw_array_struct", test_rmw_array_struct, devices=devices)
-    add_function_test(TestLValue, "test_lookup", test_lookup, devices=devices)
-    add_function_test(TestLValue, "test_lookup2", test_lookup2, devices=devices)
-    add_function_test(TestLValue, "test_unary", test_unary, devices=devices)
-    add_function_test(TestLValue, "test_rvalue", test_rvalue, devices=devices)
-    add_function_test(TestLValue, "test_intermediate", test_intermediate, devices=devices)
-    add_function_test(TestLValue, "test_array_assign", test_array_assign, devices=devices)
-    add_function_test(TestLValue, "test_array_struct_assign", test_array_struct_assign, devices=devices)
-    add_function_test(TestLValue, "test_array_struct_struct_assign", test_array_struct_struct_assign, devices=devices)
-    add_function_test(TestLValue, "test_complex", test_complex, devices=devices)
-    add_function_test(TestLValue, "test_swizzle", test_swizzle, devices=devices)
-    add_function_test(TestLValue, "test_swizzle_error_invalid_attribute", test_swizzle_error_invalid_attribute)
 
-    return TestLValue
+add_function_test(TestLValue, "test_rmw_array", test_rmw_array, devices=devices)
+add_function_test(TestLValue, "test_rmw_array_struct", test_rmw_array_struct, devices=devices)
+add_function_test(TestLValue, "test_lookup", test_lookup, devices=devices)
+add_function_test(TestLValue, "test_lookup2", test_lookup2, devices=devices)
+add_function_test(TestLValue, "test_grad", test_grad, devices=devices)
+add_function_test(TestLValue, "test_unary", test_unary, devices=devices)
+add_function_test(TestLValue, "test_rvalue", test_rvalue, devices=devices)
+add_function_test(TestLValue, "test_intermediate", test_intermediate, devices=devices)
+add_function_test(TestLValue, "test_array_assign", test_array_assign, devices=devices)
+add_function_test(TestLValue, "test_array_struct_assign", test_array_struct_assign, devices=devices)
+add_function_test(TestLValue, "test_array_struct_struct_assign", test_array_struct_struct_assign, devices=devices)
+add_function_test(TestLValue, "test_complex", test_complex, devices=devices)
+add_function_test(TestLValue, "test_swizzle", test_swizzle, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)

warp/tests/test_marching_cubes.py

@@ -5,14 +5,12 @@
 # distribution of this software and related documentation without an express
 # license agreement from NVIDIA CORPORATION is strictly prohibited.
 
-# include parent path
+import unittest
+
 import numpy as np
-import math
 
 import warp as wp
-from warp.tests.test_base import *
-
-import unittest
+from warp.tests.unittest_utils import *
 
 wp.init()
 
@@ -39,7 +37,7 @@ def test_marching_cubes(test, device):
 
     radius = dim / 4.0
 
-    wp.launch(make_field, dim=field.shape, inputs=[field, wp.vec3(dim / 2, dim / 2, dim / 2), radius], device="cuda")
+    wp.launch(make_field, dim=field.shape, inputs=[field, wp.vec3(dim / 2, dim / 2, dim / 2), radius], device=device)
 
     iso.surface(field=field, threshold=0.0)
 
@@ -52,18 +50,16 @@ def test_marching_cubes(test, device):
     iso.resize(nx=dim * 2, ny=dim * 2, nz=dim * 2, max_verts=max_verts, max_tris=max_tris)
 
 
-def register(parent):
-    devices = ["cuda"]
+devices = get_unique_cuda_test_devices()
+
 
-    class TestMarchingCubes(parent):
-        pass
+class TestMarchingCubes(unittest.TestCase):
+    pass
 
-    add_function_test(TestMarchingCubes, "test_marching_cubes", test_marching_cubes, devices=devices)
 
-    return TestMarchingCubes
+add_function_test(TestMarchingCubes, "test_marching_cubes", test_marching_cubes, devices=devices)
 
 
 if __name__ == "__main__":
     wp.build.clear_kernel_cache()
-    _ = register(unittest.TestCase)
     unittest.main(verbosity=2)