Skip to content

Commit 5a1b4e7

Browse files
committed
[no-relnote] Fix tests for compat mode
Signed-off-by: Evan Lezar <elezar@nvidia.com>
1 parent 39fd15d commit 5a1b4e7

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/e2e/nvidia-container-toolkit_test.go

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -201,19 +201,19 @@ var _ = Describe("docker", Ordered, ContinueOnFailure, func() {
201201
It("should work with the nvidia runtime in legacy mode", func(ctx context.Context) {
202202
ldconfigOut, _, err := runner.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia --gpus all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
203203
Expect(err).ToNot(HaveOccurred())
204-
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda/compat"))
204+
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda-12.9/compat/"))
205205
})
206206

207207
It("should work with the nvidia runtime in CDI mode", func(ctx context.Context) {
208208
ldconfigOut, _, err := runner.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=runtime.nvidia.com/gpu=all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
209209
Expect(err).ToNot(HaveOccurred())
210-
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda/compat"))
210+
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda-12.9/compat/"))
211211
})
212212

213-
It("should NOT work with nvidia-container-runtime-hook", func(ctx context.Context) {
213+
It("should work with nvidia-container-runtime-hook", func(ctx context.Context) {
214214
ldconfigOut, _, err := runner.Run("docker run --rm -i -e NVIDIA_DISABLE_REQUIRE=true --runtime=runc --gpus all nvcr.io/nvidia/cuda:12.9.0-base-ubi8 bash -c \"ldconfig -p | grep libcuda.so.1\"")
215215
Expect(err).ToNot(HaveOccurred())
216-
Expect(ldconfigOut).To(ContainSubstring("/usr/lib64"))
216+
Expect(ldconfigOut).To(ContainSubstring("/usr/local/cuda-12.9/compat/"))
217217
})
218218
})
219219

0 commit comments

Comments
 (0)