config_setting(
    name = "with_tpu_support_define",
    define_values = {"with_tpu_support": "true"},
    visibility = ["//visibility:public"],
)

config_setting(
    name = "with_tpu_support_flag",
    flag_values = {":enable_tpu_support": "True"},
    visibility = ["//visibility:public"],
)

selects.config_setting_group(
    name = "with_tpu_support",
    match_any = [
        ":with_tpu_support_define",

# AWS SDK must be compiled in release mode. see: https://github.com/tensorflow/tensorflow/issues/37498
common:dbg --copt -DDEBUG_BUILD

# Config to build TF TPU
common:tpu --define=with_tpu_support=true
common:tpu --define=framework_shared_object=true
common:tpu --copt=-DLIBTPU_ON_GCE
common:tpu --define=enable_mlir_bridge=true

common:rocm --copt=-Wno-gnu-offsetof-extensions