From c1eae0dedab70c5aecc183aaba075cdbb41b7f39 Mon Sep 17 00:00:00 2001
From: Evan Lezar <7723350-elezar@users.noreply.gitlab.com>
Date: Tue, 9 Jan 2024 14:37:08 +0000
Subject: [PATCH] Merge branch 'libnvdxgdmal' into 'main'

Add libnvdxgdmal library

See merge request nvidia/container-toolkit/container-toolkit!529
---
 CHANGELOG.md            | 1 +
 pkg/nvcdi/driver-wsl.go | 1 +
 2 files changed, 2 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index a5a3e77f..54ec6dc4 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -6,6 +6,7 @@
 * Fixed bug in `nvidia-ctk config` command when using `--set`. The types of applied config options are now applied correctly.
 * Log explicitly requested runtime mode.
 * Remove package dependency on libseccomp.
+* Added detection of libnvdxgdmal.so.1 on WSL2.
 
 ## v1.14.3
 * [toolkit-container] Bump CUDA base image version to 12.2.2.
diff --git a/pkg/nvcdi/driver-wsl.go b/pkg/nvcdi/driver-wsl.go
index 1aa02b8e..cda50226 100644
--- a/pkg/nvcdi/driver-wsl.go
+++ b/pkg/nvcdi/driver-wsl.go
@@ -33,6 +33,7 @@ var requiredDriverStoreFiles = []string{
 	"libnvidia-ml.so.1",             /* Core library for nvml */
 	"libnvidia-ml_loader.so",        /* Core library for nvml on WSL */
 	"libdxcore.so",                  /* Core library for dxcore support */
+	"libnvdxgdmal.so.1",             /* dxgdmal library for cuda */
 	"nvcubins.bin",                  /* Binary containing GPU code for cuda */
 	"nvidia-smi",                    /* nvidia-smi binary*/
 }