warp-lang 1.10.0__py3-none-macosx_11_0_arm64.whl

warp/tests/unittest_serial.py

@@ -0,0 +1,47 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+import warp as wp
+import warp.tests.unittest_suites
+
+
+def run_suite() -> bool:
+    """Run a test suite"""
+
+    # force rebuild of all kernels
+    wp.clear_lto_cache()
+    wp.clear_kernel_cache()
+    print("Cleared Warp kernel cache")
+
+    runner = unittest.TextTestRunner(verbosity=2, failfast=True)
+
+    # Can swap out different suites
+    suite = warp.tests.unittest_suites.default_suite()
+    # suite = warp.tests.unittest_suites.auto_discover_suite()
+    # suite = warp.tests.unittest_suites.kit_suite()
+
+    print(f"Test suite has {suite.countTestCases()} tests")
+
+    ret = not runner.run(suite).wasSuccessful()
+    return ret
+
+
+if __name__ == "__main__":
+    ret = run_suite()
+    import sys
+
+    sys.exit(ret)

warp/tests/unittest_suites.py

@@ -0,0 +1,430 @@
+# SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+"""Warp Test Suites
+
+This file is intended to define functions that return TestSuite objects, which
+can be used in parallel or serial unit tests (with optional code coverage)
+"""
+
+import os
+import sys
+import unittest
+
+START_DIRECTORY = os.path.realpath(os.path.dirname(__file__))
+TOP_LEVEL_DIRECTORY = os.path.realpath(os.path.join(START_DIRECTORY, "..", ".."))
+
+
+def _create_suite_from_test_classes(test_loader, test_classes):
+    suite = unittest.TestSuite()
+
+    for test in test_classes:
+        sub_suite = unittest.TestSuite()
+        # Note that the test_loader might have testNamePatterns set
+        sub_suite.addTest(test_loader.loadTestsFromTestCase(test))
+        suite.addTest(sub_suite)
+
+    return suite
+
+
+def auto_discover_suite(loader=unittest.defaultTestLoader, pattern="test*.py"):
+    """Uses unittest auto-discovery to build a test suite (test_*.py pattern)"""
+
+    return loader.discover(start_dir=START_DIRECTORY, pattern=pattern, top_level_dir=TOP_LEVEL_DIRECTORY)
+
+
+def _iter_class_suites(test_suite):
+    """Iterate class-level test suites - test suites that contains test cases
+
+    From unittest_parallel.py
+    """
+    has_cases = any(isinstance(suite, unittest.TestCase) for suite in test_suite)
+    if has_cases:
+        yield test_suite
+    else:
+        for suite in test_suite:
+            yield from _iter_class_suites(suite)
+
+
+def compare_unittest_suites(
+    test_loader: unittest.TestLoader, test_suite_name: str, reference_suite: unittest.TestSuite
+) -> None:
+    """Prints the tests in `test_suite` that are not in `reference_suite`."""
+
+    test_suite_fn = getattr(sys.modules[__name__], test_suite_name + "_suite")
+
+    test_suite = test_suite_fn(test_loader)
+
+    test_suite_classes_str = {
+        type(test_suite._tests[0]).__name__
+        for test_suite in list(_iter_class_suites(test_suite))
+        if test_suite.countTestCases() > 0
+    }
+
+    reference_suite_classes_str = {
+        type(test_suite._tests[0]).__name__
+        for test_suite in list(_iter_class_suites(reference_suite))
+        if test_suite.countTestCases() > 0
+    }
+
+    set_difference = reference_suite_classes_str - test_suite_classes_str
+
+    print(f"Selected test suite '{test_suite_name}'")
+    if len(set_difference) > 0:
+        print(f"Test suite '{test_suite_name}' omits the following test classes:")
+        for test_entry in set_difference:
+            print(f"    {test_entry}")
+
+    return test_suite
+
+
+def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
+    """Example of a manually constructed test suite.
+
+    Intended to be modified to create additional test suites
+    """
+    from warp.tests.cuda.test_async import TestAsync
+    from warp.tests.cuda.test_mempool import TestMempool
+    from warp.tests.cuda.test_multigpu import TestMultiGPU
+    from warp.tests.cuda.test_peer import TestPeer
+    from warp.tests.cuda.test_pinned import TestPinned
+    from warp.tests.cuda.test_streams import TestStreams
+    from warp.tests.geometry.test_bvh import TestBvh
+    from warp.tests.geometry.test_hash_grid import TestHashGrid
+    from warp.tests.geometry.test_marching_cubes import TestMarchingCubes
+    from warp.tests.geometry.test_mesh import TestMesh
+    from warp.tests.geometry.test_mesh_query_aabb import TestMeshQueryAABBMethods
+    from warp.tests.geometry.test_mesh_query_point import TestMeshQueryPoint
+    from warp.tests.geometry.test_mesh_query_ray import TestMeshQueryRay
+    from warp.tests.geometry.test_volume import TestVolume
+    from warp.tests.geometry.test_volume_write import TestVolumeWrite
+    from warp.tests.interop.test_dlpack import TestDLPack
+    from warp.tests.interop.test_jax import TestJax
+    from warp.tests.interop.test_torch import TestTorch
+    from warp.tests.test_adam import TestAdam
+    from warp.tests.test_arithmetic import TestArithmetic
+    from warp.tests.test_array import TestArray
+    from warp.tests.test_array_reduce import TestArrayReduce
+    from warp.tests.test_atomic import TestAtomic
+    from warp.tests.test_atomic_bitwise import TestAtomicBitwise
+    from warp.tests.test_atomic_cas import TestAtomicCAS
+    from warp.tests.test_bool import TestBool
+    from warp.tests.test_builtins_resolution import TestBuiltinsResolution
+    from warp.tests.test_closest_point_edge_edge import TestClosestPointEdgeEdgeMethods
+    from warp.tests.test_codegen import TestCodeGen
+    from warp.tests.test_codegen_instancing import TestCodeGenInstancing
+    from warp.tests.test_compile_consts import TestConstants
+    from warp.tests.test_conditional import TestConditional
+    from warp.tests.test_context import TestContext
+    from warp.tests.test_copy import TestCopy
+    from warp.tests.test_ctypes import TestCTypes
+    from warp.tests.test_dense import TestDense
+    from warp.tests.test_devices import TestDevices
+    from warp.tests.test_examples import (
+        TestCoreExamples,
+        TestFemDiffusionExamples,
+        TestFemExamples,
+        TestOptimExamples,
+    )
+    from warp.tests.test_fabricarray import TestFabricArray
+    from warp.tests.test_fast_math import TestFastMath
+    from warp.tests.test_fem import TestFem, TestFemShapeFunctions, TestFemUtilities
+    from warp.tests.test_fp16 import TestFp16
+    from warp.tests.test_func import TestFunc
+    from warp.tests.test_future_annotations import TestFutureAnnotations
+    from warp.tests.test_generics import TestGenerics
+    from warp.tests.test_grad import TestGrad
+    from warp.tests.test_grad_customs import TestGradCustoms
+    from warp.tests.test_grad_debug import TestGradDebug
+    from warp.tests.test_import import TestImport
+    from warp.tests.test_indexedarray import TestIndexedArray
+    from warp.tests.test_intersect import TestIntersect
+    from warp.tests.test_iter import TestIter
+    from warp.tests.test_large import TestLarge
+    from warp.tests.test_launch import TestLaunch
+    from warp.tests.test_lerp import TestLerp
+    from warp.tests.test_linear_solvers import TestLinearSolvers
+    from warp.tests.test_lvalue import TestLValue
+    from warp.tests.test_mat import TestMat
+    from warp.tests.test_mat_constructors import TestMatConstructors
+    from warp.tests.test_mat_lite import TestMatLite
+    from warp.tests.test_mat_scalar_ops import TestMatScalarOps
+    from warp.tests.test_math import TestMath
+    from warp.tests.test_module_hashing import TestModuleHashing
+    from warp.tests.test_modules_lite import TestModuleLite
+    from warp.tests.test_noise import TestNoise
+    from warp.tests.test_operators import TestOperators
+    from warp.tests.test_options import TestOptions
+    from warp.tests.test_overwrite import TestOverwrite
+    from warp.tests.test_print import TestPrint
+    from warp.tests.test_quat import TestQuat
+    from warp.tests.test_rand import TestRand
+    from warp.tests.test_reload import TestReload
+    from warp.tests.test_rounding import TestRounding
+    from warp.tests.test_runlength_encode import TestRunlengthEncode
+    from warp.tests.test_scalar_ops import TestScalarOps
+    from warp.tests.test_smoothstep import TestSmoothstep
+    from warp.tests.test_snippet import TestSnippets
+    from warp.tests.test_sparse import TestSparse
+    from warp.tests.test_spatial import TestSpatial
+    from warp.tests.test_special_values import TestSpecialValues
+    from warp.tests.test_static import TestStatic
+    from warp.tests.test_struct import TestStruct
+    from warp.tests.test_tape import TestTape
+    from warp.tests.test_transient_module import TestTransientModule
+    from warp.tests.test_triangle_closest_point import TestTriangleClosestPoint
+    from warp.tests.test_types import TestTypes
+    from warp.tests.test_utils import TestUtils
+    from warp.tests.test_vec import TestVec
+    from warp.tests.test_vec_constructors import TestVecConstructors
+    from warp.tests.test_vec_lite import TestVecLite
+    from warp.tests.test_vec_scalar_ops import TestVecScalarOps
+    from warp.tests.test_verify_fp import TestVerifyFP
+    from warp.tests.test_version import TestVersion
+    from warp.tests.tile.test_tile import TestTile
+    from warp.tests.tile.test_tile_atomic_bitwise import TestTileAtomicBitwise
+    from warp.tests.tile.test_tile_cholesky import TestTileCholesky
+    from warp.tests.tile.test_tile_load import TestTileLoad
+    from warp.tests.tile.test_tile_mathdx import TestTileMathDx
+    from warp.tests.tile.test_tile_matmul import TestTileMatmul
+    from warp.tests.tile.test_tile_reduce import TestTileReduce
+    from warp.tests.tile.test_tile_shared_memory import TestTileSharedMemory
+    from warp.tests.tile.test_tile_sort import TestTileSort
+    from warp.tests.tile.test_tile_view import TestTileView
+
+    test_classes = [
+        TestAdam,
+        TestArithmetic,
+        TestArray,
+        TestArrayReduce,
+        TestAsync,
+        TestAtomic,
+        TestAtomicBitwise,
+        TestAtomicCAS,
+        TestBool,
+        TestBuiltinsResolution,
+        TestBvh,
+        TestClosestPointEdgeEdgeMethods,
+        TestCodeGen,
+        TestCodeGenInstancing,
+        TestConditional,
+        TestConstants,
+        TestContext,
+        TestCopy,
+        TestCTypes,
+        TestDense,
+        TestDevices,
+        TestDLPack,
+        TestCoreExamples,
+        TestFemDiffusionExamples,
+        TestFemExamples,
+        TestOptimExamples,
+        TestFabricArray,
+        TestFastMath,
+        TestFem,
+        TestFemShapeFunctions,
+        TestFemUtilities,
+        TestFp16,
+        TestFunc,
+        TestFutureAnnotations,
+        TestGenerics,
+        TestGrad,
+        TestGradCustoms,
+        TestGradDebug,
+        TestHashGrid,
+        TestImport,
+        TestIndexedArray,
+        TestIntersect,
+        TestIter,
+        TestJax,
+        TestLarge,
+        TestLaunch,
+        TestLerp,
+        TestLinearSolvers,
+        TestLValue,
+        TestMarchingCubes,
+        TestMat,
+        TestMatConstructors,
+        TestMatLite,
+        TestMatScalarOps,
+        TestMath,
+        TestMempool,
+        TestMesh,
+        TestMeshQueryAABBMethods,
+        TestMeshQueryPoint,
+        TestMeshQueryRay,
+        TestModuleHashing,
+        TestModuleLite,
+        TestMultiGPU,
+        TestNoise,
+        TestOperators,
+        TestOptions,
+        TestOverwrite,
+        TestPeer,
+        TestPinned,
+        TestPrint,
+        TestQuat,
+        TestRand,
+        TestReload,
+        TestRounding,
+        TestRunlengthEncode,
+        TestScalarOps,
+        TestSmoothstep,
+        TestSnippets,
+        TestSparse,
+        TestSpatial,
+        TestSpecialValues,
+        TestStatic,
+        TestStreams,
+        TestStruct,
+        TestTape,
+        TestTile,
+        TestTileAtomicBitwise,
+        TestTileCholesky,
+        TestTileLoad,
+        TestTileMathDx,
+        TestTileMatmul,
+        TestTileReduce,
+        TestTileSharedMemory,
+        TestTileSort,
+        TestTileView,
+        TestTorch,
+        TestTransientModule,
+        TestTriangleClosestPoint,
+        TestTypes,
+        TestUtils,
+        TestVec,
+        TestVecConstructors,
+        TestVecLite,
+        TestVecScalarOps,
+        TestVerifyFP,
+        TestVersion,
+        TestVolume,
+        TestVolumeWrite,
+    ]
+
+    return _create_suite_from_test_classes(test_loader, test_classes)
+
+
+def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
+    """Tries to mimic the test suite used for testing omni.warp.core in Kit
+
+    Requires manual updates with test_ext.py for now.
+    """
+    from warp.tests.cuda.test_mempool import TestMempool
+    from warp.tests.cuda.test_peer import TestPeer
+    from warp.tests.cuda.test_pinned import TestPinned
+    from warp.tests.cuda.test_streams import TestStreams
+    from warp.tests.geometry.test_bvh import TestBvh
+    from warp.tests.geometry.test_hash_grid import TestHashGrid
+    from warp.tests.geometry.test_marching_cubes import TestMarchingCubes
+    from warp.tests.geometry.test_mesh import TestMesh
+    from warp.tests.geometry.test_mesh_query_aabb import TestMeshQueryAABBMethods
+    from warp.tests.geometry.test_mesh_query_point import TestMeshQueryPoint
+    from warp.tests.geometry.test_mesh_query_ray import TestMeshQueryRay
+    from warp.tests.geometry.test_volume import TestVolume
+    from warp.tests.geometry.test_volume_write import TestVolumeWrite
+    from warp.tests.interop.test_dlpack import TestDLPack
+    from warp.tests.test_array_reduce import TestArrayReduce
+    from warp.tests.test_bool import TestBool
+    from warp.tests.test_builtins_resolution import TestBuiltinsResolution
+    from warp.tests.test_codegen import TestCodeGen
+    from warp.tests.test_compile_consts import TestConstants
+    from warp.tests.test_conditional import TestConditional
+    from warp.tests.test_copy import TestCopy
+    from warp.tests.test_ctypes import TestCTypes
+    from warp.tests.test_devices import TestDevices
+    from warp.tests.test_fp16 import TestFp16
+    from warp.tests.test_func import TestFunc
+    from warp.tests.test_generics import TestGenerics
+    from warp.tests.test_grad_customs import TestGradCustoms
+    from warp.tests.test_grad_debug import TestGradDebug
+    from warp.tests.test_indexedarray import TestIndexedArray
+    from warp.tests.test_launch import TestLaunch
+    from warp.tests.test_lvalue import TestLValue
+    from warp.tests.test_mat_lite import TestMatLite
+    from warp.tests.test_math import TestMath
+    from warp.tests.test_module_aot import TestModuleAOT
+    from warp.tests.test_module_hashing import TestModuleHashing
+    from warp.tests.test_modules_lite import TestModuleLite
+    from warp.tests.test_noise import TestNoise
+    from warp.tests.test_operators import TestOperators
+    from warp.tests.test_rand import TestRand
+    from warp.tests.test_rounding import TestRounding
+    from warp.tests.test_runlength_encode import TestRunlengthEncode
+    from warp.tests.test_scalar_ops import TestScalarOps
+    from warp.tests.test_snippet import TestSnippets
+    from warp.tests.test_static import TestStatic
+    from warp.tests.test_tape import TestTape
+    from warp.tests.test_transient_module import TestTransientModule
+    from warp.tests.test_types import TestTypes
+    from warp.tests.test_utils import TestUtils
+    from warp.tests.test_vec_lite import TestVecLite
+    from warp.tests.tile.test_tile_reduce import TestTileReduce
+
+    test_classes = [
+        TestArrayReduce,
+        TestBool,
+        TestBuiltinsResolution,
+        TestBvh,
+        TestCodeGen,
+        TestConstants,
+        TestConditional,
+        TestCopy,
+        TestCTypes,
+        TestDevices,
+        TestDLPack,
+        TestFp16,
+        TestFunc,
+        TestGenerics,
+        TestGradCustoms,
+        TestGradDebug,
+        TestHashGrid,
+        TestIndexedArray,
+        TestLaunch,
+        TestLValue,
+        TestMarchingCubes,
+        TestMatLite,
+        TestMath,
+        TestMempool,
+        TestMesh,
+        TestMeshQueryAABBMethods,
+        TestMeshQueryPoint,
+        TestMeshQueryRay,
+        TestModuleAOT,
+        TestModuleHashing,
+        TestModuleLite,
+        TestNoise,
+        TestOperators,
+        TestPeer,
+        TestPinned,
+        TestRand,
+        TestRounding,
+        TestRunlengthEncode,
+        TestScalarOps,
+        TestSnippets,
+        TestStatic,
+        TestStreams,
+        TestTape,
+        TestTileReduce,
+        TestTransientModule,
+        TestTypes,
+        TestUtils,
+        TestVecLite,
+        TestVolume,
+        TestVolumeWrite,
+    ]
+
+    return _create_suite_from_test_classes(test_loader, test_classes)