Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
regexp_test.go 4.16 KB
Newer Older
liming6's avatar
init commit  
liming6 committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208 
package gpu

import (
	"encoding/json"
	"testing"
)

const (
	Data1 = `============================ System Management Interface =============================
======================================================================================
No KFD PIDs currently running!
======================================================================================
=================================== End of SMI Log ===================================`
	Data2 = `================================= System Management Interface ==================================
================================================================================================
PIDs for KFD processes:

PID: 142211
	   PASID: 32775
	   HCU Node(Include CPU sort): ['2']
	   HCU Index: ['0']
	   GPUID: ['15868']
	   PCI BUS: ['0000:49:00.0']
	   VRAM USED(MiB): 54682
	   VRAM USED(%): 83
	   SDMA USED: 0

PID: 142218
	   PASID: 32772
	   HCU Node(Include CPU sort): ['8']
	   HCU Index: ['6']
	   GPUID: ['37441']
	   PCI BUS: ['0000:cd:00.0']
	   VRAM USED(MiB): 43485
	   VRAM USED(%): 66
	   SDMA USED: 0

PID: 142236
	   PASID: 32774
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142216
	   PASID: 32773
	   HCU Node(Include CPU sort): ['6']
	   HCU Index: ['4']
	   GPUID: ['18915']
	   PCI BUS: ['0000:9c:00.0']
	   VRAM USED(MiB): 37438
	   VRAM USED(%): 57
	   SDMA USED: 0

PID: 142214
	   PASID: 32771
	   HCU Node(Include CPU sort): ['3']
	   HCU Index: ['1']
	   GPUID: ['51742']
	   PCI BUS: ['0000:54:00.0']
	   VRAM USED(MiB): 54815
	   VRAM USED(%): 84
	   SDMA USED: 0

PID: 142212
	   PASID: 32768
	   HCU Node(Include CPU sort): ['4']
	   HCU Index: ['2']
	   GPUID: ['14451']
	   PCI BUS: ['0000:5e:00.0']
	   VRAM USED(MiB): 40722
	   VRAM USED(%): 62
	   SDMA USED: 0

PID: 142249
	   PASID: 32780
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142239
	   PASID: 32776
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142247
	   PASID: 32781
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142217
	   PASID: 32770
	   HCU Node(Include CPU sort): ['5']
	   HCU Index: ['3']
	   GPUID: ['34940']
	   PCI BUS: ['0000:67:00.0']
	   VRAM USED(MiB): 40826
	   VRAM USED(%): 62
	   SDMA USED: 0

PID: 142245
	   PASID: 32779
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142235
	   PASID: 32784
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142215
	   PASID: 32797
	   HCU Node(Include CPU sort): ['9']
	   HCU Index: ['7']
	   GPUID: ['46537']
	   PCI BUS: ['0000:dd:00.0']
	   VRAM USED(MiB): 43279
	   VRAM USED(%): 66
	   SDMA USED: 0

PID: 142243
	   PASID: 32778
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

PID: 142213
	   PASID: 32769
	   HCU Node(Include CPU sort): ['7']
	   HCU Index: ['5']
	   GPUID: ['4240']
	   PCI BUS: ['0000:bc:00.0']
	   VRAM USED(MiB): 37436
	   VRAM USED(%): 57
	   SDMA USED: 0

PID: 142241
	   PASID: 32777
	   HCU Node(Include CPU sort):
	   HCU Index:
	   GPUID:
	   PCI BUS:
	   VRAM USED(MiB): 0
	   VRAM USED(%): inf
	   SDMA USED: 0

================================================================================================
======================================== End of SMI Log ========================================`
)

func TestRegexp(t *testing.T) {
	str := "[\"7\"]"
	ss := make([]string, 0)
	err := json.Unmarshal([]byte(str), &ss)
	if err != nil {
		t.Fatal(err)
	}
	t.Logf("%v", ss)
}

func TestParseDCUPidInfo(t *testing.T) {
	i, e := parseDCUPidInfo(Data2)
	if e != nil {
		t.Fatal(e)
	}
	for _, info := range i {
		t.Logf("%+v\n", info)
	}

	ii, e := parseDCUPidInfo(Data1)
	if e != nil {
		t.Fatal(e)
	}
	for _, info := range ii {
		t.Logf("%+v\n", info)
	}
}