NvidiaWarp-GarmentCode/warp/config.py

# 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.0-beta.6"

cuda_path = (
    None  # path to local CUDA toolchain, if None at init time warp will attempt to find the SDK using CUDA_PATH env var
)

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
quiet = False  # suppress all output except errors and warnings

host_compiler = None  # user can specify host compiler here, otherwise will attempt to find one automatically

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

graph_capture_module_load_default = True  # Default value of force_module_load for capture_begin()