Tiny add warning when cannot recognize bool env var (#5348)
This commit is contained in:
@@ -78,10 +78,24 @@ time_infos = {}
|
|||||||
|
|
||||||
HIP_FP8_E4M3_FNUZ_MAX = 224.0
|
HIP_FP8_E4M3_FNUZ_MAX = 224.0
|
||||||
|
|
||||||
|
_warned_bool_env_var_keys = set()
|
||||||
|
|
||||||
|
|
||||||
def get_bool_env_var(name: str, default: str = "false") -> bool:
|
def get_bool_env_var(name: str, default: str = "false") -> bool:
|
||||||
value = os.getenv(name, default)
|
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
|
# https://pytorch.org/docs/stable/notes/hip.html#checking-for-hip
|
||||||
|
|||||||
Reference in New Issue
Block a user