快捷方式

特性开关 (C++)

enum class FeatureGateName

使用 X 宏模式生成的特性开关枚举类定义。

enumerator X
enumerator ENUMERATE_ALL_FEATURE_FLAGS
const 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++)。

特性开关用于根据环境设置启用/禁用实验性特性。

枚举使用 X 宏模式定义。要添加特性开关,只需将 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:枚举名称必须与 UI 中的 JK 旋钮名称完全匹配。

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

注意

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发者的深入教程

查看教程

资源

查找开发资源并获得解答

查看资源