fix: guard configure_deep_gemm_num_sms when JIT disabled (#20868)

Co-authored-by: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
This commit is contained in:
Gabriel Wu
2026-03-18 11:15:20 -07:00
committed by GitHub
co-authored by Claude Opus 4.6
parent a6c7bb54eb
commit 70876ae93b
@@ -122,7 +122,7 @@ def update_deep_gemm_config(gpu_id: int, server_args: ServerArgs):
@contextmanager
def configure_deep_gemm_num_sms(num_sms):
if num_sms is None:
if num_sms is None or not ENABLE_JIT_DEEPGEMM:
yield
else:
original_num_sms = deep_gemm.get_num_sms()