mirror of
https://github.com/NVIDIA/TensorRT-LLM.git
synced 2026-02-06 03:01:50 +08:00
22 lines
694 B
Python
22 lines
694 B
Python
# This file is vendored from the Triton project. DO NOT EDIT THIS FILE DIRECTLY.
|
|
# Source: https://github.com/triton-lang/triton/tree/v3.5.1/python/triton_kernels/triton_kernels/proton_opts.py
|
|
# Triton is licensed under the MIT License.
|
|
|
|
# proton options
|
|
|
|
import os
|
|
|
|
_launch_metadata_allow_sync = None
|
|
|
|
|
|
def launch_metadata_allow_sync():
|
|
global _launch_metadata_allow_sync
|
|
if _launch_metadata_allow_sync is None:
|
|
_launch_metadata_allow_sync = not (os.getenv("PROTON_LAUNCH_METADATA_NOSYNC") == "1")
|
|
return _launch_metadata_allow_sync
|
|
|
|
|
|
def set_launch_metadata_allow_sync(allow_sync: bool):
|
|
global _launch_metadata_allow_sync
|
|
_launch_metadata_allow_sync = allow_sync
|