diff options
author | Wen-Heng (Jack) Chung <whchung@gmail.com> | 2018-06-22 23:09:43 -0500 |
---|---|---|
committer | Deven Desai <deven.desai.amd@gmail.com> | 2018-09-05 23:50:51 +0000 |
commit | 69d3b8faf41791834301a74a05e288964940427d (patch) | |
tree | 2d27d1201f7b7216ca1d5fe24466c945baa44bbc /tensorflow/workspace.bzl | |
parent | 47b1af2a3a724a5d783ae06ca0e0e78b30e0799b (diff) |
[ROCm] bazel build system and continuous integration logic
The commit contains following components to support TensorFlow on ROCm platform
- bazel build system
- continuous integration logic
Authors:
- Jack Chung: jack.chung@amd.com
- Jeffrey Poznanovic: Jeffrey.Poznanovic@amd.com
- Peng Sun: Peng.Sun@amd.com
Diffstat (limited to 'tensorflow/workspace.bzl')
-rwxr-xr-x | tensorflow/workspace.bzl | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl index 1e7c5d6790..87d1243563 100755 --- a/tensorflow/workspace.bzl +++ b/tensorflow/workspace.bzl @@ -1,6 +1,7 @@ # TensorFlow external dependencies that can be loaded in WORKSPACE files. load("//third_party/gpus:cuda_configure.bzl", "cuda_configure") +load("//third_party/gpus:rocm_configure.bzl", "rocm_configure") load("//third_party/tensorrt:tensorrt_configure.bzl", "tensorrt_configure") load("//third_party:nccl/nccl_configure.bzl", "nccl_configure") load("//third_party/mkl:build_defs.bzl", "mkl_repository") @@ -43,6 +44,7 @@ def tf_workspace(path_prefix = "", tf_repo_name = ""): sycl_configure(name = "local_config_sycl") syslibs_configure(name = "local_config_syslibs") python_configure(name = "local_config_python") + rocm_configure(name="local_config_rocm") initialize_third_party() |