Skip to content

Commit 2789e97

Browse files
committed
fixup! Create .so and SONAME symlinks for driver libraries
Signed-off-by: Evan Lezar <[email protected]>
1 parent bec2338 commit 2789e97

File tree

1 file changed

+1
-22
lines changed

1 file changed

+1
-22
lines changed

tests/e2e/nvidia-container-toolkit_test.go

Lines changed: 1 addition & 22 deletions
Original file line numberDiff line numberDiff line change
@@ -290,34 +290,13 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
290290
// lrwxrwxrwx libnvidia-ptxjitcompiler.so libnvidia-ptxjitcompiler.so.1
291291
// lrwxrwxrwx libnvidia-ptxjitcompiler.so.1 libnvidia-ptxjitcompiler.so.570.133.20
292292
// -rw-r--r-- libnvidia-ptxjitcompiler.so.570.133.20
293-
isTrackedLibrary := func(name string) bool {
294-
switch {
295-
case strings.HasPrefix(name, "libcuda.so"):
296-
fallthrough
297-
case strings.HasPrefix(name, "libcudadebugger.so"):
298-
fallthrough
299-
case strings.HasPrefix(name, "libnvidia-ml.so"):
300-
fallthrough
301-
case strings.HasPrefix(name, "libnvidia-nvvm.so"):
302-
fallthrough
303-
case strings.HasPrefix(name, "libnvidia-opencl.so"):
304-
fallthrough
305-
case strings.HasPrefix(name, "libnvidia-pkcs11-openssl3.so"):
306-
fallthrough
307-
case strings.HasPrefix(name, "libnvidia-pkcs11.so"):
308-
fallthrough
309-
case strings.HasPrefix(name, "libnvidia-ptxjitcompiler.so"):
310-
return true
311-
}
312-
return false
313-
}
314293

315294
symlinkOutputLines := strings.Split(symlinkOutput, "\n")
316295
symlinks := make(map[string][]string)
317296
var soSymlink string
318297
for _, line := range symlinkOutputLines {
319298
trimmed := strings.TrimSpace(line)
320-
if trimmed == "" || !isTrackedLibrary(trimmed) {
299+
if trimmed == "" {
321300
soSymlink = ""
322301
continue
323302
}

0 commit comments

Comments
 (0)