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

nvmdev: Add GetPhysicalFunction() for both Device and ParentDevice

Browse Source 
Signed-off-by: Christopher Desiniotis <cdesiniotis@nvidia.com>
This commit is contained in:
Christopher Desiniotis 2022-08-23 16:03:28 -07:00
parent 6ff7845b92
commit bccac280ca
2 changed files with 34 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pkg/nvmdev/nvmdev.go
View File

@ -319,6 +319,25 @@ func (m *Device) Delete() error {
return nil
}
// GetPhysicalFunction gets the physical PCI device backing a 'parent' device
func (p *ParentDevice) GetPhysicalFunction() (*nvpci.NvidiaPCIDevice, error) {
if !p.IsVF {
return p.NvidiaPCIDevice, nil
}
physfnPath, err := filepath.EvalSymlinks(path.Join(p.Path, "physfn"))
if err != nil {
return nil, fmt.Errorf("unable to resolve %s: %v", path.Join(p.Path, "physfn"), err)
}
return nvpci.NewDevice(physfnPath)
}
// GetPhysicalFunction gets the physical PCI device that a vGPU is created on
func (m *Device) GetPhysicalFunction() (*nvpci.NvidiaPCIDevice, error) {
return m.Parent.GetPhysicalFunction()
}
// IsMDEVTypeSupported checks if the mdevType is supported by the GPU
func (p *ParentDevice) IsMDEVTypeSupported(mdevType string) bool {
_, found := p.mdevPaths[mdevType]

18
pkg/nvmdev/nvmdev_test.go
View File

@ -33,6 +33,11 @@ func TestNvmdev(t *testing.T) {
require.Equal(t, 1, len(parentDevs), "Wrong number of parent GPU devices")
parentA100 := parentDevs[0]
pf, err := parentA100.GetPhysicalFunction()
require.Nil(t, err, "Error getting physical function backing the Mock A100 parent device")
require.Equal(t, "0000:3b:04.1", pf.Address, "Wrong address for Mock A100 physical function")
supported := parentA100.IsMDEVTypeSupported("A100-4C")
require.True(t, supported, "A100-4C should be a supported vGPU type")
@ -46,7 +51,14 @@ func TestNvmdev(t *testing.T) {
mdevs, err := nvmdev.GetAllDevices()
require.Nil(t, err, "Error getting NVIDIA MDEV (vGPU) devices")
require.Equal(t, 1, len(mdevs), "Wrong number of NVIDIA MDEV (vGPU) devices")
require.Equal(t, "A100-4C", mdevs[0].MDEVType, "Wrong value for mdev_type")
require.Equal(t, "vfio_mdev", mdevs[0].Driver, "Wrong driver detected for mdev device")
require.Equal(t, 200, mdevs[0].IommuGroup, "Wrong value for iommu_group")
mdevA100 := mdevs[0]
require.Equal(t, "A100-4C", mdevA100.MDEVType, "Wrong value for mdev_type")
require.Equal(t, "vfio_mdev", mdevA100.Driver, "Wrong driver detected for mdev device")
require.Equal(t, 200, mdevA100.IommuGroup, "Wrong value for iommu_group")
pf, err = mdevA100.GetPhysicalFunction()
require.Nil(t, err, "Error getting the physical function for Mock A100 mediated device")
require.Equal(t, "0000:3b:04.1", pf.Address, "Wrong address for Mock A100 physical function")
}