@@ -166,8 +166,8 @@ cc_library(
166166
167167cc_library(
168168 name = "rocm_hip",
169+ srcs = glob(["%{ rocm_root} /lib/libamdhip*.so*"]),
169170 hdrs = glob(["%{ rocm_root} /include/hip/**"]),
170- data = glob(["%{ rocm_root} /lib/libamdhip*.so*"]),
171171 include_prefix = "rocm",
172172 includes = [
173173 "%{ rocm_root} /include",
@@ -176,7 +176,6 @@ cc_library(
176176 visibility = ["//visibility:public"],
177177 deps = [
178178 ":rocm_config",
179- ":rocm_rpath",
180179 ":rocprofiler_register",
181180 ":system_libs",
182181 ],
@@ -185,13 +184,13 @@ cc_library(
185184# Used by jax_rocm_plugin to minimally link to hip runtime.
186185cc_library(
187186 name = "hip_runtime",
188- hdrs = glob(["%{ rocm_root} /include/hip/**"]),
189187 srcs = glob([
190188 "%{ rocm_root} /lib/libamdhip*.so*",
191189 "%{ rocm_root} /lib/libamd_comgr.so*",
192190 "%{ rocm_root} /lib/librccl*.so*",
193191 "%{ rocm_root} /lib/libhipsparse*.so*",
194192 ]),
193+ hdrs = glob(["%{ rocm_root} /include/hip/**"]),
195194 include_prefix = "rocm",
196195 includes = [
197196 "%{ rocm_root} /include",
@@ -221,7 +220,6 @@ cc_library(
221220 includes = [
222221 "%{ rocm_root} /include",
223222 ],
224-
225223 strip_include_prefix = "%{ rocm_root} ",
226224 visibility = ["//visibility:public"],
227225 deps = [
@@ -232,27 +230,23 @@ cc_library(
232230
233231cc_library(
234232 name = "rocfft",
235- data = glob(["%{ rocm_root} /lib/librocfft*.so*"]),
233+ srcs = glob(["%{ rocm_root} /lib/librocfft*.so*"]),
236234 include_prefix = "rocm",
237235 includes = [
238236 "%{ rocm_root} /include",
239237 ],
240238 linkstatic = 1,
241239 visibility = ["//visibility:public"],
242- deps = [
243- ":rocm_config",
244- ":rocm_rpath",
245- ],
240+ deps = [":rocm_config"],
246241)
247242
248243cc_library(
249244 name = "hipfft",
250- data = glob(["%{ rocm_root} /lib/libhipfft*.so*"]),
245+ srcs = glob(["%{ rocm_root} /lib/libhipfft*.so*"]),
251246 include_prefix = "rocm",
252247 includes = [
253248 "%{ rocm_root} /include",
254249 ],
255- linkopts = ["-Wl,-rpath,external/local_config_rocm/rocm/%{ rocm_root} /lib"],
256250 linkstatic = 1,
257251 visibility = ["//visibility:public"],
258252 deps = [
@@ -305,8 +299,8 @@ cc_library(
305299
306300cc_library(
307301 name = "rccl",
302+ srcs = glob(["%{ rocm_root} /lib/librccl*.so*"]),
308303 hdrs = glob(["%{ rocm_root} /include/rccl/**"]),
309- data = glob(["%{ rocm_root} /lib/librccl*.so*"]),
310304 include_prefix = "rocm",
311305 includes = [
312306 "%{ rocm_root} /include",
@@ -317,7 +311,6 @@ cc_library(
317311 visibility = ["//visibility:public"],
318312 deps = [
319313 ":rocm_config",
320- ":rocm_rpath",
321314 ":system_libs",
322315 ],
323316)
@@ -363,8 +356,8 @@ cc_library(
363356 visibility = ["//visibility:public"],
364357 deps = [
365358 ":rocm_config",
366- ":rocsparse",
367359 ":rocm_rpath",
360+ ":rocsparse",
368361 ],
369362)
370363
@@ -389,17 +382,17 @@ cc_library(
389382
390383cc_library(
391384 name = "rocprofiler-sdk",
392- hdrs = glob(["%{ rocm_root} /include/rocprofiler-sdk/**"]),
393385 srcs = glob(["%{ rocm_root} /lib/librocprofiler-sdk*.so*"]),
386+ hdrs = glob(["%{ rocm_root} /include/rocprofiler-sdk/**"]),
394387 include_prefix = "rocm",
395388 includes = [
396389 "%{ rocm_root} /include/",
397390 ],
398391 strip_include_prefix = "%{ rocm_root} ",
399392 visibility = ["//visibility:public"],
400393 deps = [
401- ":rocm_config",
402394 ":hip_runtime",
395+ ":rocm_config",
403396 ],
404397)
405398
0 commit comments