diff --git a/third_party/gpus/rocm_configure.bzl b/third_party/gpus/rocm_configure.bzl index c7474273fd8..39fb9a6d8cb 100644 --- a/third_party/gpus/rocm_configure.bzl +++ b/third_party/gpus/rocm_configure.bzl @@ -202,6 +202,7 @@ def _rocm_include_path(repository_ctx, rocm_config): inc_dirs.append(rocm_config.rocm_toolkit_path + "/llvm/lib/clang/8.0/include") inc_dirs.append(rocm_config.rocm_toolkit_path + "/llvm/lib/clang/9.0.0/include") inc_dirs.append(rocm_config.rocm_toolkit_path + "/llvm/lib/clang/10.0.0/include") + inc_dirs.append(rocm_config.rocm_toolkit_path + "/llvm/lib/clang/11.0.0/include") # Add rocrand and hiprand headers inc_dirs.append(rocm_config.rocm_toolkit_path + "/rocrand/include") diff --git a/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/BUILD b/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/BUILD index 0dcee093970..6b71c20f735 100755 --- a/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/BUILD +++ b/third_party/toolchains/preconfig/ubuntu16.04/gcc5-rocm/BUILD @@ -85,6 +85,7 @@ cc_toolchain_config( "/opt/rocm/llvm/lib/clang/8.0/include", "/opt/rocm/llvm/lib/clang/9.0.0/include", "/opt/rocm/llvm/lib/clang/10.0.0/include", + "/opt/rocm/llvm/lib/clang/11.0.0/include", "/opt/rocm/rocrand/include", "/opt/rocm/hiprand/include", "/opt/rocm/rocfft/include",