warp-lang 1.8.0__py3-none-manylinux_2_34_aarch64.whl → 1.9.0__py3-none-manylinux_2_34_aarch64.whl

warp/tests/unittest_suites.py

@@ -164,6 +164,7 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
     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
@@ -193,13 +194,19 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
     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.tile.test_tile import TestTile
+    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,
@@ -256,6 +263,7 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
         TestLValue,
         TestMarchingCubes,
         TestMat,
+        TestMatConstructors,
         TestMatLite,
         TestMatScalarOps,
         TestMath,
@@ -293,15 +301,21 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
         TestStruct,
         TestTape,
         TestTile,
+        TestTileCholesky,
+        TestTileLoad,
         TestTileMathDx,
+        TestTileMatmul,
         TestTileReduce,
         TestTileSharedMemory,
+        TestTileSort,
+        TestTileView,
         TestTorch,
         TestTransientModule,
         TestTriangleClosestPoint,
         TestTypes,
         TestUtils,
         TestVec,
+        TestVecConstructors,
         TestVecLite,
         TestVecScalarOps,
         TestVerifyFP,
@@ -350,6 +364,7 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
     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
@@ -396,6 +411,7 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
         TestMeshQueryAABBMethods,
         TestMeshQueryPoint,
         TestMeshQueryRay,
+        TestModuleAOT,
         TestModuleHashing,
         TestModuleLite,
         TestNoise,

warp/tests/walkthrough_debug.py

@@ -68,7 +68,7 @@ wp.init()
 wp.config.mode = "debug"
 
 # Make sure Warp was built with `build_lib.py --mode=debug`
-assert wp.context.runtime.core.is_debug_enabled(), "Warp must be built in debug mode to enable debugging kernels"
+assert wp.context.runtime.core.wp_is_debug_enabled(), "Warp must be built in debug mode to enable debugging kernels"
 
 
 @wp.kernel

warp/thirdparty/unittest_parallel.py

@@ -28,6 +28,7 @@ import sys
 import tempfile
 import time
 import unittest
+from concurrent.futures.process import BrokenProcessPool
 from contextlib import contextmanager
 from io import StringIO
 
@@ -160,6 +161,7 @@ def main(argv=None):
     group_warp.add_argument(
         "--no-shared-cache", action="store_true", help="Use a separate kernel cache per test process."
     )
+    group_warp.add_argument("--warp-debug", action="store_true", help="Set warp.config.mode to 'debug'")
     args = parser.parse_args(args=argv)
 
     if args.coverage_branch:
@@ -242,15 +244,76 @@ def main(argv=None):
                     test_manager = ParallelTestManager(manager, args, temp_dir)
                     results = pool.map(test_manager.run_tests, test_suites)
             else:
-                # NVIDIA Modification added concurrent.futures
-                with concurrent.futures.ProcessPoolExecutor(
-                    max_workers=process_count,
-                    mp_context=multiprocessing.get_context(method="spawn"),
-                    initializer=initialize_test_process,
-                    initargs=(manager.Lock(), shared_index, args, temp_dir),
-                ) as executor:
-                    test_manager = ParallelTestManager(manager, args, temp_dir)
-                    results = list(executor.map(test_manager.run_tests, test_suites, timeout=2400))
+                # NVIDIA Modification: added concurrent.futures with crash handling and per-suite isolated fallback
+                results = []
+                parallel_failed = False
+
+                try:
+                    with concurrent.futures.ProcessPoolExecutor(
+                        max_workers=process_count,
+                        mp_context=multiprocessing.get_context(method="spawn"),
+                        initializer=initialize_test_process,
+                        initargs=(manager.Lock(), shared_index, args, temp_dir),
+                    ) as executor:
+                        test_manager = ParallelTestManager(manager, args, temp_dir)
+                        # Try parallel execution first using the original map approach
+                        results = list(executor.map(test_manager.run_tests, test_suites, timeout=2400))
+
+                except BrokenProcessPool:
+                    # Process pool is broken - switch to isolated single-process fallback
+                    print(
+                        "Warning: Process pool broken during parallel execution. Switching to isolated single-process fallback.",
+                        file=sys.stderr,
+                    )
+                    parallel_failed = True
+                except Exception as e:
+                    # Handle other pool-level exceptions
+                    print(
+                        f"Warning: Process pool error: {e}. Switching to isolated single-process fallback.",
+                        file=sys.stderr,
+                    )
+                    parallel_failed = True
+
+                # Fallback to isolated single-process execution if parallel failed
+                if parallel_failed:
+                    print("Running all tests in isolated single-process mode...", file=sys.stderr)
+                    # Run all test suites in isolated single-process pools
+                    results = []
+                    for i, suite in enumerate(test_suites):
+                        try:
+                            # Create a new single-process pool for each test suite
+                            with concurrent.futures.ProcessPoolExecutor(
+                                max_workers=1,
+                                mp_context=multiprocessing.get_context(method="spawn"),
+                                initializer=initialize_test_process,
+                                initargs=(manager.Lock(), shared_index, args, temp_dir),
+                            ) as executor:
+                                test_manager = ParallelTestManager(manager, args, temp_dir)
+                                future = executor.submit(test_manager.run_tests, suite)
+                                try:
+                                    result = future.result(timeout=2400)
+                                    results.append(result)
+                                except BrokenProcessPool:
+                                    print(
+                                        f"Warning: Process crashed or was terminated unexpectedly in isolated execution for test suite {i + 1}/{len(test_suites)}. Marking tests as crashed.",
+                                        file=sys.stderr,
+                                    )
+                                    crash_result = create_crash_result(suite)
+                                    results.append(crash_result)
+                                except Exception as e:
+                                    print(
+                                        f"Warning: Error in isolated test suite {i + 1}/{len(test_suites)}: {e}. Marking tests as crashed.",
+                                        file=sys.stderr,
+                                    )
+                                    error_result = create_crash_result(suite)
+                                    results.append(error_result)
+                        except Exception as e:
+                            print(
+                                f"Warning: Failed to create isolated process for test suite {i + 1}/{len(test_suites)}: {e}. Marking tests as crashed.",
+                                file=sys.stderr,
+                            )
+                            error_result = create_crash_result(suite)
+                            results.append(error_result)
         else:
             # This entire path is an NVIDIA Modification
 
@@ -259,6 +322,9 @@ def main(argv=None):
             if args.verbose > 1:
                 print(file=sys.stderr)
 
+            if args.warp_debug:
+                wp.config.mode = "debug"
+
             # Run the tests in serial
             start_time = time.perf_counter()
 
@@ -423,6 +489,36 @@ def _iter_test_cases(test_suite):
             yield from _iter_test_cases(suite)
 
 
+def create_crash_result(test_suite):
+    """Create a result indicating the process crashed or was terminated unexpectedly while running this test suite.
+
+    This entire function is an NVIDIA modification.
+    """
+    test_count = test_suite.countTestCases()
+    crash_errors = []
+
+    # Create crash error entries for each test in the suite
+    # Note: We don't know which specific test caused the crash, just that the process crashed
+    for test in _iter_test_cases(test_suite):
+        error_msg = (
+            "Process crashed or was terminated unexpectedly while running this test suite "
+            f"(unknown which test caused the crash): {test}"
+        )
+        crash_errors.append(
+            "\n".join(
+                [
+                    unittest.TextTestResult.separator1,
+                    str(test),
+                    unittest.TextTestResult.separator2,
+                    error_msg,
+                ]
+            )
+        )
+
+    # Return the same format as run_tests: (test_count, errors, failures, skipped, expected_failures, unexpected_successes, test_records)
+    return (test_count, crash_errors, [], 0, 0, 0, [])
+
+
 class ParallelTestManager:
     def __init__(self, manager, args, temp_dir):
         self.args = args
@@ -547,6 +643,9 @@ def initialize_test_process(lock, shared_index, args, temp_dir):
     with _coverage(args, temp_dir):
         import warp as wp
 
+        if args.warp_debug:
+            wp.config.mode = "debug"
+
         if args.no_shared_cache:
             from warp.thirdparty import appdirs