aboutsummaryrefslogtreecommitdiffhomepage
path: root/tools
diff options
context:
space:
mode:
authorGravatar Wen-Heng (Jack) Chung <whchung@gmail.com>2018-06-22 23:09:43 -0500
committerGravatar Deven Desai <deven.desai.amd@gmail.com>2018-09-05 23:50:51 +0000
commit69d3b8faf41791834301a74a05e288964940427d (patch)
tree2d27d1201f7b7216ca1d5fe24466c945baa44bbc /tools
parent47b1af2a3a724a5d783ae06ca0e0e78b30e0799b (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 'tools')
-rw-r--r--tools/bazel.rc3
1 files changed, 3 insertions, 0 deletions
diff --git a/tools/bazel.rc b/tools/bazel.rc
index 601e07ffdd..afc5cf56ab 100644
--- a/tools/bazel.rc
+++ b/tools/bazel.rc
@@ -42,6 +42,9 @@ build:download_clang_use_lld --linkopt='-fuse-ld=lld'
build:cuda --crosstool_top=@local_config_cuda//crosstool:toolchain
build:cuda --define=using_cuda=true --define=using_cuda_nvcc=true
+build:rocm --crosstool_top=@local_config_rocm//crosstool:toolchain
+build:rocm --define=using_rocm=true --define=using_rocm_hipcc=true
+
build:cuda_clang --crosstool_top=@local_config_cuda//crosstool:toolchain
build:cuda_clang --define=using_cuda=true --define=using_cuda_clang=true --define=using_clang=true