github.com/johnnyeven/libtools@v0.0.0-20191126065708-61829c1adf46/third_party/toolchains/preconfig/generate/BUILD (about)

     1  licenses(["restricted"])
     2  
     3  load(":generate.bzl", "tensorflow_rbe_config")
     4  
     5  tensorflow_rbe_config(
     6      name = "ubuntu16.04-py3-clang",
     7      compiler = "clang",
     8      os = "ubuntu16.04",
     9      python_version = "3",
    10  )
    11  
    12  tensorflow_rbe_config(
    13      name = "centos6-py-gcc7",
    14      build_bazel_src = True,
    15      compiler = "gcc",
    16      compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
    17      os = "centos6",
    18      python_version = "2",
    19  )
    20  
    21  tensorflow_rbe_config(
    22      name = "centos6-py3-gcc7",
    23      build_bazel_src = True,
    24      compiler = "gcc",
    25      compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
    26      os = "centos6",
    27      python_version = "3",
    28  )
    29  
    30  tensorflow_rbe_config(
    31      name = "centos6-py3-gcc7-cuda10.0-cudnn7-tensorrt5",
    32      build_bazel_src = True,
    33      compiler = "gcc",
    34      compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
    35      cuda_version = "10.0",
    36      cudnn_install_path = "/usr/local/cuda-10.0",
    37      cudnn_version = "7",
    38      os = "centos6",
    39      python_version = "3",
    40      tensorrt_install_path = "/usr",
    41      tensorrt_version = "5",
    42  )
    43  
    44  tensorflow_rbe_config(
    45      name = "centos6-py3-gcc7-cuda10.1-cudnn7-tensorrt5",
    46      build_bazel_src = True,
    47      compiler = "gcc",
    48      compiler_prefix = "/opt/rh/devtoolset-7/root/usr/bin",
    49      cuda_version = "10.1",
    50      cudnn_install_path = "/usr/local/cuda-10.1",
    51      cudnn_version = "7",
    52      os = "centos6",
    53      python_version = "3",
    54      tensorrt_install_path = "/usr",
    55      tensorrt_version = "5",
    56  )
    57  
    58  tensorflow_rbe_config(
    59      name = "ubuntu14.04-py3-gcc7-cuda10.0-cudnn7-tensorrt5",
    60      compiler = "gcc-7",
    61      cuda_version = "10.0",
    62      cudnn_version = "7",
    63      os = "ubuntu14.04",
    64      python_version = "3",
    65      tensorrt_version = "5",
    66  )
    67  
    68  tensorflow_rbe_config(
    69      name = "ubuntu14.04-py3-gcc-cuda10.0-cudnn7-tensorrt5",
    70      compiler = "gcc",
    71      cuda_version = "10.0",
    72      cudnn_version = "7",
    73      os = "ubuntu14.04",
    74      python_version = "3",
    75      tensorrt_version = "5",
    76  )
    77  
    78  tensorflow_rbe_config(
    79      name = "ubuntu14.04-py3-clang-cuda10.0-cudnn7-tensorrt5",
    80      compiler = "clang",
    81      cuda_version = "10.0",
    82      cudnn_version = "7",
    83      os = "ubuntu14.04",
    84      python_version = "3",
    85      tensorrt_version = "5",
    86  )
    87  
    88  tensorflow_rbe_config(
    89      name = "ubuntu16.04-py-gcc7_manylinux2010",
    90      compiler = "/dt7/usr/bin/gcc",
    91      compiler_prefix = "/usr/bin",
    92      os = "ubuntu16.04-manylinux2010",
    93      python_version = "2",
    94  )
    95  
    96  tensorflow_rbe_config(
    97      name = "ubuntu16.04-py3-gcc7_manylinux2010",
    98      compiler = "/dt7/usr/bin/gcc",
    99      compiler_prefix = "/usr/bin",
   100      os = "ubuntu16.04-manylinux2010",
   101      python_version = "3.6",
   102  )
   103  
   104  tensorflow_rbe_config(
   105      name = "ubuntu16.04-py3-gcc7_manylinux2010-cuda10.0-cudnn7-tensorrt5.1",
   106      compiler = "/dt7/usr/bin/gcc",
   107      compiler_prefix = "/usr/bin",
   108      cuda_version = "10.0",
   109      cudnn_version = "7",
   110      os = "ubuntu16.04-manylinux2010",
   111      python_version = "3.6",
   112      tensorrt_install_path = "/usr/local/tensorrt",
   113      tensorrt_version = "5.1",
   114  )
   115  
   116  tensorflow_rbe_config(
   117      name = "ubuntu16.04-py3_opt-gcc5-rocm",
   118      compiler = "gcc",
   119      os = "ubuntu16.04",
   120      python_version = "3",
   121      rocm_version = "2.5",  # Any version will do.
   122  )