scheduler: export GPUPartitionIndexOfNVIDIAHopper

Signed-off-by: wangjianyu.wjy <[email protected]>
koordinator-sh · Jan 20, 2025 · 8fe662c · 8fe662c
1 parent 236bcc9
commit 8fe662c
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper.go b/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper.go
@@ -26,7 +26,7 @@ import (
 )
 
 var (
-	gpuPartitionIndexOfNVIDIAHopper = GPUPartitionIndexer{
+	GPUPartitionIndexOfNVIDIAHopper = GPUPartitionIndexer{
 		1: []*PartitionsOfAllocationScore{
 			{
 				Partitions: []*apiext.GPUPartition{
@@ -153,7 +153,7 @@ var (
 		model := node.Labels[apiext.LabelGPUModel]
 		switch model {
 		case "H100", "H800", "H20":
-			partitionIndexer = gpuPartitionIndexOfNVIDIAHopper
+			partitionIndexer = GPUPartitionIndexOfNVIDIAHopper
 		}
 		return partitionIndexer, partitionPolicy == apiext.GPUPartitionPolicyHonor
 	}

diff --git a/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper_test.go b/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper_test.go
@@ -120,7 +120,7 @@ func TestGetGPUPartitionIndexer(t *testing.T) {
 					},
 				},
 			},
-			want: gpuPartitionIndexOfNVIDIAHopper,
+			want: GPUPartitionIndexOfNVIDIAHopper,
 		},
 	}
 	for _, tt := range tests {
-Original file line number
+Diff line change
@@ Expand Up / @@ -120,7 +120,7 @@ func TestGetGPUPartitionIndexer(t *testing.T) { @@
     					},
     				},
     			},
-    			want: gpuPartitionIndexOfNVIDIAHopper,
+    			want: GPUPartitionIndexOfNVIDIAHopper,
     		},
     	}
     	for _, tt := range tests {
@@ Expand Down @@