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",
"rocm/include/rocrand", "rocm/include/rocrand",
"rocm/include/roctracer",
], ],
visibility = ["//visibility:public"], visibility = ["//visibility:public"],
) )
@ -109,6 +110,7 @@ cc_library(
":hiprand", ":hiprand",
":miopen", ":miopen",
":hipsparse", ":hipsparse",
":roctracer",
":rocsolver", ":rocsolver",
], ],
) )
@ -144,6 +146,11 @@ cc_library(
data = ["rocm/lib/%{hipsparse_lib}"], data = ["rocm/lib/%{hipsparse_lib}"],
) )
cc_library(
name = "roctracer",
data = ["rocm/lib/%{roctracer_lib}"],
)
cc_library( cc_library(
name = "rocsolver", name = "rocsolver",
srcs = ["rocm/lib/%{rocsolver_lib}"], 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"), ("MIOpen", rocm_config.rocm_toolkit_path + "/miopen"),
("rccl", rocm_config.rocm_toolkit_path + "/rccl"), ("rccl", rocm_config.rocm_toolkit_path + "/rccl"),
("hipsparse", rocm_config.rocm_toolkit_path + "/hipsparse"), ("hipsparse", rocm_config.rocm_toolkit_path + "/hipsparse"),
("roctracer64", rocm_config.rocm_toolkit_path + "/roctracer"),
("rocsolver", rocm_config.rocm_toolkit_path + "/rocsolver"), ("rocsolver", rocm_config.rocm_toolkit_path + "/rocsolver"),
] ]
] ]
@ -458,6 +459,7 @@ def _create_dummy_repository(repository_ctx):
"%{rocfft_lib}": _lib_name("rocfft"), "%{rocfft_lib}": _lib_name("rocfft"),
"%{hiprand_lib}": _lib_name("hiprand"), "%{hiprand_lib}": _lib_name("hiprand"),
"%{hipsparse_lib}": _lib_name("hipsparse"), "%{hipsparse_lib}": _lib_name("hipsparse"),
"%{roctracer_lib}": _lib_name("roctracer64"),
"%{rocsolver_lib}": _lib_name("rocsolver"), "%{rocsolver_lib}": _lib_name("rocsolver"),
"%{copy_rules}": "", "%{copy_rules}": "",
"%{rocm_headers}": "", "%{rocm_headers}": "",
@ -635,6 +637,7 @@ def _create_local_rocm_repository(repository_ctx):
"%{miopen_lib}": rocm_libs["MIOpen"].file_name, "%{miopen_lib}": rocm_libs["MIOpen"].file_name,
"%{rccl_lib}": rocm_libs["rccl"].file_name, "%{rccl_lib}": rocm_libs["rccl"].file_name,
"%{hipsparse_lib}": rocm_libs["hipsparse"].file_name, "%{hipsparse_lib}": rocm_libs["hipsparse"].file_name,
"%{roctracer_lib}": rocm_libs["roctracer64"].file_name,
"%{rocsolver_lib}": rocm_libs["rocsolver"].file_name, "%{rocsolver_lib}": rocm_libs["rocsolver"].file_name,
"%{copy_rules}": "\n".join(copy_rules), "%{copy_rules}": "\n".join(copy_rules),
"%{rocm_headers}": ('":rocm-include",\n' + "%{rocm_headers}": ('":rocm-include",\n' +