warp-lang 1.5.0__py3-none-manylinux2014_x86_64.whl → 1.6.0__py3-none-manylinux2014_x86_64.whl

warp/tests/unittest_suites.py

@@ -167,7 +167,8 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
     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_sim_grad import TestSimGradients
+
+    # from warp.tests.test_sim_grad import TestSimGradients Disabled, flaky
     from warp.tests.test_sim_kinematics import TestSimKinematics
     from warp.tests.test_smoothstep import TestSmoothstep
     from warp.tests.test_snippet import TestSnippets
@@ -276,7 +277,7 @@ def default_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader)
         TestRounding,
         TestRunlengthEncode,
         TestScalarOps,
-        TestSimGradients,
+        # TestSimGradients, Disabled, flaky
         TestSimKinematics,
         TestSmoothstep,
         TestSnippets,
@@ -315,15 +316,18 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
     """
     from warp.tests.test_array import TestArray
     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_bvh import TestBvh
     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_copy import TestCopy
     from warp.tests.test_ctypes import TestCTypes
     from warp.tests.test_devices import TestDevices
     from warp.tests.test_dlpack import TestDLPack
     from warp.tests.test_fabricarray import TestFabricArray
+    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
@@ -331,26 +335,35 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
     from warp.tests.test_hash_grid import TestHashGrid
     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_marching_cubes import TestMarchingCubes
     from warp.tests.test_mat_lite import TestMatLite
     from warp.tests.test_math import TestMath
     from warp.tests.test_matmul_lite import TestMatmulLite
+    from warp.tests.test_mempool import TestMempool
     from warp.tests.test_mesh import TestMesh
     from warp.tests.test_mesh_query_aabb import TestMeshQueryAABBMethods
     from warp.tests.test_mesh_query_point import TestMeshQueryPoint
     from warp.tests.test_mesh_query_ray import TestMeshQueryRay
+    from warp.tests.test_mlp import TestMLP
     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_peer import TestPeer
+    from warp.tests.test_pinned import TestPinned
     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_snippet import TestSnippets
     from warp.tests.test_sparse import TestSparse
     from warp.tests.test_static import TestStatic
     from warp.tests.test_streams import TestStreams
     from warp.tests.test_tape import TestTape
+    from warp.tests.test_tile_reduce import TestTileReduce
     from warp.tests.test_transient_module import TestTransientModule
     from warp.tests.test_types import TestTypes
     from warp.tests.test_utils import TestUtils
@@ -361,15 +374,18 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
     test_classes = [
         TestArray,
         TestArrayReduce,
+        TestBool,
+        TestBuiltinsResolution,
         TestBvh,
         TestCodeGen,
-        TestCodeGenInstancing,
         TestConstants,
         TestConditional,
+        TestCopy,
         TestCTypes,
         TestDevices,
         TestDLPack,
         TestFabricArray,
+        TestFp16,
         TestFunc,
         TestGenerics,
         TestGradCustoms,
@@ -377,26 +393,35 @@ def kit_suite(test_loader: unittest.TestLoader = unittest.defaultTestLoader):
         TestHashGrid,
         TestIndexedArray,
         TestLaunch,
+        TestLValue,
         TestMarchingCubes,
         TestMatLite,
         TestMath,
         TestMatmulLite,
+        TestMempool,
         TestMesh,
         TestMeshQueryAABBMethods,
         TestMeshQueryPoint,
         TestMeshQueryRay,
+        TestMLP,
         TestModuleHashing,
         TestModuleLite,
         TestNoise,
         TestOperators,
+        TestPeer,
+        TestPinned,
         TestQuat,
         TestRand,
+        TestReload,
         TestRounding,
         TestRunlengthEncode,
+        TestScalarOps,
+        TestSnippets,
         TestSparse,
         TestStatic,
         TestStreams,
         TestTape,
+        TestTileReduce,
         TestTransientModule,
         TestTypes,
         TestUtils,

warp/tests/unittest_utils.py

@@ -7,7 +7,7 @@
 
 import ctypes
 import ctypes.util
-import importlib
+import importlib.util
 import os
 import sys
 import time
@@ -58,7 +58,6 @@ def get_selected_cuda_test_devices(mode: Optional[str] = None):
     """
 
     if mode is None:
-        global test_mode
         mode = test_mode
 
     if mode == "basic":
@@ -98,7 +97,6 @@ def get_test_devices(mode: Optional[str] = None):
           "all": Returns all available devices.
     """
     if mode is None:
-        global test_mode
         mode = test_mode
 
     devices = []
@@ -128,8 +126,13 @@ def get_cuda_test_devices(mode=None):
     return [d for d in devices if d.is_cuda]
 
 
-# redirects and captures all stdout output (including from C-libs)
-class StdOutCapture:
+class StreamCapture:
+    def __init__(self, stream_name):
+        self.stream_name = stream_name  # 'stdout' or 'stderr'
+        self.saved = None
+        self.target = None
+        self.tempfile = None
+
     def begin(self):
         # Flush the stream buffers managed by libc.
         # This is needed at the moment due to Carbonite not flushing the logs
@@ -137,14 +140,15 @@ class StdOutCapture:
         if LIBC is not None:
             LIBC.fflush(None)
 
-        # save original
-        self.saved = sys.stdout
+        # Get the stream object (sys.stdout or sys.stderr)
+        self.saved = getattr(sys, self.stream_name)
         self.target = os.dup(self.saved.fileno())
 
         # create temporary capture stream
         import io
         import tempfile
 
+        # Create temporary capture stream
         self.tempfile = io.TextIOWrapper(
             tempfile.TemporaryFile(buffering=0),
             encoding="utf-8",
@@ -153,33 +157,46 @@ class StdOutCapture:
             write_through=True,
         )
 
+        # Redirect the stream
         os.dup2(self.tempfile.fileno(), self.saved.fileno())
-
-        sys.stdout = self.tempfile
+        setattr(sys, self.stream_name, self.tempfile)
 
     def end(self):
         # The following sleep doesn't seem to fix the test_print failure on Windows
         # if sys.platform == "win32":
         #    # Workaround for what seems to be a Windows-specific bug where
-        #    # the output of CUDA's `printf` is not being immediately flushed
-        #    # despite the context synchronisation.
+        #    # the output of CUDA's printf is not being immediately flushed
+        #    # despite the context synchronization.
         #    time.sleep(0.01)
-
         if LIBC is not None:
             LIBC.fflush(None)
 
+        # Restore the original stream
         os.dup2(self.target, self.saved.fileno())
         os.close(self.target)
 
+        # Read the captured output
         self.tempfile.seek(0)
         res = self.tempfile.buffer.read()
         self.tempfile.close()
 
-        sys.stdout = self.saved
+        # Restore the stream object
+        setattr(sys, self.stream_name, self.saved)
 
         return str(res.decode("utf-8"))
 
 
+# Subclasses for specific streams
+class StdErrCapture(StreamCapture):
+    def __init__(self):
+        super().__init__("stderr")
+
+
+class StdOutCapture(StreamCapture):
+    def __init__(self):
+        super().__init__("stdout")
+
+
 class CheckOutput:
     def __init__(self, test):
         self.test = test

warp/thirdparty/unittest_parallel.py

@@ -306,9 +306,9 @@ def main(argv=None):
 
         # Test report
         print(unittest.TextTestResult.separator2, file=sys.stderr)
-        print(f'Ran {tests_run} {"tests" if tests_run > 1 else "test"} in {test_duration:.3f}s', file=sys.stderr)
+        print(f"Ran {tests_run} {'tests' if tests_run > 1 else 'test'} in {test_duration:.3f}s", file=sys.stderr)
         print(file=sys.stderr)
-        print(f'{"OK" if is_success else "FAILED"}{" (" + ", ".join(infos) + ")" if infos else ""}', file=sys.stderr)
+        print(f"{'OK' if is_success else 'FAILED'}{' (' + ', '.join(infos) + ')' if infos else ''}", file=sys.stderr)
 
         if test_records and args.junit_report_xml:
             # NVIDIA modification to report results in Junit XML format