# This file is expanded from a template by cuda_configure.bzl
# Update cuda_configure.bzl#verify_build_defines when adding new variables.

load(":cc_toolchain_config.bzl", "cc_toolchain_config")

licenses(["restricted"])

package(default_visibility = ["//visibility:public"])

toolchain(
    name = "toolchain-linux-x86_64",
    exec_compatible_with = [
        "@bazel_tools//platforms:linux",
        "@bazel_tools//platforms:x86_64",
    ],
    target_compatible_with = [
        "@bazel_tools//platforms:linux",
        "@bazel_tools//platforms:x86_64",
    ],
    toolchain = ":cc-compiler-local",
    toolchain_type = "@bazel_tools//tools/cpp:toolchain_type",
)

cc_toolchain_suite(
    name = "toolchain",
    toolchains = {
        "local|compiler": ":cc-compiler-local",
        "darwin|compiler": ":cc-compiler-darwin",
        "x64_windows|msvc-cl": ":cc-compiler-windows",
        "x64_windows": ":cc-compiler-windows",
        "arm": ":cc-compiler-local",
        "aarch64": ":cc-compiler-local",
        "k8": ":cc-compiler-local",
        "piii": ":cc-compiler-local",
        "ppc": ":cc-compiler-local",
        "darwin": ":cc-compiler-darwin",
    },
)

cc_toolchain(
    name = "cc-compiler-local",
    all_files = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    compiler_files = ":empty",
    dwp_files = ":empty",
    linker_files = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    objcopy_files = ":empty",
    strip_files = ":empty",
    # To support linker flags that need to go to the start of command line
    # we need the toolchain to support parameter files. Parameter files are
    # last on the command line and contain all shared libraries to link, so all
    # regular options will be left of them.
    supports_param_files = 1,
    toolchain_config = ":cc-compiler-local-config",
    toolchain_identifier = "local_linux",
)

cc_toolchain_config(
    name = "cc-compiler-local-config",
    builtin_include_directories = [
        "/usr/include/c++/5",
        "/usr/include/x86_64-linux-gnu/c++/5",
        "/usr/include/c++/5/backward",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include",
        "/usr/local/include",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include-fixed",
        "/usr/include/x86_64-linux-gnu",
        "/usr/include",
        "/opt/rocm/include",
        "/opt/rocm/hsa/include",
        "/opt/rocm/include/hip",
        "/opt/rocm/include/hip/hcc_detail",
        "/opt/rocm/hip/include",
        "/opt/rocm/llvm/lib/clang/8.0/include",
        "/opt/rocm/llvm/lib/clang/9.0.0/include",
        "/opt/rocm/rocrand/include",
        "/opt/rocm/hiprand/include",
        "/opt/rocm/rocfft/include",
        "/opt/rocm/rocblas/include",
        "/opt/rocm/miopen/include",
        "/opt/rocm/hcc/include",
        "/opt/rocm/hcc/compiler/lib/clang/7.0.0/include/",
        "/opt/rocm/hcc/lib/clang/7.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/8.0.0/include/",
        "/opt/rocm/hcc/lib/clang/8.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/9.0.0/include/",
        "/opt/rocm/hcc/lib/clang/9.0.0/include",
    ],
    cpu = "local",
    extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"],
    host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    host_compiler_prefix = "/usr/bin",
    host_compiler_warnings = [
        "-Wunused-but-set-parameter",
        "-Wno-free-nonheap-object",
    ],
    host_unfiltered_compile_flags = [
        "-DTENSORFLOW_USE_ROCM=1",
        "-D__HIP_PLATFORM_HCC__",
        "-DEIGEN_USE_HIP",
    ],
    linker_bin_path = "/opt/rocm/hcc/compiler/bin",
)

cc_toolchain(
    name = "cc-compiler-darwin",
    all_files = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    compiler_files = ":empty",
    dwp_files = ":empty",
    linker_files = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    objcopy_files = ":empty",
    strip_files = ":empty",
    supports_param_files = 0,
    toolchain_config = ":cc-compiler-local-darwin",
    toolchain_identifier = "local_darwin",
)

cc_toolchain_config(
    name = "cc-compiler-local-darwin",
    builtin_include_directories = [
        "/usr/include/c++/5",
        "/usr/include/x86_64-linux-gnu/c++/5",
        "/usr/include/c++/5/backward",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include",
        "/usr/local/include",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include-fixed",
        "/usr/include/x86_64-linux-gnu",
        "/usr/include",
        "/opt/rocm/include",
        "/opt/rocm/hsa/include",
        "/opt/rocm/include/hip",
        "/opt/rocm/include/hip/hcc_detail",
        "/opt/rocm/hip/include",
        "/opt/rocm/llvm/lib/clang/8.0/include",
        "/opt/rocm/llvm/lib/clang/9.0.0/include",
        "/opt/rocm/rocrand/include",
        "/opt/rocm/hiprand/include",
        "/opt/rocm/rocfft/include",
        "/opt/rocm/rocblas/include",
        "/opt/rocm/miopen/include",
        "/opt/rocm/hcc/include",
        "/opt/rocm/hcc/compiler/lib/clang/7.0.0/include/",
        "/opt/rocm/hcc/lib/clang/7.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/8.0.0/include/",
        "/opt/rocm/hcc/lib/clang/8.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/9.0.0/include/",
        "/opt/rocm/hcc/lib/clang/9.0.0/include",
    ],
    cpu = "darwin",
    extra_no_canonical_prefixes_flags = ["-fno-canonical-system-headers"],
    host_compiler_path = "clang/bin/crosstool_wrapper_driver_is_not_gcc",
    host_compiler_prefix = "/usr/bin",
    host_compiler_warnings = [
        "-Wunused-but-set-parameter",
        "-Wno-free-nonheap-object",
    ],
    host_unfiltered_compile_flags = [
        "-DTENSORFLOW_USE_ROCM=1",
        "-D__HIP_PLATFORM_HCC__",
        "-DEIGEN_USE_HIP",
    ],
    linker_bin_path = "/opt/rocm/hcc/compiler/bin",
)

cc_toolchain(
    name = "cc-compiler-windows",
    all_files = ":empty",
    compiler_files = ":empty",
    dwp_files = ":empty",
    linker_files = ":empty",
    objcopy_files = ":empty",
    strip_files = ":empty",
    supports_param_files = 1,
    toolchain_config = ":cc-compiler-windows-config",
    toolchain_identifier = "local_windows",
)

cc_toolchain_config(
    name = "cc-compiler-windows-config",
    builtin_include_directories = [
        "/usr/include/c++/5",
        "/usr/include/x86_64-linux-gnu/c++/5",
        "/usr/include/c++/5/backward",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include",
        "/usr/local/include",
        "/usr/lib/gcc/x86_64-linux-gnu/5/include-fixed",
        "/usr/include/x86_64-linux-gnu",
        "/usr/include",
        "/opt/rocm/include",
        "/opt/rocm/hsa/include",
        "/opt/rocm/include/hip",
        "/opt/rocm/include/hip/hcc_detail",
        "/opt/rocm/hip/include",
        "/opt/rocm/llvm/lib/clang/8.0/include",
        "/opt/rocm/llvm/lib/clang/9.0.0/include",
        "/opt/rocm/rocrand/include",
        "/opt/rocm/hiprand/include",
        "/opt/rocm/rocfft/include",
        "/opt/rocm/rocblas/include",
        "/opt/rocm/miopen/include",
        "/opt/rocm/hcc/include",
        "/opt/rocm/hcc/compiler/lib/clang/7.0.0/include/",
        "/opt/rocm/hcc/lib/clang/7.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/8.0.0/include/",
        "/opt/rocm/hcc/lib/clang/8.0.0/include",
        "/opt/rocm/hcc/compiler/lib/clang/9.0.0/include/",
        "/opt/rocm/hcc/lib/clang/9.0.0/include",
    ],
    cpu = "x64_windows",
    msvc_cl_path = "msvc_not_used",
    msvc_env_include = "msvc_not_used",
    msvc_env_lib = "msvc_not_used",
    msvc_env_path = "msvc_not_used",
    msvc_env_tmp = "msvc_not_used",
    msvc_lib_path = "msvc_not_used",
    msvc_link_path = "msvc_not_used",
    msvc_ml_path = "msvc_not_used",
)

filegroup(
    name = "empty",
    srcs = [],
)

filegroup(
    name = "crosstool_wrapper_driver_is_not_gcc",
    srcs = ["clang/bin/crosstool_wrapper_driver_is_not_gcc"],
)

filegroup(
    name = "windows_msvc_wrapper_files",
    srcs = glob(["windows/msvc_*"]),
)
