github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/gpus/rocm/BUILD.tpl (about)

     1  licenses(["restricted"])  # MPL2, portions GPL v3, LGPL v3, BSD-like
     2  
     3  package(default_visibility = ["//visibility:public"])
     4  
     5  config_setting(
     6      name = "using_hipcc",
     7      values = {
     8          "define": "using_rocm_hipcc=true",
     9      },
    10  )
    11  
    12  cc_library(
    13      name = "rocm_headers",
    14      hdrs = [
    15          "rocm/rocm_config.h",
    16          %{rocm_headers}
    17      ],
    18      includes = [
    19          ".",
    20          "rocm/include",
    21          "rocm/include/rocrand",
    22      ],
    23      visibility = ["//visibility:public"],
    24  )
    25  
    26  cc_library(
    27      name = "hip",
    28      srcs = ["rocm/lib/%{hip_lib}"],
    29      data = ["rocm/lib/%{hip_lib}"],
    30      includes = [
    31          ".",
    32          "rocm/include",
    33      ],
    34      linkstatic = 1,
    35      visibility = ["//visibility:public"],
    36  )
    37  
    38  cc_library(
    39      name = "rocblas",
    40      srcs = ["rocm/lib/%{rocblas_lib}"],
    41      data = ["rocm/lib/%{rocblas_lib}"],
    42      includes = [
    43          ".",
    44          "rocm/include",
    45      ],
    46      linkstatic = 1,
    47      visibility = ["//visibility:public"],
    48  )
    49  
    50  cc_library(
    51      name = "rocfft",
    52      srcs = ["rocm/lib/%{rocfft_lib}"],
    53      data = ["rocm/lib/%{rocfft_lib}"],
    54      includes = [
    55          ".",
    56          "rocm/include",
    57      ],
    58      linkstatic = 1,
    59      visibility = ["//visibility:public"],
    60  )
    61  
    62  cc_library(
    63      name = "hiprand",
    64      srcs = ["rocm/lib/%{hiprand_lib}"],
    65      data = ["rocm/lib/%{hiprand_lib}"],
    66      includes = [
    67          ".",
    68          "rocm/include",
    69          "rocm/include/rocrand",
    70      ],
    71      linkstatic = 1,
    72      visibility = ["//visibility:public"],
    73  )
    74  
    75  cc_library(
    76      name = "miopen",
    77      srcs = ["rocm/lib/%{miopen_lib}"],
    78      data = ["rocm/lib/%{miopen_lib}"],
    79      includes = [
    80          ".",
    81          "rocm/include",
    82      ],
    83      linkstatic = 1,
    84      visibility = ["//visibility:public"],
    85  )
    86  
    87  cc_library(
    88      name = "rccl",
    89      srcs = ["rocm/lib/%{rccl_lib}"],
    90      data = ["rocm/lib/%{rccl_lib}"],
    91      includes = [
    92          ".",
    93          "rocm/include",
    94      ],
    95      linkstatic = 1,
    96      visibility = ["//visibility:public"],
    97  )
    98  
    99  cc_library(
   100      name = "rocm",
   101      visibility = ["//visibility:public"],
   102      deps = [
   103          ":rocm_headers",
   104          ":hip",
   105          ":rocblas",
   106          ":rocfft",
   107          ":hiprand",
   108          ":miopen",
   109      ],
   110  )
   111  
   112  %{copy_rules}