Skip to content

Commit

Permalink
Adjust some rocm library locations
Browse files Browse the repository at this point in the history
  • Loading branch information
jayfurmanek committed Aug 7, 2023
1 parent e337a60 commit e5a5bf8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions third_party/gpus/rocm_configure.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -339,14 +339,14 @@ def _find_libs(repository_ctx, rocm_config, hipfft_or_rocfft, miopen_path, rccl_
libs_paths = [
(name, _rocm_lib_paths(repository_ctx, name, path))
for name, path in [
("amdhip64", rocm_config.rocm_toolkit_path + "/hip"),
("amdhip64", rocm_config.rocm_toolkit_path),
("rocblas", rocm_config.rocm_toolkit_path),
(hipfft_or_rocfft, rocm_config.rocm_toolkit_path),
("hiprand", rocm_config.rocm_toolkit_path),
("MIOpen", miopen_path),
("rccl", rccl_path),
("hipsparse", rocm_config.rocm_toolkit_path),
("roctracer64", rocm_config.rocm_toolkit_path + "/roctracer"),
("roctracer64", rocm_config.rocm_toolkit_path),
("rocsolver", rocm_config.rocm_toolkit_path),
]
]
Expand Down

0 comments on commit e5a5bf8

Please sign in to comment.