Revert "Revert "enable csgmv automatically on cuda"" (#14277)

This commit is contained in:
b8zhong
2025-12-03 13:12:30 -08:00
committed by GitHub
parent 8ab5d8b40d
commit 9d82340298
2 changed files with 7 additions and 0 deletions

View File

@@ -3711,6 +3711,12 @@ def cached_triton_kernel(key_fn=None):
"""
def decorator(fn):
# Auto-enable the custom kernel cache for CUDA, where it is
# known to be compatible.
if is_cuda() and not envs.SGLANG_USE_CUSTOM_TRITON_KERNEL_CACHE.is_set():
logger.debug("Detected platform CUDA, using custom triton kernel cache.")
return CachedKernel(fn, key_fn)
if envs.SGLANG_USE_CUSTOM_TRITON_KERNEL_CACHE.get():
logger.debug(
f"{envs.SGLANG_USE_CUSTOM_TRITON_KERNEL_CACHE.name} = True. Using custom triton kernel cache."