kse-01/tensorflow/third_party/gpus/rocm/BUILD.tpl
github-classroom[bot] 1122cdd8b0
Initial commit
2023-10-09 11:37:31 +00:00

147 lines
2.9 KiB (Stored with Git LFS)
Smarty

load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
licenses(["restricted"]) # MPL2, portions GPL v3, LGPL v3, BSD-like
package(default_visibility = ["//visibility:public"])
config_setting(
name = "using_hipcc",
values = {
"define": "using_rocm_hipcc=true",
},
)
cc_library(
name = "rocm_headers",
hdrs = [
"rocm/rocm_config.h",
%{rocm_headers}
],
includes = [
".",
"rocm/include",
"rocm/include/rocrand",
],
visibility = ["//visibility:public"],
)
cc_library(
name = "hip",
srcs = ["rocm/lib/%{hip_lib}"],
data = ["rocm/lib/%{hip_lib}"],
includes = [
".",
"rocm/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "rocblas",
srcs = ["rocm/lib/%{rocblas_lib}"],
data = ["rocm/lib/%{rocblas_lib}"],
includes = [
".",
"rocm/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "rocfft",
srcs = ["rocm/lib/%{rocfft_lib}"],
data = ["rocm/lib/%{rocfft_lib}"],
includes = [
".",
"rocm/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "hiprand",
srcs = ["rocm/lib/%{hiprand_lib}"],
data = ["rocm/lib/%{hiprand_lib}"],
includes = [
".",
"rocm/include",
"rocm/include/rocrand",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "miopen",
srcs = ["rocm/lib/%{miopen_lib}"],
data = ["rocm/lib/%{miopen_lib}"],
includes = [
".",
"rocm/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "rccl",
srcs = ["rocm/lib/%{rccl_lib}"],
data = ["rocm/lib/%{rccl_lib}"],
includes = [
".",
"rocm/include",
],
linkstatic = 1,
visibility = ["//visibility:public"],
)
cc_library(
name = "rocm",
visibility = ["//visibility:public"],
deps = [
":rocm_headers",
":hip",
":rocblas",
":rocfft",
":hiprand",
":miopen",
],
)
bzl_library(
name = "build_defs_bzl",
srcs = ["build_defs.bzl"],
)
cc_library(
name = "rocprim",
srcs = [
"rocm/include/hipcub/hipcub_version.hpp",
"rocm/include/rocprim/rocprim_version.hpp",
],
hdrs = glob([
"rocm/include/hipcub/**",
"rocm/include/rocprim/**",
]),
includes = [
".",
"rocm/include/hipcub",
"rocm/include/rocprim",
],
visibility = ["//visibility:public"],
deps = [
"@local_config_rocm//rocm:rocm_headers",
],
)
cc_import(
name = "hipsparse",
hdrs = glob(["rocm/include/hipsparse/**",]),
shared_library = "rocm/lib/%{hipsparse_lib}",
visibility = ["//visibility:public"],
)
%{copy_rules}