warp-lang 1.0.2__py3-none-win_amd64.whl → 1.2.0__py3-none-win_amd64.whl

warp/tests/unused_test_misc.py

@@ -1,71 +1,69 @@
-# Copyright (c) 2021 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 numpy as np
-
-import warp as wp
-
-wp.init()
-
-
-@wp.kernel
-def arange(out: wp.array(dtype=int)):
-    tid = wp.tid()
-    out[tid] = tid
-
-
-device = "cuda:0"
-cmds = []
-
-n = 10
-arrays = []
-
-for i in range(5):
-    arrays.append(wp.zeros(n, dtype=int, device=device))
-
-# setup CUDA graph
-wp.capture_begin()
-
-# launch kernels and keep command object around
-for i in range(5):
-    cmd = wp.launch(arange, dim=n, inputs=[arrays[i]], device=device, record_cmd=True)
-    cmds.append(cmd)
-
-graph = wp.capture_end()
-
-# ---------------------------------------
-
-ref = np.arange(0, n, dtype=int)
-wp.capture_launch(graph)
-
-for i in range(5):
-    print(arrays[i].numpy())
-
-
-# ---------------------------------------
-
-n = 16
-arrays = []
-
-for i in range(5):
-    arrays.append(wp.zeros(n, dtype=int, device=device))
-
-# update graph params
-for i in range(5):
-    cmd.set_dim(n)
-    cmd.set_param(arrays[i])
-
-    cmd.update_graph()
-
-
-wp.capture_launch(graph)
-wp.synchronize()
-
-ref = np.arange(0, n, dtype=int)
-
-for i in range(5):
-    print(arrays[i].numpy())
+# Copyright (c) 2021 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 numpy as np
+
+import warp as wp
+
+
+@wp.kernel
+def arange(out: wp.array(dtype=int)):
+    tid = wp.tid()
+    out[tid] = tid
+
+
+device = "cuda:0"
+cmds = []
+
+n = 10
+arrays = []
+
+for _i in range(5):
+    arrays.append(wp.zeros(n, dtype=int, device=device))
+
+# setup CUDA graph
+wp.capture_begin()
+
+# launch kernels and keep command object around
+for i in range(5):
+    cmd = wp.launch(arange, dim=n, inputs=[arrays[i]], device=device, record_cmd=True)
+    cmds.append(cmd)
+
+graph = wp.capture_end()
+
+# ---------------------------------------
+
+ref = np.arange(0, n, dtype=int)
+wp.capture_launch(graph)
+
+for i in range(5):
+    print(arrays[i].numpy())
+
+
+# ---------------------------------------
+
+n = 16
+arrays = []
+
+for _i in range(5):
+    arrays.append(wp.zeros(n, dtype=int, device=device))
+
+# update graph params
+for i in range(5):
+    cmd.set_dim(n)
+    cmd.set_param(arrays[i])
+
+    cmd.update_graph()
+
+
+wp.capture_launch(graph)
+wp.synchronize()
+
+ref = np.arange(0, n, dtype=int)
+
+for i in range(5):
+    print(arrays[i].numpy())

warp/tests/walkthrough_debug.py

@@ -1,85 +1,85 @@
-# Copyright (c) 2023 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.
-
-####################################################################################################
-#
-# This file demonstrates step-through debugging support of the C++ code generated for a Warp kernel
-# running on the CPU.
-#
-# This is not a unit test; it should be run interactively.
-#
-# For a fully integrated experience use Visual Studio Code and install the "Python C++ Debugger"
-# and "CodeLLDB" extensions. Add the following configurations to your .vscode/launch.json file:
-#
-
-"""
-        {
-            "name": "Warp Debugger",
-            "type": "pythoncpp",
-            "request": "launch",
-            "pythonLaunchName": "Python: Current File",
-            "cppAttachName": "(lldb) Attach",
-        },
-        {
-            "name": "Python: Current File",
-            "type": "python",
-            "request": "launch",
-            "program": "${file}",
-            "console": "integratedTerminal",
-            "stopOnEntry": false,
-        },
-        {
-            "name": "(lldb) Attach",
-            "type": "lldb",
-            "request": "attach",
-        },
-"""
-
-#
-# Then run this .py file using the "Warp Debugger" configuration.
-#
-# Check out the following resources for more information about launch configurations and
-# troubleshooting common VSCode debugger integration issues:
-# • https://vscode-docs.readthedocs.io/en/stable/editor/debugging/#launch-configurations
-# • https://code.visualstudio.com/docs/cpp/cpp-debug#_debugging
-#
-####################################################################################################
-
-import warp as wp
-
-# The init() function prints the directory of the kernel cache which contains the .cpp files
-# generated from Warp kernels. You can put breakpoints in these C++ files through Visual Studio Code,
-# but it's generally more convenient to use wp.breakpoint(). See the example below.
-wp.init()
-
-# Enable kernels to be compiled with debug info and disable optimizations
-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"
-
-
-@wp.kernel
-def example_breakpoint(n: int):
-    a = int(0)
-
-    for i in range(0, n):
-        if a == 5:
-            # Your debugger should halt at the C++ code corresponding with the next line,
-            # namely a call to the __debugbreak() intrinsic function.
-            wp.breakpoint()
-
-            break
-
-        a += 1
-
-    wp.expect_eq(a, 5)
-
-
-if __name__ == "__main__":
-    wp.build.clear_kernel_cache()
-    wp.launch(example_breakpoint, dim=1, inputs=[10], device="cpu")
+# Copyright (c) 2023 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.
+
+####################################################################################################
+#
+# This file demonstrates step-through debugging support of the C++ code generated for a Warp kernel
+# running on the CPU.
+#
+# This is not a unit test; it should be run interactively.
+#
+# For a fully integrated experience use Visual Studio Code and install the "Python C++ Debugger"
+# and "CodeLLDB" extensions. Add the following configurations to your .vscode/launch.json file:
+#
+
+"""
+{
+    "name": "Warp Debugger",
+    "type": "pythoncpp",
+    "request": "launch",
+    "pythonLaunchName": "Python: Current File",
+    "cppAttachName": "(lldb) Attach",
+},
+{
+    "name": "Python: Current File",
+    "type": "debugpy",
+    "request": "launch",
+    "program": "${file}",
+    "console": "integratedTerminal",
+    "stopOnEntry": false,
+},
+{
+    "name": "(lldb) Attach",
+    "type": "lldb",
+    "request": "attach",
+},
+"""
+
+#
+# Then run this .py file using the "Warp Debugger" configuration.
+#
+# Check out the following resources for more information about launch configurations and
+# troubleshooting common VSCode debugger integration issues:
+# • https://vscode-docs.readthedocs.io/en/stable/editor/debugging/#launch-configurations
+# • https://code.visualstudio.com/docs/cpp/cpp-debug#_debugging
+#
+####################################################################################################
+
+import warp as wp
+
+# The init() function prints the directory of the kernel cache which contains the .cpp files
+# generated from Warp kernels. You can put breakpoints in these C++ files through Visual Studio Code,
+# but it's generally more convenient to use wp.breakpoint(). See the example below.
+wp.init()
+
+# Enable kernels to be compiled with debug info and disable optimizations
+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"
+
+
+@wp.kernel
+def example_breakpoint(n: int):
+    a = int(0)
+
+    for _i in range(0, n):
+        if a == 5:
+            # Your debugger should halt at the C++ code corresponding with the next line,
+            # namely a call to the __debugbreak() intrinsic function.
+            wp.breakpoint()
+
+            break
+
+        a += 1
+
+    wp.expect_eq(a, 5)
+
+
+if __name__ == "__main__":
+    wp.build.clear_kernel_cache()
+    wp.launch(example_breakpoint, dim=1, inputs=[10], device="cpu")