UniqAI/nvidia-container-toolkit
1
0
Fork 0
mirror of https://github.com/NVIDIA/nvidia-container-toolkit synced 2025-06-26 18:18:24 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix mode detection on Thor-based systems

Browse Source 
This change updates github.com/NVIDIA/go-nvlib from v0.7.1 to v0.7.2
to allow Thor systems to be detected as Tegra-based. This allows fixes
automatic mode detection to work on these systems.

Signed-off-by: Evan Lezar <elezar@nvidia.com>
This commit is contained in:
Evan Lezar
2025-05-13 21:25:11 +02:00
parent adb5e6719d
commit a4dc28bb3f
13 changed files with 236 additions and 93 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
vendor/github.com/NVIDIA/go-nvlib/pkg/nvpci/mmio/mmio.go generated vendored
View File

@@ -107,7 +107,7 @@ func (m *mmio) BigEndian() Mmio {
}
func (m *mmio) Close() error {
err := syscall.Munmap(*m.Bytes.Raw())
err := syscall.Munmap(*m.Raw())
if err != nil {
return fmt.Errorf("failed to munmap file: %v", err)
}
@@ -117,7 +117,7 @@ func (m *mmio) Close() error {
func (m *mmio) Sync() error {
_, _, errno := syscall.Syscall(
syscall.SYS_MSYNC,
uintptr(unsafe.Pointer(&(*m.Bytes.Raw())[0])),
uintptr(unsafe.Pointer(&(*m.Raw())[0])),
uintptr(m.Len()),
uintptr(syscall.MS_SYNC|syscall.MS_INVALIDATE))
if errno != 0 {

4
vendor/github.com/NVIDIA/go-nvlib/pkg/nvpci/mmio/mock.go generated vendored
View File

@@ -70,8 +70,8 @@ func (m *mockMmio) Sync() error {
if !m.rw {
return fmt.Errorf("opened read-only")
}
for i := range *m.Bytes.Raw() {
(*m.source)[m.offset+i] = (*m.Bytes.Raw())[i]
for i := range *m.Raw() {
(*m.source)[m.offset+i] = (*m.Raw())[i]
}
return nil
}

10
vendor/github.com/NVIDIA/go-nvlib/pkg/nvpci/mock.go generated vendored
View File

@@ -98,7 +98,7 @@ func (m *MockNvpci) AddMockA100(address string, numaNode int, sriov *SriovInfo)
if err != nil {
return err
}
_, err = numa.WriteString(fmt.Sprintf("%v", numaNode))
_, err = fmt.Fprintf(numa, "%v", numaNode)
if err != nil {
return err
}
@@ -137,7 +137,7 @@ func createNVIDIAgpuFiles(deviceDir string) error {
if err != nil {
return err
}
_, err = vendor.WriteString(fmt.Sprintf("0x%x", PCINvidiaVendorID))
_, err = fmt.Fprintf(vendor, "0x%x", PCINvidiaVendorID)
if err != nil {
return err
}
@@ -146,7 +146,7 @@ func createNVIDIAgpuFiles(deviceDir string) error {
if err != nil {
return err
}
_, err = class.WriteString(fmt.Sprintf("0x%x", PCI3dControllerClass))
_, err = fmt.Fprintf(class, "0x%x", PCI3dControllerClass)
if err != nil {
return err
}
@@ -188,7 +188,7 @@ func createNVIDIAgpuFiles(deviceDir string) error {
if err != nil {
return err
}
_, err = resource.WriteString(fmt.Sprintf("0x%x 0x%x 0x%x", bar0[0], bar0[1], bar0[2]))
_, err = fmt.Fprintf(resource, "0x%x 0x%x 0x%x", bar0[0], bar0[1], bar0[2])
if err != nil {
return err
}
@@ -246,7 +246,7 @@ func (m *MockNvpci) createVf(pfAddress string, id, iommu_group, numaNode int) er
if err != nil {
return err
}
_, err = numa.WriteString(fmt.Sprintf("%v", numaNode))
_, err = fmt.Fprintf(numa, "%v", numaNode)
if err != nil {
return err
}