Merge branch 'ccrouzet/init-with-tape' into 'main'

Initialize Warp When Entering a `wp.Tape` Context

See merge request omniverse/warp!766

CHANGELOG.md

@@ -72,6 +72,7 @@
 - Fix printing Boolean and 8-bit integer values.
 - Fix array interface type strings used for Boolean and 8-bit integer values.
 - Fix initialization error when setting struct members.
+- Fix Warp not being initialized upon entering a `wp.Tape` context.
 
 ## [1.3.3] - 2024-09-04
 

warp/tape.py

@@ -50,6 +50,8 @@ def __init__(self):
         self.loss = None
 
     def __enter__(self):
+        wp.context.init()
+
         if wp.context.runtime.tape is not None:
             raise RuntimeError("Warp: Error, entering a tape while one is already active")
 

warp/tests/test_implicit_init.py

@@ -375,6 +375,27 @@ class TestImplicitInitStructMemberInit(unittest.TestCase):
 )
 
 
+#   Tape
+# ------------------------------------------------------------------------------
+
+
+def test_tape(test, device):
+    with wp.Tape():
+        pass
+
+
+class TestImplicitInitTape(unittest.TestCase):
+    pass
+
+
+add_function_test(
+    TestImplicitInitTape,
+    "test_tape",
+    test_tape,
+    check_output=False,
+)
+
+
 if __name__ == "__main__":
     # Do not clear the kernel cache or call anything that would initialize Warp
     # since these tests are specifically aiming to catch issues where Warp isn't