OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2025-12-06 00:19:58 +01:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Merge pull request #48443 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47491

Browse Source 
[r2.5 port][ROCm] Port PR 47491 to r2.5
This commit is contained in:
Mihai Maruseac 2021-04-23 08:24:22 -07:00 committed by GitHub
commit eb3b2914c6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 35 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
third_party/gpus/rocm_configure.bzl vendored
View File

@ -592,6 +592,37 @@ def _create_local_rocm_repository(repository_ctx):
),
]
# explicitly copy (into the local_config_rocm repo) the $ROCM_PATH/hiprand/include and
# $ROCM_PATH/rocrand/include dirs, only once the softlink to them in $ROCM_PATH/include
# dir has been removed. This removal will happen in a near-future ROCm release.
hiprand_include = ""
hiprand_include_softlink = rocm_config.rocm_toolkit_path + "/include/hiprand"
softlink_exists = files_exist(repository_ctx, [hiprand_include_softlink], bash_bin)
if not softlink_exists[0]:
hiprand_include = '":hiprand-include",\n'
copy_rules.append(
make_copy_dir_rule(
repository_ctx,
name = "hiprand-include",
src_dir = rocm_toolkit_path + "/hiprand/include",
out_dir = "rocm/include/hiprand",
)
)
rocrand_include = ""
rocrand_include_softlink = rocm_config.rocm_toolkit_path + "/include/rocrand"
softlink_exists = files_exist(repository_ctx, [rocrand_include_softlink], bash_bin)
if not softlink_exists[0]:
rocrand_include = '":rocrand-include",\n'
copy_rules.append(
make_copy_dir_rule(
repository_ctx,
name = "rocrand-include",
src_dir = rocm_toolkit_path + "/rocrand/include",
out_dir = "rocm/include/rocrand",
)
)
rocm_libs = _find_libs(repository_ctx, rocm_config, hipfft_or_rocfft, bash_bin)
rocm_lib_srcs = []
rocm_lib_outs = []
@ -619,6 +650,7 @@ def _create_local_rocm_repository(repository_ctx):
],
))
# Set up BUILD file for rocm/
repository_ctx.template(
"rocm/build_defs.bzl",
@ -652,7 +684,9 @@ def _create_local_rocm_repository(repository_ctx):
'":rocblas-include",\n' +
'":miopen-include",\n' +
'":rccl-include",\n' +
'":hipsparse-include",' +
hiprand_include +
rocrand_include +
'":hipsparse-include",\n' +
'":rocsolver-include"'),
},
)