warp-lang 1.0.1__py3-none-win_amd64.whl → 1.1.0__py3-none-win_amd64.whl

warp/config.py

@@ -1,36 +1,40 @@
-# Copyright (c) 2022 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.
-
-version = "1.0.1"
-
-verify_fp = False  # verify inputs and outputs are finite after each launch
-verify_cuda = False  # if true will check CUDA errors after each kernel launch / memory operation
-print_launches = False  # if true will print out launch information
-
-mode = "release"
-verbose = False  # print extra informative messages
-verbose_warnings = False  # whether file and line info gets included in Warp warnings
-quiet = False  # suppress all output except errors and warnings
-
-cache_kernels = True
-kernel_cache_dir = None  # path to kernel cache directory, if None a default path will be used
-
-cuda_output = (
-    None  # preferred CUDA output format for kernels ("ptx" or "cubin"), determined automatically if unspecified
-)
-
-ptx_target_arch = 70  # target architecture for PTX generation, defaults to the lowest architecture that supports all of Warp's features
-
-enable_backward = True  # whether to compiler the backward passes of the kernels
-
-llvm_cuda = False  # use Clang/LLVM instead of NVRTC to compile CUDA
-
-enable_graph_capture_module_load_by_default = True  # Default value of force_module_load for capture_begin()
-
-enable_mempools_at_init = True  # Whether CUDA devices will be initialized with mempools enabled (if supported)
-
-max_unroll = 16
+# Copyright (c) 2022 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.
+
+from typing import Optional
+
+version: str = "1.1.0"
+
+verify_fp: bool = False  # verify inputs and outputs are finite after each launch
+verify_cuda: bool = False  # if true will check CUDA errors after each kernel launch / memory operation
+print_launches: bool = False  # if true will print out launch information
+
+mode: str = "release"
+verbose: bool = False  # print extra informative messages
+verbose_warnings: bool = False  # whether file and line info gets included in Warp warnings
+quiet: bool = False  # suppress all output except errors and warnings
+
+cache_kernels: bool = True
+kernel_cache_dir: bool = None  # path to kernel cache directory, if None a default path will be used
+
+cuda_output: Optional[str] = (
+    None  # preferred CUDA output format for kernels ("ptx" or "cubin"), determined automatically if unspecified
+)
+
+ptx_target_arch: int = 70  # target architecture for PTX generation, defaults to the lowest architecture that supports all of Warp's features
+
+enable_backward: bool = True  # whether to compiler the backward passes of the kernels
+
+llvm_cuda: bool = False  # use Clang/LLVM instead of NVRTC to compile CUDA
+
+enable_graph_capture_module_load_by_default: bool = (
+    True  # Default value of force_module_load for capture_begin() if CUDA driver does not support at least CUDA 12.3
+)
+
+enable_mempools_at_init: bool = True  # Whether CUDA devices will be initialized with mempools enabled (if supported)
+
+max_unroll: int = 16

warp/constants.py

@@ -1,45 +1,45 @@
-# 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.
-
-import math
-
-from warp.types import constant
-
-__all__ = [
-    "E",
-    "e",
-    "INF",
-    "inf",
-    "LOG2E",
-    "log2e",
-    "LOG10E",
-    "log10e",
-    "LN2",
-    "ln2",
-    "LN10",
-    "ln10",
-    "PHI",
-    "phi",
-    "PI",
-    "pi",
-    "HALF_PI",
-    "half_pi",
-    "TAU",
-    "tau",
-]
-
-E = e = constant(2.71828182845904523536)  # e
-LOG2E = log2e = constant(1.44269504088896340736)  # log2(e)
-LOG10E = log10e = constant(0.43429448190325182765)  # log10(e)
-LN2 = ln2 = constant(0.69314718055994530942)  # ln(2)
-LN10 = ln10 = constant(2.30258509299404568402)  # ln(10)
-PHI = phi = constant(1.61803398874989484820)  # golden constant
-PI = pi = constant(3.14159265358979323846)  # pi
-HALF_PI = half_pi = constant(1.57079632679489661923)  # half pi
-TAU = tau = constant(6.28318530717958647692)  # 2 * pi
-
-INF = inf = constant(math.inf)
+# 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.
+
+import math
+
+from warp.types import constant
+
+__all__ = [
+    "E",
+    "e",
+    "INF",
+    "inf",
+    "LOG2E",
+    "log2e",
+    "LOG10E",
+    "log10e",
+    "LN2",
+    "ln2",
+    "LN10",
+    "ln10",
+    "PHI",
+    "phi",
+    "PI",
+    "pi",
+    "HALF_PI",
+    "half_pi",
+    "TAU",
+    "tau",
+]
+
+E = e = constant(2.71828182845904523536)  # e
+LOG2E = log2e = constant(1.44269504088896340736)  # log2(e)
+LOG10E = log10e = constant(0.43429448190325182765)  # log10(e)
+LN2 = ln2 = constant(0.69314718055994530942)  # ln(2)
+LN10 = ln10 = constant(2.30258509299404568402)  # ln(10)
+PHI = phi = constant(1.61803398874989484820)  # golden constant
+PI = pi = constant(3.14159265358979323846)  # pi
+HALF_PI = half_pi = constant(1.57079632679489661923)  # half pi
+TAU = tau = constant(6.28318530717958647692)  # 2 * pi
+
+INF = inf = constant(math.inf)