Skip to content

Commit b2e05e3

Browse files
nvjullindivakar-amd
authored andcommitted
[Bugfix] Added more env vars to hash (vllm-project#22449)
Signed-off-by: Julien Lin <[email protected]>
1 parent 0a09281 commit b2e05e3

File tree

1 file changed

+36
-10
lines changed

1 file changed

+36
-10
lines changed

vllm/envs.py

Lines changed: 36 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1206,14 +1206,6 @@ def compute_hash() -> str:
12061206
affect the choice of different kernels or attention backends should
12071207
also be included in the factors list.
12081208
"""
1209-
factors: list[Any] = []
1210-
1211-
# summarize environment variables
1212-
def factorize(name: str):
1213-
if __getattr__(name):
1214-
factors.append(__getattr__(name))
1215-
else:
1216-
factors.append("None")
12171209

12181210
# The values of envs may affects the computation graph.
12191211
# TODO(DefTruth): hash all environment variables?
@@ -1228,11 +1220,45 @@ def factorize(name: str):
12281220
"VLLM_DP_SIZE",
12291221
"VLLM_USE_STANDALONE_COMPILE",
12301222
"VLLM_FUSED_MOE_CHUNK_SIZE",
1223+
"VLLM_FLASHINFER_MOE_BACKEND",
1224+
"VLLM_V1_USE_PREFILL_DECODE_ATTENTION",
1225+
"VLLM_USE_AITER_UNIFIED_ATTENTION",
1226+
"VLLM_ATTENTION_BACKEND",
1227+
"VLLM_USE_FLASHINFER_SAMPLER",
1228+
"VLLM_FLASHINFER_FORCE_TENSOR_CORES",
1229+
"VLLM_DISABLED_KERNELS",
1230+
"VLLM_USE_DEEP_GEMM",
12311231
"VLLM_USE_TRTLLM_FP4_GEMM",
1232+
"VLLM_USE_FLASHINFER_MOE_FP8",
1233+
"VLLM_USE_FLASHINFER_MOE_FP4",
1234+
"VLLM_USE_FLASHINFER_MOE_MXFP4_MXFP8",
1235+
"VLLM_USE_FLASHINFER_MOE_MXFP4_BF16",
1236+
"VLLM_USE_CUDNN_PREFILL",
1237+
"VLLM_USE_TRTLLM_ATTENTION",
1238+
"VLLM_ROCM_USE_AITER",
1239+
"VLLM_ROCM_USE_AITER_PAGED_ATTN",
1240+
"VLLM_ROCM_USE_AITER_LINEAR",
1241+
"VLLM_ROCM_USE_AITER_MOE",
1242+
"VLLM_ROCM_USE_AITER_RMSNORM",
1243+
"VLLM_ROCM_USE_AITER_MLA",
1244+
"VLLM_ROCM_USE_AITER_MHA",
1245+
"VLLM_ROCM_USE_SKINNY_GEMM",
1246+
"VLLM_ROCM_FP8_PADDING",
1247+
"VLLM_ROCM_MOE_PADDING",
1248+
"VLLM_ROCM_CUSTOM_PAGED_ATTN",
1249+
"VLLM_ROCM_QUICK_REDUCE_QUANTIZATION",
1250+
"VLLM_ROCM_QUICK_REDUCE_CAST_BF16_TO_FP16",
1251+
"VLLM_ROCM_QUICK_REDUCE_MAX_SIZE_BYTES_MB",
12321252
]
12331253
for key in environment_variables_to_hash:
1234-
if key in environment_variables:
1235-
factorize(key)
1254+
# if this goes out of sync with environment_variables,
1255+
# it's not a user error, it's a bug
1256+
assert key in environment_variables, \
1257+
"Please update environment_variables_to_hash in envs.py"
1258+
1259+
factors = [
1260+
environment_variables[key]() for key in environment_variables_to_hash
1261+
]
12361262

12371263
hash_str = hashlib.md5(str(factors).encode(),
12381264
usedforsecurity=False).hexdigest()

0 commit comments

Comments
 (0)