Skip to content

Commit 831fef8

Browse files
committed
Avoid copying data files over if non hermetic build is used
1 parent 769e105 commit 831fef8

File tree

1 file changed

+52
-14
lines changed

1 file changed

+52
-14
lines changed

third_party/gpus/rocm/BUILD.tpl

Lines changed: 52 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -196,13 +196,19 @@ cc_library(
196196
],
197197
)
198198

199+
rocblas_libs = glob([
200+
"%{rocm_root}/lib/librocblas*.so*",
201+
"%{rocm_root}/lib/rocblas/**",
202+
])
203+
199204
cc_library(
200205
name = "rocblas",
201206
hdrs = glob(["%{rocm_root}/include/rocblas/**"]),
202-
data = glob([
203-
"%{rocm_root}/lib/librocblas*.so*",
204-
"%{rocm_root}/lib/rocblas/**",
205-
]),
207+
data = select({
208+
":build_hermetic": rocblas_libs,
209+
":multiple_rocm_paths": rocblas_libs,
210+
"//conditions:default": [],
211+
}),
206212
include_prefix = "rocm",
207213
includes = [
208214
"%{rocm_root}/include",
@@ -254,13 +260,19 @@ cc_library(
254260
deps = [":rocm_config"],
255261
)
256262

263+
miopen_libs = glob([
264+
"%{rocm_root}/share/miopen/**",
265+
])
266+
257267
cc_library(
258268
name = "miopen",
259269
srcs = glob(["%{rocm_root}/lib/libMIOpen*.so*"]),
260270
hdrs = glob(["%{rocm_root}/include/miopen/**"]),
261-
data = glob([
262-
"%{rocm_root}/share/miopen/**",
263-
]),
271+
data = select({
272+
":build_hermetic": miopen_libs,
273+
":multiple_rocm_paths": miopen_libs,
274+
"//conditions:default": [],
275+
}),
264276
include_prefix = "rocm",
265277
includes = [
266278
"%{rocm_root}/include",
@@ -334,10 +346,16 @@ cc_library(
334346
deps = [":rocm_config"],
335347
)
336348

349+
roctracer_libs = glob(["%{rocm_root}/lib/libroctracer*.so*"])
350+
337351
cc_library(
338352
name = "roctracer",
339353
hdrs = glob(["%{rocm_root}/include/roctracer/**"]),
340-
data = glob(["%{rocm_root}/lib/libroctracer*.so*"]),
354+
data = select({
355+
":build_hermetic": roctracer_libs,
356+
":multiple_rocm_paths": roctracer_libs,
357+
"//conditions:default": [],
358+
}),
341359
include_prefix = "rocm",
342360
includes = [
343361
"%{rocm_root}/include/",
@@ -385,11 +403,19 @@ cc_library(
385403
deps = [":rocm_config"],
386404
)
387405

406+
hipsolver_libs = glob([
407+
"%{rocm_root}/lib/libhipsolver*.so*",
408+
])
409+
388410
cc_library(
389411
name = "hipsolver",
390412
srcs = glob(["%{rocm_root}/lib/libhipsolver*.so*"]),
391413
hdrs = glob(["%{rocm_root}/include/hipsolver/**"]),
392-
data = glob(["%{rocm_root}/lib/libhipsolver*.so*"]),
414+
data = select({
415+
":build_hermetic": hipsolver_libs,
416+
":multiple_rocm_paths": hipsolver_libs,
417+
"//conditions:default": [],
418+
}),
393419
include_prefix = "rocm",
394420
includes = [
395421
"%{rocm_root}/include/",
@@ -399,11 +425,17 @@ cc_library(
399425
deps = [":rocm_config"],
400426
)
401427

428+
hipblas_libs = glob(["%{rocm_root}/lib/libhipblas.so*"])
429+
402430
cc_library(
403431
name = "hipblas",
404432
srcs = glob(["%{rocm_root}/lib/libhipblas.so*"]),
405433
hdrs = glob(["%{rocm_root}/include/hipblas/**"]),
406-
data = glob(["%{rocm_root}/lib/libhipblas.so*"]),
434+
data = select({
435+
":build_hermetic": hipblas_libs,
436+
":multiple_rocm_paths": hipblas_libs,
437+
"//conditions:default": [],
438+
}),
407439
include_prefix = "rocm",
408440
includes = [
409441
"%{rocm_root}/include/",
@@ -428,13 +460,19 @@ cc_library(
428460
deps = [":rocm_config"],
429461
)
430462

463+
hipblaslt_libs = glob([
464+
"%{rocm_root}/lib/hipblaslt/**",
465+
"%{rocm_root}/lib/libhipblaslt.so*",
466+
])
467+
431468
cc_library(
432469
name = "hipblaslt",
433470
hdrs = glob(["%{rocm_root}/include/hipblaslt/**"]),
434-
data = glob([
435-
"%{rocm_root}/lib/hipblaslt/**",
436-
"%{rocm_root}/lib/libhipblaslt.so*",
437-
]),
471+
data = select({
472+
":build_hermetic": hipblaslt_libs,
473+
":multiple_rocm_paths": hipblaslt_libs,
474+
"//conditions:default": [],
475+
}),
438476
include_prefix = "rocm",
439477
includes = [
440478
"%{rocm_root}/include/",

0 commit comments

Comments
 (0)