netbird/gvisor
0
0
Fork 0
mirror of https://github.com/netbirdio/gvisor.git synced 2026-05-22 17:12:49 -07:00
Code Issues Packages Projects Releases Wiki Activity

Fix glob pattern to search for all TPU devices in /sys/devices.

Browse Source 
Also fix the device file path used to fetch device ID.

Co-authored-by: Lucas Manning <lucasmanning@google.com>
PiperOrigin-RevId: 583112662
This commit is contained in:
Ayush Ranjan
2023-11-16 11:39:23 -08:00
committed by gVisor bot
co-authored by Lucas Manning
parent f154acfb7b
commit 7f5733ea38
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
runsc/boot/vfs.go
+2 -2
View File
@@ -1196,7 +1196,7 @@ func tpuProxyRegisterDevices(info *containerInfo, vfsObj *vfs.VirtualFilesystem)
}
// At this point /sys/devices/pci0000:00/<pci_address>/accel/accel# contains
// all the TPU devices on the host. Enumerate them and register TPU devices.
pciAddrs, err := filepath.Glob("/sys/devices/pci0000:00/*")
pciAddrs, err := filepath.Glob("/sys/devices/pci0000:00/*/accel/accel*")
if err != nil {
return fmt.Errorf("enumerating PCI device files: %w", err)
}
@@ -1211,7 +1211,7 @@ func tpuProxyRegisterDevices(info *containerInfo, vfsObj *vfs.VirtualFilesystem)
return fmt.Errorf("parsing PCI device number: %w", err)
}
var deviceIDBytes []byte
if deviceIDBytes, err = os.ReadFile(path.Join(pciPath, "device")); err != nil {
if deviceIDBytes, err = os.ReadFile(path.Join(pciPath, "device/device")); err != nil {
return fmt.Errorf("reading PCI device ID: %w", err)
}
deviceIDStr := strings.Replace(string(deviceIDBytes), "0x", "", -1)