Fix API call for GetVgpuCompatibility

Signed-off-by: Kevin Klues <kklues@nvidia.com>
NVIDIA · Apr 12, 2024 · f01937e · f01937e
1 parent 3d3e863
commit f01937e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/nvml/api.go b/pkg/nvml/api.go
@@ -343,7 +343,7 @@ type Interface interface {
 	VgpuInstanceGetAccountingMode(vgpuInstance VgpuInstance) (EnableState, Return)
 	VgpuInstanceGetAccountingPids(vgpuInstance VgpuInstance) ([]int, Return)
 	VgpuInstanceGetAccountingStats(vgpuInstance VgpuInstance, pid int) (AccountingStats, Return)
-	GetVgpuCompatibility(nvmlVgpuMetadata *nvmlVgpuMetadata, pgpuMetadata *nvmlVgpuPgpuMetadata) (VgpuPgpuCompatibility, Return)
+	GetVgpuCompatibility(vgpuMetadata *VgpuMetadata, pgpuMetadata *VgpuPgpuMetadata) (VgpuPgpuCompatibility, Return)
 	GetVgpuVersion() (VgpuVersion, VgpuVersion, Return)
 	SetVgpuVersion(vgpuVersion *VgpuVersion) Return
 	VgpuInstanceClearAccountingPids(vgpuInstance VgpuInstance) Return

diff --git a/pkg/nvml/vgpu.go b/pkg/nvml/vgpu.go
@@ -423,9 +423,9 @@ func (vgpuInstance nvmlVgpuInstance) GetAccountingStats(pid int) (AccountingStat
 }
 
 // nvml.GetVgpuCompatibility()
-func (l *library) GetVgpuCompatibility(nvmlVgpuMetadata *nvmlVgpuMetadata, pgpuMetadata *nvmlVgpuPgpuMetadata) (VgpuPgpuCompatibility, Return) {
+func (l *library) GetVgpuCompatibility(vgpuMetadata *VgpuMetadata, pgpuMetadata *VgpuPgpuMetadata) (VgpuPgpuCompatibility, Return) {
 	var compatibilityInfo VgpuPgpuCompatibility
-	ret := nvmlGetVgpuCompatibility(nvmlVgpuMetadata, pgpuMetadata, &compatibilityInfo)
+	ret := nvmlGetVgpuCompatibility(&vgpuMetadata.nvmlVgpuMetadata, &pgpuMetadata.nvmlVgpuPgpuMetadata, &compatibilityInfo)
 	return compatibilityInfo, ret
 }