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

warp/tests/test_module_hashing.py

@@ -0,0 +1,111 @@
+# 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.
+
+# TODO: add more tests for kernels and generics
+
+import os
+import tempfile
+import unittest
+from importlib import util
+
+import warp as wp
+from warp.tests.unittest_utils import *
+
+FUNC_OVERLOAD_1 = """# -*- coding: utf-8 -*-
+import warp as wp
+
+@wp.func
+def fn():
+    wp.print(17)
+
+@wp.func
+def fn(value: int):
+    wp.print(value)
+"""
+
+# should be same hash as FUNC_OVERLOAD_1
+FUNC_OVERLOAD_2 = """# -*- coding: utf-8 -*-
+import warp as wp
+
+@wp.func
+def fn():
+    wp.print(17)
+
+@wp.func
+def fn(value: int):
+    wp.print(value)
+"""
+
+# should be different hash than FUNC_OVERLOAD_1 (first overload is different)
+FUNC_OVERLOAD_3 = """# -*- coding: utf-8 -*-
+import warp as wp
+
+@wp.func
+def fn():
+    wp.print(42)
+
+@wp.func
+def fn(value: int):
+    wp.print(value)
+"""
+
+# should be different hash than FUNC_OVERLOAD_1 (second overload is different)
+FUNC_OVERLOAD_4 = """# -*- coding: utf-8 -*-
+import warp as wp
+
+@wp.func
+def fn():
+    wp.print(17)
+
+@wp.func
+def fn(value: int):
+    wp.print(value + 1)
+"""
+
+
+def load_code_as_module(code, name):
+    file, file_path = tempfile.mkstemp(suffix=".py")
+
+    try:
+        with os.fdopen(file, "w") as f:
+            f.write(code)
+
+        spec = util.spec_from_file_location(name, file_path)
+        module = util.module_from_spec(spec)
+        spec.loader.exec_module(module)
+    finally:
+        os.remove(file_path)
+
+    return wp.get_module(module.__name__)
+
+
+def test_function_overload_hashing(test, device):
+    m1 = load_code_as_module(FUNC_OVERLOAD_1, "func_overload_1")
+    m2 = load_code_as_module(FUNC_OVERLOAD_2, "func_overload_2")
+    m3 = load_code_as_module(FUNC_OVERLOAD_3, "func_overload_3")
+    m4 = load_code_as_module(FUNC_OVERLOAD_4, "func_overload_4")
+
+    hash1 = m1.hash_module()
+    hash2 = m2.hash_module()
+    hash3 = m3.hash_module()
+    hash4 = m4.hash_module()
+
+    test.assertEqual(hash2, hash1)
+    test.assertNotEqual(hash3, hash1)
+    test.assertNotEqual(hash4, hash1)
+
+
+class TestModuleHashing(unittest.TestCase):
+    pass
+
+
+add_function_test(TestModuleHashing, "test_function_overload_hashing", test_function_overload_hashing)
+
+
+if __name__ == "__main__":
+    wp.build.clear_kernel_cache()
+    unittest.main(verbosity=2)

warp/tests/test_modules_lite.py

@@ -10,9 +10,6 @@ import unittest
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
-
 devices = get_test_devices()
 
 

warp/tests/test_multigpu.py

@@ -13,8 +13,6 @@ import warp as wp
 from warp.tests.unittest_utils import *
 from warp.utils import check_iommu
 
-wp.init()
-
 
 @wp.kernel
 def inc(a: wp.array(dtype=float)):

warp/tests/test_noise.py

@@ -12,10 +12,6 @@ import numpy as np
 import warp as wp
 from warp.tests.unittest_utils import *
 
-# import matplotlib.pyplot as plt
-
-wp.init()
-
 
 @wp.kernel
 def pnoise(

warp/tests/test_operators.py

@@ -10,8 +10,6 @@ import unittest
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 @wp.kernel
 def test_operators_scalar_float():

warp/tests/test_options.py

@@ -12,8 +12,6 @@ import unittest
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 @wp.kernel
 def scale(

warp/tests/test_peer.py

@@ -10,8 +10,6 @@ import unittest
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def get_device_pair_with_peer_access_support():
     devices = wp.get_cuda_devices()

warp/tests/test_pinned.py

@@ -12,8 +12,6 @@ import numpy as np
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def test_pinned(test: unittest.TestCase, device):
     assert wp.get_device(device).is_cuda, "Test device must be a CUDA device"

warp/tests/test_print.py

@@ -11,8 +11,6 @@ import unittest
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 @wp.kernel
 def test_print_kernel():

warp/tests/test_quat.py

@@ -13,8 +13,6 @@ import warp as wp
 import warp.sim
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 np_float_types = [np.float32, np.float64, np.float16]
 
 kernel_cache = {}

warp/tests/test_rand.py

@@ -12,11 +12,6 @@ import numpy as np
 import warp as wp
 from warp.tests.unittest_utils import *
 
-# import matplotlib.pyplot as plt
-
-
-wp.init()
-
 
 @wp.kernel
 def test_kernel(
@@ -113,6 +108,46 @@ def test_rand(test, device):
     test.assertTrue(err < 1e-04)
 
 
+@wp.kernel
+def randn_kernel(
+    x: wp.array(dtype=float),
+):
+    tid = wp.tid()
+    r = wp.rand_init(tid)
+    x[tid] = wp.randn(r)
+
+
+def test_randn(test, device):
+    N = 100000000
+
+    samples = wp.zeros(N, dtype=float, device=device)
+
+    wp.launch(randn_kernel, inputs=[samples], dim=N, device=device)
+
+    randn_samples = samples.numpy()
+
+    test.assertFalse(np.any(np.isinf(randn_samples)))
+    test.assertFalse(np.any(np.isnan(randn_samples)))
+
+    randn_true = np.array(
+        [
+            -1.8213255,
+            0.27881497,
+            -1.1122388,
+            0.5936895,
+            0.04976363,
+            0.69087356,
+            0.2935363,
+            0.8405019,
+            -0.8436684,
+            0.53108305,
+        ]
+    )
+
+    err = np.max(np.abs(randn_samples[0:10] - randn_true))
+    test.assertTrue(err < 1e-04)
+
+
 @wp.kernel
 def sample_cdf_kernel(kernel_seed: int, cdf: wp.array(dtype=float), samples: wp.array(dtype=int)):
     tid = wp.tid()
@@ -278,6 +313,7 @@ class TestRand(unittest.TestCase):
 
 
 add_function_test(TestRand, "test_rand", test_rand, devices=devices)
+add_function_test(TestRand, "test_randn", test_randn, devices=devices)
 add_function_test(TestRand, "test_sample_cdf", test_sample_cdf, devices=devices)
 add_function_test(TestRand, "test_sampling_methods", test_sampling_methods, devices=devices)
 add_function_test(TestRand, "test_poisson", test_poisson, devices=devices)

warp/tests/test_reload.py

@@ -22,8 +22,6 @@ import warp.tests.aux_test_reference_reference as test_reference_reference
 import warp.tests.aux_test_square as test_square
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def reload_module(module):
     # Clearing the .pyc file associated with a module is a necessary workaround
@@ -68,8 +66,6 @@ def test_redefine(test, device):
 
 square_two = """import warp as wp
 
-wp.init()
-
 
 @wp.func
 def sqr(x: float):
@@ -87,8 +83,6 @@ def run(expect, device):
 
 square_four = """import warp as wp
 
-wp.init()
-
 
 @wp.func
 def multiply(x: float):
@@ -148,8 +142,6 @@ template_ref = """# 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():
@@ -160,8 +152,6 @@ template_refref = """# This file is used to test reloading module references.
 
 import warp as wp
 
-wp.init()
-
 
 @wp.func
 def more_magic():

warp/tests/test_rounding.py

@@ -15,8 +15,6 @@ from warp.tests.unittest_utils import *
 compare_to_numpy = False
 print_results = False
 
-wp.init()
-
 
 @wp.kernel
 def test_kernel(

warp/tests/test_runlength_encode.py

@@ -14,8 +14,6 @@ import warp as wp
 from warp.tests.unittest_utils import *
 from warp.utils import runlength_encode
 
-wp.init()
-
 
 def test_runlength_encode_int(test, device, n):
     rng = np.random.default_rng(123)

warp/tests/test_sim_grad.py

@@ -13,8 +13,6 @@ import warp as wp
 import warp.sim
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 @wp.kernel
 def evaluate_loss(

warp/tests/test_sim_kinematics.py

@@ -13,8 +13,6 @@ import warp as wp
 import warp.sim
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def test_fk_ik(test, device):
     builder = wp.sim.ModelBuilder()

warp/tests/test_smoothstep.py

@@ -70,8 +70,6 @@ TEST_DATA = {
     ),
 }
 
-wp.init()
-
 
 def test_smoothstep(test, device):
     def make_kernel_fn(data_type):

warp/tests/test_snippet.py

@@ -5,8 +5,6 @@ import numpy as np
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def test_basic(test, device):
     snippet = """

warp/tests/test_sparse.py

@@ -28,8 +28,6 @@ from warp.sparse import (
 )
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 
 def _get_block(mat, row, col, block_shape):
     return mat[row * block_shape[0] : (row + 1) * block_shape[0], col * block_shape[1] : (col + 1) * block_shape[1]]

warp/tests/test_spatial.py

@@ -12,8 +12,6 @@ import numpy as np
 import warp as wp
 from warp.tests.unittest_utils import *
 
-wp.init()
-
 np_float_types = [np.float32, np.float64, np.float16]
 
 kernel_cache = {}