[WIP] : add ability to dynamically detect gdrcopy module

cmd/nvidia-device-plugin/plugin-manager.go

@@ -28,6 +28,7 @@ import (
 	"github.com/NVIDIA/k8s-device-plugin/internal/cdi"
 	"github.com/NVIDIA/k8s-device-plugin/internal/imex"
 	"github.com/NVIDIA/k8s-device-plugin/internal/plugin"
+	"github.com/NVIDIA/k8s-device-plugin/internal/utils"
 )
 
 // GetPlugins returns a set of plugins for the specified configuration.
@@ -54,7 +55,7 @@ func GetPlugins(ctx context.Context, infolib info.Interface, nvmllib nvml.Interf
 		cdi.WithNvidiaCTKPath(*config.Flags.Plugin.NvidiaCTKPath),
 		cdi.WithDeviceIDStrategy(*config.Flags.Plugin.DeviceIDStrategy),
 		cdi.WithVendor("k8s.device-plugin.nvidia.com"),
-		cdi.WithGdrcopyEnabled(*config.Flags.GDRCopyEnabled),
+		cdi.WithGdrcopyEnabled(*config.Flags.GDRCopyEnabled || utils.IsGdrdrvLoaded()),
 		cdi.WithGdsEnabled(*config.Flags.GDSEnabled),
 		cdi.WithMofedEnabled(*config.Flags.MOFEDEnabled),
 		cdi.WithImexChannels(imexChannels),

deployments/helm/nvidia-device-plugin/templates/daemonset-device-plugin.yml

@@ -220,6 +220,9 @@ spec:
             mountPath: /mps
           - name: cdi-root
             mountPath: /var/run/cdi
+          - name: proc-modules
+            mountPath: /host/proc/modules
+            readOnly: true
         {{- if $options.hasConfigMap }}
           - name: available-configs
             mountPath: /available-configs
@@ -252,6 +255,10 @@ spec:
           hostPath:
             path: /var/run/cdi
             type: DirectoryOrCreate
+        - name: proc-modules
+          hostPath:
+            path: /proc/modules
+            type: File
       {{- if $options.hasConfigMap }}
         - name: available-configs
           configMap:

internal/plugin/server.go

@@ -33,6 +33,7 @@ import (
 	"github.com/NVIDIA/k8s-device-plugin/internal/cdi"
 	"github.com/NVIDIA/k8s-device-plugin/internal/imex"
 	"github.com/NVIDIA/k8s-device-plugin/internal/rm"
+	"github.com/NVIDIA/k8s-device-plugin/internal/utils"
 
 	"github.com/google/uuid"
 	"google.golang.org/grpc"
@@ -352,7 +353,7 @@ func (plugin *nvidiaDevicePlugin) getAllocateResponse(requestIds []string) (*plu
 	if plugin.config.Flags.Plugin.PassDeviceSpecs != nil && *plugin.config.Flags.Plugin.PassDeviceSpecs {
 		response.Devices = append(response.Devices, plugin.apiDeviceSpecs(*plugin.config.Flags.NvidiaDevRoot, requestIds)...)
 	}
-	if plugin.config.Flags.GDRCopyEnabled != nil && *plugin.config.Flags.GDRCopyEnabled {
+	if (plugin.config.Flags.GDRCopyEnabled != nil && *plugin.config.Flags.GDRCopyEnabled) || utils.IsGdrdrvLoaded() {
 		response.Envs["NVIDIA_GDRCOPY"] = "enabled"
 	}
 	if plugin.config.Flags.GDSEnabled != nil && *plugin.config.Flags.GDSEnabled {

internal/utils/utils.go

@@ -0,0 +1,61 @@
+/**
+# Copyright (c) NVIDIA CORPORATION.  All rights reserved.
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+**/
+
+package utils
+
+import (
+	"bufio"
+	"os"
+	"slices"
+	"strings"
+)
+
+// Supported driver module constants.
+const (
+	GDRCOPY_DRIVER_MODULE = "gdrdrv"
+)
+
+// IsGdrdrvLoaded checks if the "gdrdrv" driver module is loaded.
+func IsGdrdrvLoaded() bool {
+	return isDriverLoaded(GDRCOPY_DRIVER_MODULE)
+}
+
+// isDriverLoaded checks if the specified driver module is loaded by reading the modules file.
+// It first checks "/host/proc/modules" (for containerized environments) and falls back to "/proc/modules" if not found.
+func isDriverLoaded(module string) bool {
+	paths := []string{"/host/proc/modules", "/proc/modules"}
+	return slices.ContainsFunc(paths, func(path string) bool {
+		return isDriverLoadedWithPath(module, path)
+	})
+}
+
+// isDriverLoadedWithPath checks if the specified driver module is loaded by reading the specified modules file.
+func isDriverLoadedWithPath(module, modulesPath string) bool {
+	f, err := os.Open(modulesPath)
+	if err != nil {
+		return false
+	}
+	defer f.Close()
+
+	scanner := bufio.NewScanner(f)
+	for scanner.Scan() {
+		line := scanner.Text()
+		if strings.HasPrefix(line, module+" ") {
+			return true
+		}
+	}
+	return false
+}