ClearML/go-nvlib
1
0
Fork 0
mirror of https://github.com/clearml/go-nvlib synced 2025-01-31 02:47:02 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix bug where MigProfile.Equals() would not work with wrapper type

Browse Source 
Signed-off-by: Kevin Klues <kklues@nvidia.com>
This commit is contained in:
Kevin Klues 2023-03-26 13:14:19 +00:00
parent 2b4f40a90b
commit 8d1b98baa6
2 changed files with 92 additions and 27 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

51
pkg/nvlib/device/mig_profile.go
View File

@ -142,7 +142,7 @@ func (d *devicelib) ParseMigProfile(profile string) (MigProfile, error) {
}
// String returns the string representation of a Profile
func (p *MigProfileInfo) String() string {
func (p MigProfileInfo) String() string {
var suffix string
if len(p.Attributes) > 0 {
suffix = "+" + strings.Join(p.Attributes, ",")
@ -154,39 +154,36 @@ func (p *MigProfileInfo) String() string {
}
// GetInfo returns detailed info about a Profile
func (p *MigProfileInfo) GetInfo() MigProfileInfo {
return *p
func (p MigProfileInfo) GetInfo() MigProfileInfo {
return p
}
// Equals checks if two Profiles are identical or not
func (p *MigProfileInfo) Equals(other MigProfile) bool {
switch o := other.(type) {
case *MigProfileInfo:
if p.C != o.C {
return false
}
if p.G != o.G {
return false
}
if p.GB != o.GB {
return false
}
if p.GIProfileID != o.GIProfileID {
return false
}
if p.CIProfileID != o.CIProfileID {
return false
}
if p.CIEngProfileID != o.CIEngProfileID {
return false
}
return true
func (p MigProfileInfo) Equals(other MigProfile) bool {
o := other.GetInfo()
if p.C != o.C {
return false
}
return false
if p.G != o.G {
return false
}
if p.GB != o.GB {
return false
}
if p.GIProfileID != o.GIProfileID {
return false
}
if p.CIProfileID != o.CIProfileID {
return false
}
if p.CIEngProfileID != o.CIEngProfileID {
return false
}
return true
}
// Matches checks if a MigProfile matches the string passed in
func (p *MigProfileInfo) Matches(profile string) bool {
func (p MigProfileInfo) Matches(profile string) bool {
c, g, gb, attrs, err := parseMigProfile(profile)
if err != nil {
return false

68
pkg/nvlib/device/mig_profile_test.go
View File

@ -24,6 +24,10 @@ import (
"gitlab.com/nvidia/cloud-native/go-nvlib/pkg/nvml"
)
type MigProfileInfoWrapper struct {
MigProfileInfo
}
func newMockDeviceLib() Interface {
mockDevice := &nvml.DeviceMock{
GetNameFunc: func() (string, nvml.Return) {
@ -362,6 +366,70 @@ func TestParseMigProfile(t *testing.T) {
}
}
func TestParseMigProfileEquals(t *testing.T) {
testCases := []struct {
description string
profile1 string
profile2 string
valid bool
}{
{
"Exactly equal",
"1g.5gb",
"1g.5gb",
true,
},
{
"Equal when expanded",
"1c.1g.5gb",
"1g.5gb",
true,
},
{
"Equal with attributes",
"1g.5gb+me",
"1g.5gb+me",
true,
},
{
"Not equal C slices",
"1c.2g.10gb",
"2c.2g.10gb",
false,
},
{
"Not equal G slices",
"1c.1g.10gb",
"1c.2g.10gb",
false,
},
{
"Not equal attributes",
"1g.5gb",
"1g.5gb+me",
false,
},
}
d := newMockDeviceLib()
for _, tc := range testCases {
t.Run(tc.description, func(t *testing.T) {
p1, err := d.ParseMigProfile(tc.profile1)
require.Nil(t, err)
p2, err := d.ParseMigProfile(tc.profile2)
require.Nil(t, err)
wrapper := MigProfileInfoWrapper{p2.GetInfo()}
if tc.valid {
require.True(t, p1.Equals(p2))
require.True(t, p1.Equals(wrapper))
} else {
require.False(t, p1.Equals(p2))
require.False(t, p1.Equals(wrapper))
}
})
}
}
func TestGetMigMemorySizeGB(t *testing.T) {
type testCase struct {
totalDeviceMemory uint64