快捷方式

功能门 (C++)

enum class FeatureGateName

功能门的枚举类定义,使用 X-macro 模式生成。

enumerator X
enumerator ENUMERATE_ALL_FEATURE_FLAGS
std::string to_string(const FeatureGateName &value)

获取 FeatureGateName 枚举的字符串值。

bool check_feature_gate_key(const std::string &key)

查找给定键的功能门值。

bool is_feature_enabled(const FeatureGateName &feature)

对于给定的 FeatureGateName,检查相应的功能是否已启用。

ENUMERATE_ALL_FEATURE_FLAGS

FBGEMM_GPU 功能门枚举 (C++)。

功能门用于根据环境设置启用/禁用实验性功能。

ENUM 是使用 X-macro 模式定义的。要添加一个功能门,只需将 X(FEATURE_NAME) 附加到 ENUMERATE_ALL_FEATURE_FLAGS 宏。然后,要使用该功能门,请参见下面的示例。

示例

namespace config = fbgemm_gpu::config;

void foo() {
  if (config::is_feature_enabled(config::FeatureGateName::FEATURE_NAME))
  {
    // Do something if feature is enabled
    ...
  } else {
    // Do something different if feature is disabled
    ...
  }
}

虽然不是必需的,但最好在 Python 的 fbgemm_gpu.config.FeatureGateName 中镜像枚举值。

对于 fbcode:ENUM 名称必须与 UI 中的 JK 旋钮名称完全匹配。

对于 OSS:环境变量将被评估为 f”FBGEMM_{ENUM}”

注意

文档

访问全面的 PyTorch 开发者文档

查看文档

教程

为初学者和高级开发者提供深入的教程

查看教程

资源

查找开发资源并让您的问题得到解答

查看资源