Environment VariablesΒΆ

Environment Variable

Information

CUROBO_TORCH_COMPILE_DISABLE

CUROBO_TORCH_COMPILE_DISABLE=0 will enable kernel fusion with torch.compile. If this variable is not set, then kernel fusion is disabled.

CUROBO_TORCH_COMPILE_FORCE

CUROBO_TORCH_COMPILE_FORCE=1 will force running of kernel fusion with torch.compile bypassing any checks in cuRobo.

CUROBO_TORCH_CUDA_GRAPH_RESET

CUROBO_TORCH_CUDA_GRAPH_RESET=1 will enable resetting of CUDA graph when memory references change. Only supported in PyTorch 2.0 and above.

CUROBO_USE_LRU_CACHE

CUROBO_USE_LRU_CACHE=1 will enable LRU cache for warp templated kernel functions. If this variable is not set, then LRU cache is disabled.

CUROBO_WARP_RUNTIME_KERNEL_DISABLE

CUROBO_WARP_RUNTIME_KERNEL_DISABLE=1 will disable warp runtime kernel generation. If this variable is not set, then warp runtime kernel generation is enabled.