Skip to content

Commit 596cbc2

Browse files
committed
Fix bazel targets
1 parent aa5c077 commit 596cbc2

File tree

1 file changed

+9
-16
lines changed

1 file changed

+9
-16
lines changed

third_party/gpus/rocm/BUILD.tpl

Lines changed: 9 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -166,8 +166,8 @@ cc_library(
166166

167167
cc_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.
186185
cc_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

233231
cc_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

248243
cc_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

306300
cc_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

390383
cc_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

Comments
 (0)