Add AI accelerator device mounting support in CreateControllerContainer

Need this to talk to device via Vulkan, ROCm, etc.

Signed-off-by: Eric Curtin <[email protected]>

Author: ericcurtin

pkg/standalone/containers.go

@@ -8,6 +8,7 @@ import (
 	"fmt"
 	"io"
 	"os"
+	"path/filepath"
 	"strconv"
 	"strings"
 	"time"
@@ -272,6 +273,25 @@ func CreateControllerContainer(ctx context.Context, dockerClient *client.Client,
 		hostConfig.DeviceRequests = []container.DeviceRequest{{Count: -1, Capabilities: [][]string{{"gpu"}}}}
 	}
 
+	// Enable access to AI accelerator devices if they exist
+	devicePaths := []string{"/dev/dri", "/dev/kfd", "/dev/accel", "/dev/davinci*", "/dev/devmm_svm", "/dev/hisi_hdc"}
+	for _, path := range devicePaths {
+		devices, err := filepath.Glob(path)
+		if err != nil {
+			// Skip on glob error, don't fail container creation
+			continue
+		}
+		for _, device := range devices {
+			if _, err := os.Stat(device); err == nil {
+				hostConfig.Devices = append(hostConfig.Devices, container.DeviceMapping{
+					PathOnHost:        device,
+					PathInContainer:   device,
+					CgroupPermissions: "rwm",
+				})
+			}
+		}
+	}
+
 	// Create the container. If we detect that a concurrent installation is in
 	// progress (as indicated by a conflicting container name (which should have
 	// been detected just before installation)), then we'll allow the error to