diff options
| author |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 10:22:55 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-27 10:22:55 -0700 |
| commit | 62e60166de65d6604b897f2575a5accc86160496 (patch) | |
| tree | 4f0c3b8d7915de7f328bfeaaa1a0312f7df481a2 /tensorflow/workspace.bzl | |
| parent | cd1bdeafecf39bc55409b75cf27cecf273237ca2 (diff) | |
| parent | 69d3b8faf41791834301a74a05e288964940427d (diff) | |
Merge pull request #20277 from ROCmSoftwarePlatform:upstream-staging
PiperOrigin-RevId: 214793113
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 9128a1aafe..4bf2ff3fb5 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() |