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

Adding roctracer to local_config_rocm repo

Browse Source
This commit is contained in:
Deven Desai 2021-03-02 03:33:18 +00:00
parent 63019e2b06
commit 505751314d
2 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
third_party/gpus/rocm/BUILD.tpl vendored
View File

@ -21,6 +21,7 @@ cc_library(
".",
"rocm/include",
"rocm/include/rocrand",
"rocm/include/roctracer",
],
visibility = ["//visibility:public"],
)
@ -109,6 +110,7 @@ cc_library(
":hiprand",
":miopen",
":hipsparse",
":roctracer",
":rocsolver",
],
)
@ -144,6 +146,11 @@ cc_library(
data = ["rocm/lib/%{hipsparse_lib}"],
)
cc_library(
name = "roctracer",
data = ["rocm/lib/%{roctracer_lib}"],
)
cc_library(
name = "rocsolver",
srcs = ["rocm/lib/%{rocsolver_lib}"],

3
third_party/gpus/rocm_configure.bzl vendored
View File

@ -332,6 +332,7 @@ def _find_libs(repository_ctx, rocm_config, bash_bin):
("MIOpen", rocm_config.rocm_toolkit_path + "/miopen"),
("rccl", rocm_config.rocm_toolkit_path + "/rccl"),
("hipsparse", rocm_config.rocm_toolkit_path + "/hipsparse"),
("roctracer64", rocm_config.rocm_toolkit_path + "/roctracer"),
("rocsolver", rocm_config.rocm_toolkit_path + "/rocsolver"),
]
]
@ -458,6 +459,7 @@ def _create_dummy_repository(repository_ctx):
"%{rocfft_lib}": _lib_name("rocfft"),
"%{hiprand_lib}": _lib_name("hiprand"),
"%{hipsparse_lib}": _lib_name("hipsparse"),
"%{roctracer_lib}": _lib_name("roctracer64"),
"%{rocsolver_lib}": _lib_name("rocsolver"),
"%{copy_rules}": "",
"%{rocm_headers}": "",
@ -635,6 +637,7 @@ def _create_local_rocm_repository(repository_ctx):
"%{miopen_lib}": rocm_libs["MIOpen"].file_name,
"%{rccl_lib}": rocm_libs["rccl"].file_name,
"%{hipsparse_lib}": rocm_libs["hipsparse"].file_name,
"%{roctracer_lib}": rocm_libs["roctracer64"].file_name,
"%{rocsolver_lib}": rocm_libs["rocsolver"].file_name,
"%{copy_rules}": "\n".join(copy_rules),
"%{rocm_headers}": ('":rocm-include",\n' +