mirror of
https://github.com/clearml/go-nvlib
synced 2025-03-15 08:08:19 +00:00
nvmdev: Add GetPhysicalFunction() for both Device and ParentDevice
Signed-off-by: Christopher Desiniotis <cdesiniotis@nvidia.com>
This commit is contained in:
parent
6ff7845b92
commit
bccac280ca
@ -319,6 +319,25 @@ func (m *Device) Delete() error {
|
|||||||
return nil
|
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
|
// IsMDEVTypeSupported checks if the mdevType is supported by the GPU
|
||||||
func (p *ParentDevice) IsMDEVTypeSupported(mdevType string) bool {
|
func (p *ParentDevice) IsMDEVTypeSupported(mdevType string) bool {
|
||||||
_, found := p.mdevPaths[mdevType]
|
_, found := p.mdevPaths[mdevType]
|
||||||
|
@ -33,6 +33,11 @@ func TestNvmdev(t *testing.T) {
|
|||||||
require.Equal(t, 1, len(parentDevs), "Wrong number of parent GPU devices")
|
require.Equal(t, 1, len(parentDevs), "Wrong number of parent GPU devices")
|
||||||
|
|
||||||
parentA100 := parentDevs[0]
|
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")
|
supported := parentA100.IsMDEVTypeSupported("A100-4C")
|
||||||
require.True(t, supported, "A100-4C should be a supported vGPU type")
|
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()
|
mdevs, err := nvmdev.GetAllDevices()
|
||||||
require.Nil(t, err, "Error getting NVIDIA MDEV (vGPU) devices")
|
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, 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")
|
mdevA100 := mdevs[0]
|
||||||
require.Equal(t, 200, mdevs[0].IommuGroup, "Wrong value for iommu_group")
|
|
||||||
|
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")
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user