Skip to content

Commit

Permalink
Fix miopen issue
Browse files Browse the repository at this point in the history
  • Loading branch information
alekstheod committed Nov 19, 2024
1 parent 3659239 commit 87a11fc
Showing 1 changed file with 11 additions and 3 deletions.
14 changes: 11 additions & 3 deletions third_party/tsl/third_party/gpus/rocm/BUILD.tpl
Original file line number Diff line number Diff line change
Expand Up @@ -102,6 +102,8 @@ cc_library(
includes = [
"%{rocm_root}/include",
],
# workaround to bring tensile files to the same fs layout as expected in the lib
# rocblas assumes that tensile files are located in ../roblas/libraries directory
linkopts = ["-rpath local_config_rocm/rocm/rocm_dis/lib"],
strip_include_prefix = "%{rocm_root}",
visibility = ["//visibility:public"],
Expand Down Expand Up @@ -150,14 +152,18 @@ cc_library(

cc_library(
name = "miopen",
srcs = glob(["%{rocm_root}/lib/libMIOpen*.so*"]),
hdrs = glob(["%{rocm_root}/include/rccl/**"]),
data = glob(["%{rocm_root}/lib/libMIOpen*.so*"]),
data = glob([
"%{rocm_root}/lib/libMIOpen*.so*",
"%{rocm_root}/share/miopen/**",
]),
include_prefix = "rocm",
includes = [
"%{rocm_root}/include",
],
linkstatic = 1,
# workaround to bring miopen db files to the same fs layout as expected in the lib
# rocblas assumes that miopen db files are located in ../share/miopen/db directory
linkopts = ["-rpath local_config_rocm/rocm/rocm_dis/lib"],
strip_include_prefix = "%{rocm_root}",
visibility = ["//visibility:public"],
deps = [":rocm_config"],
Expand Down Expand Up @@ -291,6 +297,8 @@ cc_library(
includes = [
"%{rocm_root}/include/",
],
# workaround to bring tensile files to the same fs layout as expected in the lib
# hibplatslt assumes that tensile files are located in ../hipblaslt/libraries directory
linkopts = ["-rpath local_config_rocm/rocm/rocm_dis/lib"],
strip_include_prefix = "%{rocm_root}",
visibility = ["//visibility:public"],
Expand Down

0 comments on commit 87a11fc

Please sign in to comment.