ClearML/go-nvlib
1
0
Fork 0
mirror of https://github.com/clearml/go-nvlib synced 2025-03-03 10:42:09 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add test for NUMA node addition to nvpci

Browse Source 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
This commit is contained in:
Kevin Klues 2021-03-22 16:25:49 +00:00
parent 5adc7bd87c
commit 4066c09810
2 changed files with 48 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
pkg/nvpci/mock.go
View File

@ -53,7 +53,7 @@ func (m *MockNvpci) Cleanup() {
os.RemoveAll(m.pciDevicesRoot) os.RemoveAll(m.pciDevicesRoot)
} }
func (m *MockNvpci) AddMockA100(address string) error { func (m *MockNvpci) AddMockA100(address string, numaNode int) error {
deviceDir := filepath.Join(m.pciDevicesRoot, address) deviceDir := filepath.Join(m.pciDevicesRoot, address)
err := os.MkdirAll(deviceDir, 0755) err := os.MkdirAll(deviceDir, 0755)
if err != nil { if err != nil {
@ -87,6 +87,15 @@ func (m *MockNvpci) AddMockA100(address string) error {
return err return err
} }
numa, err := os.Create(filepath.Join(deviceDir, "numa_node"))
if err != nil {
return err
}
_, err = numa.WriteString(fmt.Sprintf("%v", numaNode))
if err != nil {
return err
}
config, err := os.Create(filepath.Join(deviceDir, "config")) config, err := os.Create(filepath.Join(deviceDir, "config"))
if err != nil { if err != nil {
return err return err

39
pkg/nvpci/nvpci_test.go
View File

@ -31,7 +31,7 @@ func TestNvpci(t *testing.T) {
require.Nil(t, err, "Error creating NewMockNvpci") require.Nil(t, err, "Error creating NewMockNvpci")
defer nvpci.Cleanup() defer nvpci.Cleanup()
err = nvpci.AddMockA100("0000:80:05.1") err = nvpci.AddMockA100("0000:80:05.1", 0)
require.Nil(t, err, "Error adding Mock A100 device to MockNvpci") require.Nil(t, err, "Error adding Mock A100 device to MockNvpci")
devices, err := nvpci.GetGPUs() devices, err := nvpci.GetGPUs()
@ -39,6 +39,7 @@ func TestNvpci(t *testing.T) {
require.Equal(t, 1, len(devices), "Wrong number of GPU devices") require.Equal(t, 1, len(devices), "Wrong number of GPU devices")
require.Equal(t, 1, len(devices[0].Resources), "Wrong number GPU resources found") require.Equal(t, 1, len(devices[0].Resources), "Wrong number GPU resources found")
require.Equal(t, "0000:80:05.1", devices[0].Address, "Wrong Address found for device") require.Equal(t, "0000:80:05.1", devices[0].Address, "Wrong Address found for device")
require.Equal(t, 0, devices[0].NumaNode, "Wrong NUMA node found for device")
config, err := devices[0].Config.Read() config, err := devices[0].Config.Read()
require.Nil(t, err, "Error reading config") require.Nil(t, err, "Error reading config")
@ -62,3 +63,39 @@ func TestNvpci(t *testing.T) {
require.Equal(t, int(resource0.End-resource0.Start+1), bar0.Len()) require.Equal(t, int(resource0.End-resource0.Start+1), bar0.Len())
require.Equal(t, ga100PmcID, bar0.Read32(0)) require.Equal(t, ga100PmcID, bar0.Read32(0))
} }
func TestNvpciNUMANode(t *testing.T) {
testCases := []struct {
Description string
NumaNode int
}{
{
Description: "Numa Node -1",
NumaNode: -1,
},
{
Description: "Numa Node 0",
NumaNode: 0,
},
{
Description: "Numa Node 1",
NumaNode: 1,
},
}
for _, tc := range testCases {
t.Run(tc.Description, func(t *testing.T) {
nvpci, err := NewMockNvpci()
require.Nil(t, err, "Error creating NewMockNvpci")
defer nvpci.Cleanup()
err = nvpci.AddMockA100("0000:80:05.1", tc.NumaNode)
require.Nil(t, err, "Error adding Mock A100 device to MockNvpci")
devices, err := nvpci.GetGPUs()
require.Nil(t, err, "Error getting GPUs")
require.Equal(t, 1, len(devices), "Wrong number of GPU devices")
require.Equal(t, tc.NumaNode, devices[0].NumaNode, "Wrong NUMA node found for device")
})
}
}