diff --git a/docs/usage/environment_variables.md b/docs/usage/environment_variables.md index 313449a40..6cae70535 100644 --- a/docs/usage/environment_variables.md +++ b/docs/usage/environment_variables.md @@ -72,5 +72,7 @@ environment_variables: dict[str, Callable[[], Any]] = { "FD_USE_DEEP_GEMM": lambda: bool(int(os.getenv("FD_USE_DEEP_GEMM", "0"))), + # Whether to use Machete for wint4 dense GEMM. + "FD_USE_MACHETE": lambda: os.getenv("FD_USE_MACHETE", "0"), } ``` diff --git a/docs/zh/usage/environment_variables.md b/docs/zh/usage/environment_variables.md index 74b135eae..74d0fc9dd 100644 --- a/docs/zh/usage/environment_variables.md +++ b/docs/zh/usage/environment_variables.md @@ -71,5 +71,8 @@ environment_variables: dict[str, Callable[[], Any]] = { # 是否使用DeepGemm后端的FP8 blockwise MoE. "FD_USE_DEEP_GEMM": lambda: bool(int(os.getenv("FD_USE_DEEP_GEMM", "0"))), + + # 是否使用 Machete 后端的 wint4 GEMM. + "FD_USE_MACHETE": lambda: os.getenv("FD_USE_MACHETE", "0"), } ```