fix parseNvidiaNumaInfo index out of range (#889)

Signed-off-by: bin <[email protected]>

Author: flpanbin

pkg/device-plugin/nvidiadevice/nvinternal/plugin/register.go

@@ -94,7 +94,7 @@ func parseNvidiaNumaInfo(idx int, nvidiaTopoStr string) (int, error) {
 		}
 		klog.V(5).InfoS("nvidia-smi topo -m row output", "row output", words, "length", len(words))
 		if strings.Contains(words[0], fmt.Sprint(idx)) {
-			if words[numaAffinityColumnIndex] == "N/A" {
+			if len(words) <= numaAffinityColumnIndex || words[numaAffinityColumnIndex] == "N/A" {
 				klog.InfoS("current card has not established numa topology", "gpu row info", words, "index", idx)
 				return 0, nil
 			}

pkg/device-plugin/nvidiadevice/nvinternal/plugin/register_test.go

@@ -67,6 +67,14 @@ func Test_parseNvidiaNumaInfo(t *testing.T) {
 			want:    0,
 			wantErr: false,
 		},
+		{
+			name: "NUMA Affinity is empty",
+			idx:  0,
+			nvidiaTopoStr: `GPU0	CPU Affinity	NUMA Affinity	GPU NUMA ID
+GPU0	X`,
+			want:    0,
+			wantErr: false,
+		},
 	}
 
 	for _, tt := range tests {