快捷方式

特性開關 (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 的全面開發者文件

檢視文件

教程

獲取針對初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得解答

檢視資源