Tiny add warning when cannot recognize bool env var (#5348)

This commit is contained in:
fzyzcjy
2025-04-21 14:11:29 +08:00
committed by GitHub
parent 70645f4d7d
commit 685d8980c3

View File

@@ -78,10 +78,24 @@ time_infos = {}
HIP_FP8_E4M3_FNUZ_MAX = 224.0
_warned_bool_env_var_keys = set()
def get_bool_env_var(name: str, default: str = "false") -> bool:
value = os.getenv(name, default)
return value.lower() in ("true", "1")
value = value.lower()
truthy_values = ("true", "1")
falsy_values = ("false", "0")
if (value not in truthy_values) and (value not in falsy_values):
if value not in _warned_bool_env_var_keys:
logger.warning(
f"get_bool_env_var({name}) see non-understandable value={value} and treat as false"
)
_warned_bool_env_var_keys.add(value)
return value in truthy_values
# https://pytorch.org/docs/stable/notes/hip.html#checking-for-hip