github/intel-device-plugins-for-kubernetes
7
0
Fork 0
mirror of https://github.com/intel/intel-device-plugins-for-kubernetes.git synced 2025-06-03 03:59:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #54 from zhenyw/gpu

Browse Source 
gpu_plugin: skip drm control node
This commit is contained in:
Alexander D. Kanevskiy 2018-07-26 15:03:04 +03:00 committed by GitHub
commit 6c08dbdb64
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
cmd/gpu_plugin/gpu_plugin.go
View File

@ -35,7 +35,8 @@ import (
const ( const (
sysfsDrmDirectory = "/sys/class/drm" sysfsDrmDirectory = "/sys/class/drm"
devfsDriDirectory = "/dev/dri" devfsDriDirectory = "/dev/dri"
gpuDeviceRE = `^card[0-9]*$` gpuDeviceRE = `^card[0-9]+$`
controlDeviceRE = `^controlD[0-9]+$`
vendorString = "0x8086" vendorString = "0x8086"
// Device plugin settings. // Device plugin settings.
@ -58,6 +59,7 @@ func newDeviceManager() *deviceManager {
// Discovers all GPU devices available on the local node by walking `/sys/class/drm` directory. // Discovers all GPU devices available on the local node by walking `/sys/class/drm` directory.
func (dm *deviceManager) discoverGPUs(sysfsDrmDir string, devfsDriDir string) error { func (dm *deviceManager) discoverGPUs(sysfsDrmDir string, devfsDriDir string) error {
reg := regexp.MustCompile(gpuDeviceRE) reg := regexp.MustCompile(gpuDeviceRE)
ctlReg := regexp.MustCompile(controlDeviceRE)
files, err := ioutil.ReadDir(sysfsDrmDir) files, err := ioutil.ReadDir(sysfsDrmDir)
if err != nil { if err != nil {
return fmt.Errorf("Can't read sysfs folder: %v", err) return fmt.Errorf("Can't read sysfs folder: %v", err)
@ -79,6 +81,10 @@ func (dm *deviceManager) discoverGPUs(sysfsDrmDir string, devfsDriDir string) er
} }
for _, drmFile := range drmFiles { for _, drmFile := range drmFiles {
if ctlReg.MatchString(drmFile.Name()) {
//Skipping possible drm control node
continue
}
devPath := path.Join(devfsDriDir, drmFile.Name()) devPath := path.Join(devfsDriDir, drmFile.Name())
if _, err := os.Stat(devPath); err != nil { if _, err := os.Stat(devPath); err != nil {
continue continue