warp-lang 0.9.0__py3-none-win_amd64.whl → 0.11.0__py3-none-win_amd64.whl

warp/tests/test_all.py

@@ -1,215 +0,0 @@
-# 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 unittest
-import os
-
-import warp as wp
-
-# Uncomment to run the tests on all devices
-# import warp.tests.test_base
-# warp.tests.test_base.test_mode = "all"
-
-from warp.tests.test_base import get_test_devices
-
-import warp.tests.test_codegen
-import warp.tests.test_mesh_query_aabb
-import warp.tests.test_mesh_query_point
-import warp.tests.test_mesh_query_ray
-import warp.tests.test_conditional
-import warp.tests.test_operators
-import warp.tests.test_rounding
-import warp.tests.test_hash_grid
-import warp.tests.test_ctypes
-import warp.tests.test_rand
-import warp.tests.test_noise
-import warp.tests.test_tape
-import warp.tests.test_compile_consts
-import warp.tests.test_volume
-import warp.tests.test_mlp
-import warp.tests.test_grad
-import warp.tests.test_intersect
-import warp.tests.test_array
-import warp.tests.test_launch
-import warp.tests.test_import
-import warp.tests.test_func
-import warp.tests.test_fp16
-import warp.tests.test_reload
-import warp.tests.test_struct
-import warp.tests.test_closest_point_edge_edge
-import warp.tests.test_multigpu
-import warp.tests.test_quat
-import warp.tests.test_atomic
-import warp.tests.test_adam
-import warp.tests.test_transient_module
-import warp.tests.test_lerp
-import warp.tests.test_smoothstep
-import warp.tests.test_model
-import warp.tests.test_fast_math
-import warp.tests.test_streams
-import warp.tests.test_torch
-import warp.tests.test_pinned
-import warp.tests.test_matmul
-import warp.tests.test_options
-import warp.tests.test_dlpack
-import warp.tests.test_vec
-import warp.tests.test_mat
-import warp.tests.test_arithmetic
-import warp.tests.test_spatial
-import warp.tests.test_math
-import warp.tests.test_generics
-import warp.tests.test_indexedarray
-import warp.tests.test_copy
-import warp.tests.test_mesh
-
-
-def register_tests(parent):
-    tests = []
-
-    tests.append(warp.tests.test_codegen.register(parent))
-    tests.append(warp.tests.test_mesh_query_aabb.register(parent))
-    tests.append(warp.tests.test_mesh_query_point.register(parent))
-    tests.append(warp.tests.test_mesh_query_ray.register(parent))
-    tests.append(warp.tests.test_conditional.register(parent))
-    tests.append(warp.tests.test_operators.register(parent))
-    tests.append(warp.tests.test_rounding.register(parent))
-    tests.append(warp.tests.test_hash_grid.register(parent))
-    tests.append(warp.tests.test_ctypes.register(parent))
-    tests.append(warp.tests.test_rand.register(parent))
-    tests.append(warp.tests.test_noise.register(parent))
-    tests.append(warp.tests.test_tape.register(parent))
-    tests.append(warp.tests.test_compile_consts.register(parent))
-    tests.append(warp.tests.test_volume.register(parent))
-    tests.append(warp.tests.test_mlp.register(parent))
-    tests.append(warp.tests.test_grad.register(parent))
-    tests.append(warp.tests.test_intersect.register(parent))
-    tests.append(warp.tests.test_array.register(parent))
-    tests.append(warp.tests.test_launch.register(parent))
-    tests.append(warp.tests.test_import.register(parent))
-    tests.append(warp.tests.test_func.register(parent))
-    tests.append(warp.tests.test_fp16.register(parent))
-    tests.append(warp.tests.test_reload.register(parent))
-    tests.append(warp.tests.test_struct.register(parent))
-    tests.append(warp.tests.test_closest_point_edge_edge.register(parent))
-    tests.append(warp.tests.test_multigpu.register(parent))
-    tests.append(warp.tests.test_quat.register(parent))
-    tests.append(warp.tests.test_atomic.register(parent))
-    tests.append(warp.tests.test_adam.register(parent))
-    tests.append(warp.tests.test_transient_module.register(parent))
-    tests.append(warp.tests.test_lerp.register(parent))
-    tests.append(warp.tests.test_smoothstep.register(parent))
-    tests.append(warp.tests.test_model.register(parent))
-    tests.append(warp.tests.test_fast_math.register(parent))
-    tests.append(warp.tests.test_streams.register(parent))
-    tests.append(warp.tests.test_torch.register(parent))
-    tests.append(warp.tests.test_pinned.register(parent))
-    tests.append(warp.tests.test_matmul.register(parent))
-    tests.append(warp.tests.test_options.register(parent))
-    tests.append(warp.tests.test_dlpack.register(parent))
-    tests.append(warp.tests.test_vec.register(parent))
-    tests.append(warp.tests.test_mat.register(parent))
-    tests.append(warp.tests.test_arithmetic.register(parent))
-    tests.append(warp.tests.test_spatial.register(parent))
-    tests.append(warp.tests.test_math.register(parent))
-    tests.append(warp.tests.test_generics.register(parent))
-    tests.append(warp.tests.test_indexedarray.register(parent))
-    tests.append(warp.tests.test_copy.register(parent))
-    tests.append(warp.tests.test_mesh.register(parent))
-
-    return tests
-
-
-class TeamCityTestResult(unittest.TextTestResult):
-    """This class will report each test result to TeamCity"""
-
-    def __init__(self, stream, descriptions, verbosity):
-        super(TeamCityTestResult, self).__init__(stream, descriptions, verbosity)
-
-    def addSuccess(self, test):
-        super(TeamCityTestResult, self).addSuccess(test)
-        self.reportSuccess(test)
-
-    def addError(self, test, err):
-        super(TeamCityTestResult, self).addError(test, err)
-        self.reportFailure(test)
-
-    def addFailure(self, test, err):
-        super(TeamCityTestResult, self).addFailure(test, err)
-        self.reportFailure(test)
-
-    def addSkip(self, test, reason):
-        super(TeamCityTestResult, self).addSkip(test, reason)
-
-    def addExpectedFailure(self, test, err):
-        super(TeamCityTestResult, self).addExpectedFailure(test, err)
-        self.reportSuccess(test)
-
-    def addUnexpectedSuccess(self, test):
-        super(TeamCityTestResult, self).addUnexpectedSuccess(test)
-        self.reportFailure(test)
-
-    def reportSuccess(self, test):
-        test_id = test.id()
-        print(f"##teamcity[testStarted name='{test_id}']")
-        print(f"##teamcity[testFinished name='{test_id}']")
-
-    def reportFailure(self, test):
-        test_id = test.id()
-        print(f"##teamcity[testStarted name='{test_id}']")
-        print(f"##teamcity[testFailed name='{test_id}']")
-        print(f"##teamcity[testFinished name='{test_id}']")
-
-
-class TeamCityTestRunner(unittest.TextTestRunner):
-    """Test runner that will report test results to TeamCity if running in TeamCity"""
-
-    def __init__(self, **kwargs):
-        self.running_in_teamcity = os.environ.get("TEAMCITY_VERSION") is not None
-        if self.running_in_teamcity:
-            kwargs["resultclass"] = TeamCityTestResult
-        super(TeamCityTestRunner, self).__init__(**kwargs)
-
-    def run(self, test, name):
-        if self.running_in_teamcity:
-            print(f"##teamcity[testSuiteStarted name='{name}']")
-
-        result = super(TeamCityTestRunner, self).run(test)
-
-        if self.running_in_teamcity:
-            print(f"##teamcity[testSuiteFinished name='{name}']")
-            if not result.wasSuccessful():
-                print(f"##teamcity[buildStatus status='FAILURE']")
-
-        return result
-
-
-def run():
-    test_suite = unittest.TestSuite()
-
-    tests = register_tests(unittest.TestCase)
-
-    for test in tests:
-        test_suite.addTest(unittest.defaultTestLoader.loadTestsFromTestCase(test))
-
-    # force rebuild of all kernels
-    wp.build.clear_kernel_cache()
-
-    # load all modules
-    for device in get_test_devices():
-        wp.force_load(device)
-
-    runner = TeamCityTestRunner(verbosity=2, failfast=False)
-    ret = not runner.run(test_suite, "WarpTests").wasSuccessful()
-    return ret
-
-
-if __name__ == "__main__":
-    ret = run()
-
-    import sys
-
-    sys.exit(ret)

warp/tests/test_array_scan.py

@@ -1,60 +0,0 @@
-import sys
-import os
-
-sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
-
-import warp as wp
-import numpy as np
-
-wp.config.mode = "release"
-wp.config.verbose = True
-wp.config.verify_cuda = True
-
-wp.init()
-
-n = 100000
-num_runs = 16
-
-
-def test_for_type(dtype, device):
-    dtype_str = dtype.__name__
-    if dtype == int:
-        values = np.random.randint(-1e6, 1e6, n, dtype=dtype)
-    else:
-        values = np.random.uniform(-1, 1, n)
-
-    results_ref = np.cumsum(values)
-
-    in_values = wp.array(values, dtype=dtype, device=device)
-    out_values_inc = wp.zeros(len(values), dtype=dtype, device=device)
-    out_values_exc = wp.zeros(len(values), dtype=dtype, device=device)
-
-    wp.utils.array_scan(in_values, out_values_inc, True)
-    wp.utils.array_scan(in_values, out_values_exc, False)
-
-    tolerance = 0 if dtype == int else 1e-3
-
-    results_inc = out_values_inc.numpy().squeeze()
-    results_exc = out_values_exc.numpy().squeeze()
-    error_inc = np.max(np.abs(results_inc - results_ref)) / abs(results_ref[-1])
-    error_exc = max(np.max(np.abs(results_exc[1:] - results_ref[:-1])), abs(results_exc[0])) / abs(results_ref[-2])
-    if error_inc > tolerance:
-        print(f"FAIL! Max error in inclusive scan for {dtype_str}: {error_inc}")
-    else:
-        print(f"PASS! Max error in inclusive scan for {dtype_str}: {error_inc}")
-
-    if error_exc > tolerance:
-        print(f"FAIL! Max error in exclusive scan for {dtype_str}: {error_exc}")
-    # else:
-    #     print(f"PASS! Max error in exclusive scan for {dtype_str}: {error_exc}")
-
-
-np.random.seed(1008)
-for device in ("cuda", "cpu"):
-    print(f"\n\nTesting {device}")
-    for i in range(num_runs):
-        print(f"Run: {i+1}")
-        print("---------")
-
-        test_for_type(int, device)
-        test_for_type(float, device)

warp/tests/test_base.py

@@ -1,208 +0,0 @@
-# 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 unittest
-import os
-import sys
-
-import numpy as np
-import warp as wp
-
-# default test mode (see get_test_devices())
-#   "basic" - only run on CPU and first GPU device
-#   "unique" - run on CPU and all unique GPU arches
-#   "all" - run on all devices
-test_mode = "basic"
-
-
-def get_test_devices(mode=None):
-    if mode is None:
-        global test_mode
-        mode = test_mode
-
-    devices = []
-
-    # only run on CPU and first GPU device
-    if mode == "basic":
-        if wp.is_cpu_available():
-            devices.append(wp.get_device("cpu"))
-        if wp.is_cuda_available():
-            devices.append(wp.get_device("cuda:0"))
-
-    # run on CPU and all unique GPU arches
-    elif mode == "unique":
-        if wp.is_cpu_available():
-            devices.append(wp.get_device("cpu"))
-
-        cuda_devices = wp.get_cuda_devices()
-
-        unique_cuda_devices = {}
-        for d in cuda_devices:
-            if d.arch not in unique_cuda_devices:
-                unique_cuda_devices[d.arch] = d
-
-        devices.extend(list(unique_cuda_devices.values()))
-
-    # run on all devices
-    elif mode == "all":
-        devices = wp.get_devices()
-
-    return devices
-
-
-# redirects and captures all stdout output (including from C-libs)
-class StdOutCapture:
-    def begin(self):
-        # save original
-        self.saved = sys.stdout
-        self.target = os.dup(self.saved.fileno())
-
-        # create temporary capture stream
-        import io, tempfile
-
-        self.tempfile = io.TextIOWrapper(
-            tempfile.TemporaryFile(buffering=0), encoding="utf-8", errors="replace", newline="", write_through=True
-        )
-
-        os.dup2(self.tempfile.fileno(), self.saved.fileno())
-
-        sys.stdout = self.tempfile
-
-    def end(self):
-        os.dup2(self.target, self.saved.fileno())
-        os.close(self.target)
-
-        self.tempfile.seek(0)
-        res = self.tempfile.buffer.read()
-        self.tempfile.close()
-
-        sys.stdout = self.saved
-
-        return str(res.decode("utf-8"))
-
-
-class CheckOutput:
-    def __init__(self, test):
-        self.test = test
-
-    def __enter__(self):
-        # wp.force_load()
-
-        self.capture = StdOutCapture()
-        self.capture.begin()
-
-    def __exit__(self, exc_type, exc_value, traceback):
-        # ensure any stdout output is flushed
-        wp.synchronize()
-
-        s = self.capture.end()
-        if s != "":
-            print(s.rstrip())
-
-        # fail if test produces unexpected output (e.g.: from wp.expect_eq() builtins)
-        # we allow strings starting of the form "Module xxx load on device xxx"
-        # for lazy loaded modules
-        if s != "" and not s.startswith("Module"):
-            self.test.fail(f"Unexpected output:\n'{s.rstrip()}'")
-
-
-def assert_array_equal(result, expect):
-    a = result.numpy()
-    b = expect.numpy()
-
-    if (a == b).all() == False:
-        raise AssertionError(f"Unexpected result, got: {a} expected: {b}")
-
-
-def assert_np_equal(result, expect, tol=0.0):
-    a = result.flatten()
-    b = expect.flatten()
-
-    if tol == 0.0:
-        if (a == b).all() == False:
-            raise AssertionError(f"Unexpected result, got: {a} expected: {b}")
-
-    else:
-        delta = a - b
-        err = np.max(np.abs(delta))
-        if err > tol:
-            raise AssertionError(
-                f"Maximum expected error exceeds tolerance got: {a}, expected: {b}, with err: {err} > {tol}"
-            )
-
-
-def create_test_func(func, device, **kwargs):
-    # pass args to func
-    def test_func(self):
-        with CheckOutput(self):
-            func(self, device, **kwargs)
-
-    return test_func
-
-
-def sanitize_identifier(s):
-    """replace all non-identifier characters with '_'"""
-
-    s = str(s)
-    if s.isidentifier():
-        return s
-    else:
-        import re
-
-        return re.sub("\W|^(?=\d)", "_", s)
-
-
-def add_function_test(cls, name, func, devices=None, **kwargs):
-    if devices is None:
-        setattr(cls, name, create_test_func(func, None, **kwargs))
-    else:
-        for device in devices:
-            setattr(cls, name + "_" + sanitize_identifier(device), create_test_func(func, device, **kwargs))
-
-
-def add_kernel_test(cls, kernel, dim, name=None, expect=None, inputs=None, devices=None):
-    def test_func(self, device):
-        args = []
-        if inputs:
-            args.extend(inputs)
-
-        if expect:
-            # allocate outputs to match results
-            result = wp.array(expect, dtype=int, device=device)
-            output = wp.zeros_like(result)
-
-            args.append(output)
-
-        # force load so that we don't generate any log output during launch
-        kernel.module.load(device)
-
-        with CheckOutput(self):
-            wp.launch(kernel, dim=dim, inputs=args, device=device)
-
-        # check output values
-        if expect:
-            assert_array_equal(output, result)
-
-    if name is None:
-        name = kernel.key
-
-    # device is required for kernel tests, so use all devices if none were given
-    if devices is None:
-        devices = get_test_devices()
-
-    # register test func with class for the given devices
-    for d in devices:
-        # use a lambda to forward the device to the inner test function
-        test_lambda = lambda test, device=d: test_func(test, device)
-        setattr(cls, name + "_" + sanitize_identifier(d), test_lambda)
-
-
-# helper that first calls the test function to generate all kernel permuations
-# so that compilation is done in one-shot instead of per-test
-def add_function_test_register_kernel(cls, name, func, devices=None, **kwargs):
-    func(None, None, **kwargs, register_kernels=True)
-    add_function_test(cls, name, func, devices=devices, **kwargs)

warp/tests/test_unresolved_func.py

@@ -1,7 +0,0 @@
-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/test_unresolved_symbol.py

@@ -1,7 +0,0 @@
-import warp as wp
-
-
-@wp.kernel
-def unresolved_symbol_kernel():
-    # this should trigger an exception due to unresolved symbol
-    x = missing_symbol

warp_lang-0.9.0.dist-info/METADATA

@@ -1,20 +0,0 @@
-Metadata-Version: 2.1
-Name: warp-lang
-Version: 0.9.0
-Summary: A Python framework for high-performance simulation and graphics programming
-Home-page: https://github.com/NVIDIA/warp
-Author: NVIDIA
-Author-email: mmacklin@nvidia.com
-License: NVSCL
-Project-URL: Documentation, https://nvidia.github.io/warp
-Classifier: Programming Language :: Python :: 3.7
-Classifier: Programming Language :: Python :: 3.8
-Classifier: Programming Language :: Python :: 3.9
-Classifier: Programming Language :: Python :: 3.10
-Classifier: Programming Language :: Python :: 3.11
-Classifier: License :: Other/Proprietary License
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.7
-Description-Content-Type: text/markdown
-Requires-Dist: numpy
-