support for new mig profiles

Signed-off-by: Varun Ramachandra Sekar <vsekar@nvidia.com>

Author: varunrsekar

pkg/nvlib/device/mig_profile.go

@@ -28,7 +28,8 @@ import (
 
 const (
 	// AttributeMediaExtensions holds the string representation for the media extension MIG profile attribute.
-	AttributeMediaExtensions = "me"
+	AttributeMediaExtensions    = "me"
+	AttributeMediaExtensionsAll = "me.all"
 )
 
 // MigProfile represents a specific MIG profile.
@@ -59,14 +60,21 @@ func (d *devicelib) NewMigProfile(giProfileID, ciProfileID, ciEngProfileID int,
 	switch giProfileID {
 	case nvml.GPU_INSTANCE_PROFILE_1_SLICE,
 		nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1,
-		nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV2:
+		nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV2,
+		nvml.GPU_INSTANCE_PROFILE_1_SLICE_GFX,
+		nvml.GPU_INSTANCE_PROFILE_1_SLICE_NO_ME,
+		nvml.GPU_INSTANCE_PROFILE_1_SLICE_ALL_ME:
 		giSlices = 1
 	case nvml.GPU_INSTANCE_PROFILE_2_SLICE,
-		nvml.GPU_INSTANCE_PROFILE_2_SLICE_REV1:
+		nvml.GPU_INSTANCE_PROFILE_2_SLICE_REV1,
+		nvml.GPU_INSTANCE_PROFILE_2_SLICE_GFX,
+		nvml.GPU_INSTANCE_PROFILE_2_SLICE_NO_ME,
+		nvml.GPU_INSTANCE_PROFILE_2_SLICE_ALL_ME:
 		giSlices = 2
 	case nvml.GPU_INSTANCE_PROFILE_3_SLICE:
 		giSlices = 3
-	case nvml.GPU_INSTANCE_PROFILE_4_SLICE:
+	case nvml.GPU_INSTANCE_PROFILE_4_SLICE,
+		nvml.GPU_INSTANCE_PROFILE_4_SLICE_GFX:
 		giSlices = 4
 	case nvml.GPU_INSTANCE_PROFILE_6_SLICE:
 		giSlices = 6
@@ -104,6 +112,9 @@ func (d *devicelib) NewMigProfile(giProfileID, ciProfileID, ciEngProfileID int,
 	case nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1,
 		nvml.GPU_INSTANCE_PROFILE_2_SLICE_REV1:
 		attrs = append(attrs, AttributeMediaExtensions)
+	case nvml.GPU_INSTANCE_PROFILE_1_SLICE_ALL_ME,
+		nvml.GPU_INSTANCE_PROFILE_2_SLICE_ALL_ME:
+		attrs = append(attrs, AttributeMediaExtensionsAll)
 	}
 
 	p := &MigProfileInfo{

pkg/nvlib/device/mig_profile_test.go

@@ -48,16 +48,23 @@ func newMockDeviceLib() Interface {
 			info := nvml.GpuInstanceProfileInfo{}
 			switch Profile {
 			case nvml.GPU_INSTANCE_PROFILE_1_SLICE,
-				nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1:
+				nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV1,
+				nvml.GPU_INSTANCE_PROFILE_1_SLICE_GFX,
+				nvml.GPU_INSTANCE_PROFILE_1_SLICE_NO_ME,
+				nvml.GPU_INSTANCE_PROFILE_1_SLICE_ALL_ME:
 				info.MemorySizeMB = 5 * 1024
 			case nvml.GPU_INSTANCE_PROFILE_1_SLICE_REV2:
 				info.MemorySizeMB = 10 * 1024
 			case nvml.GPU_INSTANCE_PROFILE_2_SLICE,
-				nvml.GPU_INSTANCE_PROFILE_2_SLICE_REV1:
+				nvml.GPU_INSTANCE_PROFILE_2_SLICE_REV1,
+				nvml.GPU_INSTANCE_PROFILE_2_SLICE_GFX,
+				nvml.GPU_INSTANCE_PROFILE_2_SLICE_NO_ME,
+				nvml.GPU_INSTANCE_PROFILE_2_SLICE_ALL_ME:
 				info.MemorySizeMB = 10 * 1024
 			case nvml.GPU_INSTANCE_PROFILE_3_SLICE:
 				info.MemorySizeMB = 20 * 1024
-			case nvml.GPU_INSTANCE_PROFILE_4_SLICE:
+			case nvml.GPU_INSTANCE_PROFILE_4_SLICE,
+				nvml.GPU_INSTANCE_PROFILE_4_SLICE_GFX:
 				info.MemorySizeMB = 20 * 1024
 			case nvml.GPU_INSTANCE_PROFILE_7_SLICE:
 				info.MemorySizeMB = 40 * 1024