diff --git a/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper.go b/pkg/scheduler/plugins/deviceshare/allocator_gpu_helper.go index 9d3db4f99..29c51bde1 100644 --- 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 index 07afd7cc2..05682b434 100644 --- 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 {