From f11164958088ef8786417e8f09c66574c4ce83ca Mon Sep 17 00:00:00 2001 From: Baizhou Zhang Date: Sat, 20 Sep 2025 00:20:33 -0700 Subject: [PATCH] Replace os.environ in layernorm.py (#10684) --- python/sglang/srt/layers/layernorm.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/sglang/srt/layers/layernorm.py b/python/sglang/srt/layers/layernorm.py index b766ff60b..87a392d55 100644 --- a/python/sglang/srt/layers/layernorm.py +++ b/python/sglang/srt/layers/layernorm.py @@ -14,7 +14,6 @@ """Fused operators for normalization layers.""" import logging -import os from typing import Optional, Tuple, Union import torch @@ -81,7 +80,7 @@ class RMSNorm(CustomOp): ) if _use_aiter: self._forward_method = self.forward_aiter - if os.environ["SGLANG_ENABLE_DETERMINISTIC_INFERENCE"] == "1": + if get_bool_env_var("SGLANG_ENABLE_DETERMINISTIC_INFERENCE"): self._forward_method = self.forward_native def forward_cuda(