def use_tanh_for_sigmoid() -> bool:
    """
    Whether the we want to use tanh to approximate sigmoid for the device code generation.
    This controls both the code generation for AITemplate codegen and CUTLASS.
    Default value to get from environment variable is "0".
    """
    return os.getenv("AIT_USE_TANH_FOR_SIGMOID", "0") == "1"
Whether the we want to use tanh to approximate sigmoid for the device code generation.
This controls both the code generation for AITemplate codegen and CUTLASS.
Default value to get from environment variable is "0".
python/aitemplate/utils/environ.py