1 /*
2  * Copyright 2018 Advanced Micro Devices, Inc.
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  *
22  *
23  */
24 #include <linux/list.h>
25 #include "amdgpu.h"
26 #include "amdgpu_xgmi.h"
27 #include "amdgpu_ras.h"
28 #include "soc15.h"
29 #include "df/df_3_6_offset.h"
30 #include "xgmi/xgmi_4_0_0_smn.h"
31 #include "xgmi/xgmi_4_0_0_sh_mask.h"
32 #include "wafl/wafl2_4_0_0_smn.h"
33 #include "wafl/wafl2_4_0_0_sh_mask.h"
34 
35 #include "amdgpu_reset.h"
36 
37 #define smnPCS_XGMI3X16_PCS_ERROR_STATUS 0x11a0020c
38 #define smnPCS_GOPX1_PCS_ERROR_STATUS    0x12200210
39 
40 static DEFINE_MUTEX(xgmi_mutex);
41 
42 #define AMDGPU_MAX_XGMI_DEVICE_PER_HIVE		4
43 
44 static LIST_HEAD(xgmi_hive_list);
45 
46 static const int xgmi_pcs_err_status_reg_vg20[] = {
47 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS,
48 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x100000,
49 };
50 
51 static const int wafl_pcs_err_status_reg_vg20[] = {
52 	smnPCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS,
53 	smnPCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS + 0x100000,
54 };
55 
56 static const int xgmi_pcs_err_status_reg_arct[] = {
57 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS,
58 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x100000,
59 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x500000,
60 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x600000,
61 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x700000,
62 	smnXGMI0_PCS_GOPX16_PCS_ERROR_STATUS + 0x800000,
63 };
64 
65 /* same as vg20*/
66 static const int wafl_pcs_err_status_reg_arct[] = {
67 	smnPCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS,
68 	smnPCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS + 0x100000,
69 };
70 
71 static const int xgmi3x16_pcs_err_status_reg_aldebaran[] = {
72 	smnPCS_XGMI3X16_PCS_ERROR_STATUS,
73 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x100000,
74 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x200000,
75 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x300000,
76 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x400000,
77 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x500000,
78 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x600000,
79 	smnPCS_XGMI3X16_PCS_ERROR_STATUS + 0x700000
80 };
81 
82 static const int walf_pcs_err_status_reg_aldebaran[] = {
83 	smnPCS_GOPX1_PCS_ERROR_STATUS,
84 	smnPCS_GOPX1_PCS_ERROR_STATUS + 0x100000
85 };
86 
87 static const struct amdgpu_pcs_ras_field xgmi_pcs_ras_fields[] = {
88 	{"XGMI PCS DataLossErr",
89 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, DataLossErr)},
90 	{"XGMI PCS TrainingErr",
91 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, TrainingErr)},
92 	{"XGMI PCS CRCErr",
93 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, CRCErr)},
94 	{"XGMI PCS BERExceededErr",
95 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, BERExceededErr)},
96 	{"XGMI PCS TxMetaDataErr",
97 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, TxMetaDataErr)},
98 	{"XGMI PCS ReplayBufParityErr",
99 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ReplayBufParityErr)},
100 	{"XGMI PCS DataParityErr",
101 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, DataParityErr)},
102 	{"XGMI PCS ReplayFifoOverflowErr",
103 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ReplayFifoOverflowErr)},
104 	{"XGMI PCS ReplayFifoUnderflowErr",
105 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ReplayFifoUnderflowErr)},
106 	{"XGMI PCS ElasticFifoOverflowErr",
107 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ElasticFifoOverflowErr)},
108 	{"XGMI PCS DeskewErr",
109 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, DeskewErr)},
110 	{"XGMI PCS DataStartupLimitErr",
111 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, DataStartupLimitErr)},
112 	{"XGMI PCS FCInitTimeoutErr",
113 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, FCInitTimeoutErr)},
114 	{"XGMI PCS RecoveryTimeoutErr",
115 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, RecoveryTimeoutErr)},
116 	{"XGMI PCS ReadySerialTimeoutErr",
117 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ReadySerialTimeoutErr)},
118 	{"XGMI PCS ReadySerialAttemptErr",
119 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, ReadySerialAttemptErr)},
120 	{"XGMI PCS RecoveryAttemptErr",
121 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, RecoveryAttemptErr)},
122 	{"XGMI PCS RecoveryRelockAttemptErr",
123 	 SOC15_REG_FIELD(XGMI0_PCS_GOPX16_PCS_ERROR_STATUS, RecoveryRelockAttemptErr)},
124 };
125 
126 static const struct amdgpu_pcs_ras_field wafl_pcs_ras_fields[] = {
127 	{"WAFL PCS DataLossErr",
128 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, DataLossErr)},
129 	{"WAFL PCS TrainingErr",
130 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, TrainingErr)},
131 	{"WAFL PCS CRCErr",
132 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, CRCErr)},
133 	{"WAFL PCS BERExceededErr",
134 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, BERExceededErr)},
135 	{"WAFL PCS TxMetaDataErr",
136 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, TxMetaDataErr)},
137 	{"WAFL PCS ReplayBufParityErr",
138 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ReplayBufParityErr)},
139 	{"WAFL PCS DataParityErr",
140 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, DataParityErr)},
141 	{"WAFL PCS ReplayFifoOverflowErr",
142 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ReplayFifoOverflowErr)},
143 	{"WAFL PCS ReplayFifoUnderflowErr",
144 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ReplayFifoUnderflowErr)},
145 	{"WAFL PCS ElasticFifoOverflowErr",
146 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ElasticFifoOverflowErr)},
147 	{"WAFL PCS DeskewErr",
148 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, DeskewErr)},
149 	{"WAFL PCS DataStartupLimitErr",
150 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, DataStartupLimitErr)},
151 	{"WAFL PCS FCInitTimeoutErr",
152 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, FCInitTimeoutErr)},
153 	{"WAFL PCS RecoveryTimeoutErr",
154 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, RecoveryTimeoutErr)},
155 	{"WAFL PCS ReadySerialTimeoutErr",
156 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ReadySerialTimeoutErr)},
157 	{"WAFL PCS ReadySerialAttemptErr",
158 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, ReadySerialAttemptErr)},
159 	{"WAFL PCS RecoveryAttemptErr",
160 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, RecoveryAttemptErr)},
161 	{"WAFL PCS RecoveryRelockAttemptErr",
162 	 SOC15_REG_FIELD(PCS_GOPX1_0_PCS_GOPX1_PCS_ERROR_STATUS, RecoveryRelockAttemptErr)},
163 };
164 
165 /**
166  * DOC: AMDGPU XGMI Support
167  *
168  * XGMI is a high speed interconnect that joins multiple GPU cards
169  * into a homogeneous memory space that is organized by a collective
170  * hive ID and individual node IDs, both of which are 64-bit numbers.
171  *
172  * The file xgmi_device_id contains the unique per GPU device ID and
173  * is stored in the /sys/class/drm/card${cardno}/device/ directory.
174  *
175  * Inside the device directory a sub-directory 'xgmi_hive_info' is
176  * created which contains the hive ID and the list of nodes.
177  *
178  * The hive ID is stored in:
179  *   /sys/class/drm/card${cardno}/device/xgmi_hive_info/xgmi_hive_id
180  *
181  * The node information is stored in numbered directories:
182  *   /sys/class/drm/card${cardno}/device/xgmi_hive_info/node${nodeno}/xgmi_device_id
183  *
184  * Each device has their own xgmi_hive_info direction with a mirror
185  * set of node sub-directories.
186  *
187  * The XGMI memory space is built by contiguously adding the power of
188  * two padded VRAM space from each node to each other.
189  *
190  */
191 
192 static struct attribute amdgpu_xgmi_hive_id = {
193 	.name = "xgmi_hive_id",
194 	.mode = S_IRUGO
195 };
196 
197 static struct attribute *amdgpu_xgmi_hive_attrs[] = {
198 	&amdgpu_xgmi_hive_id,
199 	NULL
200 };
201 ATTRIBUTE_GROUPS(amdgpu_xgmi_hive);
202 
203 static ssize_t amdgpu_xgmi_show_attrs(struct kobject *kobj,
204 	struct attribute *attr, char *buf)
205 {
206 	struct amdgpu_hive_info *hive = container_of(
207 		kobj, struct amdgpu_hive_info, kobj);
208 
209 	if (attr == &amdgpu_xgmi_hive_id)
210 		return snprintf(buf, PAGE_SIZE, "%llu\n", hive->hive_id);
211 
212 	return 0;
213 }
214 
215 static void amdgpu_xgmi_hive_release(struct kobject *kobj)
216 {
217 	struct amdgpu_hive_info *hive = container_of(
218 		kobj, struct amdgpu_hive_info, kobj);
219 
220 	amdgpu_reset_put_reset_domain(hive->reset_domain);
221 	hive->reset_domain = NULL;
222 
223 	mutex_destroy(&hive->hive_lock);
224 	kfree(hive);
225 }
226 
227 static const struct sysfs_ops amdgpu_xgmi_hive_ops = {
228 	.show = amdgpu_xgmi_show_attrs,
229 };
230 
231 struct kobj_type amdgpu_xgmi_hive_type = {
232 	.release = amdgpu_xgmi_hive_release,
233 	.sysfs_ops = &amdgpu_xgmi_hive_ops,
234 	.default_groups = amdgpu_xgmi_hive_groups,
235 };
236 
237 static ssize_t amdgpu_xgmi_show_device_id(struct device *dev,
238 				     struct device_attribute *attr,
239 				     char *buf)
240 {
241 	struct drm_device *ddev = dev_get_drvdata(dev);
242 	struct amdgpu_device *adev = drm_to_adev(ddev);
243 
244 	return sysfs_emit(buf, "%llu\n", adev->gmc.xgmi.node_id);
245 
246 }
247 
248 #define AMDGPU_XGMI_SET_FICAA(o)	((o) | 0x456801)
249 static ssize_t amdgpu_xgmi_show_error(struct device *dev,
250 				      struct device_attribute *attr,
251 				      char *buf)
252 {
253 	struct drm_device *ddev = dev_get_drvdata(dev);
254 	struct amdgpu_device *adev = drm_to_adev(ddev);
255 	uint32_t ficaa_pie_ctl_in, ficaa_pie_status_in;
256 	uint64_t fica_out;
257 	unsigned int error_count = 0;
258 
259 	ficaa_pie_ctl_in = AMDGPU_XGMI_SET_FICAA(0x200);
260 	ficaa_pie_status_in = AMDGPU_XGMI_SET_FICAA(0x208);
261 
262 	if ((!adev->df.funcs) ||
263 	    (!adev->df.funcs->get_fica) ||
264 	    (!adev->df.funcs->set_fica))
265 		return -EINVAL;
266 
267 	fica_out = adev->df.funcs->get_fica(adev, ficaa_pie_ctl_in);
268 	if (fica_out != 0x1f)
269 		pr_err("xGMI error counters not enabled!\n");
270 
271 	fica_out = adev->df.funcs->get_fica(adev, ficaa_pie_status_in);
272 
273 	if ((fica_out & 0xffff) == 2)
274 		error_count = ((fica_out >> 62) & 0x1) + (fica_out >> 63);
275 
276 	adev->df.funcs->set_fica(adev, ficaa_pie_status_in, 0, 0);
277 
278 	return sysfs_emit(buf, "%u\n", error_count);
279 }
280 
281 
282 static DEVICE_ATTR(xgmi_device_id, S_IRUGO, amdgpu_xgmi_show_device_id, NULL);
283 static DEVICE_ATTR(xgmi_error, S_IRUGO, amdgpu_xgmi_show_error, NULL);
284 
285 static int amdgpu_xgmi_sysfs_add_dev_info(struct amdgpu_device *adev,
286 					 struct amdgpu_hive_info *hive)
287 {
288 	int ret = 0;
289 	char node[10] = { 0 };
290 
291 	/* Create xgmi device id file */
292 	ret = device_create_file(adev->dev, &dev_attr_xgmi_device_id);
293 	if (ret) {
294 		dev_err(adev->dev, "XGMI: Failed to create device file xgmi_device_id\n");
295 		return ret;
296 	}
297 
298 	/* Create xgmi error file */
299 	ret = device_create_file(adev->dev, &dev_attr_xgmi_error);
300 	if (ret)
301 		pr_err("failed to create xgmi_error\n");
302 
303 
304 	/* Create sysfs link to hive info folder on the first device */
305 	if (hive->kobj.parent != (&adev->dev->kobj)) {
306 		ret = sysfs_create_link(&adev->dev->kobj, &hive->kobj,
307 					"xgmi_hive_info");
308 		if (ret) {
309 			dev_err(adev->dev, "XGMI: Failed to create link to hive info");
310 			goto remove_file;
311 		}
312 	}
313 
314 	sprintf(node, "node%d", atomic_read(&hive->number_devices));
315 	/* Create sysfs link form the hive folder to yourself */
316 	ret = sysfs_create_link(&hive->kobj, &adev->dev->kobj, node);
317 	if (ret) {
318 		dev_err(adev->dev, "XGMI: Failed to create link from hive info");
319 		goto remove_link;
320 	}
321 
322 	goto success;
323 
324 
325 remove_link:
326 	sysfs_remove_link(&adev->dev->kobj, adev_to_drm(adev)->unique);
327 
328 remove_file:
329 	device_remove_file(adev->dev, &dev_attr_xgmi_device_id);
330 
331 success:
332 	return ret;
333 }
334 
335 static void amdgpu_xgmi_sysfs_rem_dev_info(struct amdgpu_device *adev,
336 					  struct amdgpu_hive_info *hive)
337 {
338 	char node[10];
339 	memset(node, 0, sizeof(node));
340 
341 	device_remove_file(adev->dev, &dev_attr_xgmi_device_id);
342 	device_remove_file(adev->dev, &dev_attr_xgmi_error);
343 
344 	if (hive->kobj.parent != (&adev->dev->kobj))
345 		sysfs_remove_link(&adev->dev->kobj,"xgmi_hive_info");
346 
347 	sprintf(node, "node%d", atomic_read(&hive->number_devices));
348 	sysfs_remove_link(&hive->kobj, node);
349 
350 }
351 
352 
353 
354 struct amdgpu_hive_info *amdgpu_get_xgmi_hive(struct amdgpu_device *adev)
355 {
356 	struct amdgpu_hive_info *hive = NULL;
357 	int ret;
358 
359 	if (!adev->gmc.xgmi.hive_id)
360 		return NULL;
361 
362 	if (adev->hive) {
363 		kobject_get(&adev->hive->kobj);
364 		return adev->hive;
365 	}
366 
367 	mutex_lock(&xgmi_mutex);
368 
369 	list_for_each_entry(hive, &xgmi_hive_list, node)  {
370 		if (hive->hive_id == adev->gmc.xgmi.hive_id)
371 			goto pro_end;
372 	}
373 
374 	hive = kzalloc(sizeof(*hive), GFP_KERNEL);
375 	if (!hive) {
376 		dev_err(adev->dev, "XGMI: allocation failed\n");
377 		hive = NULL;
378 		goto pro_end;
379 	}
380 
381 	/* initialize new hive if not exist */
382 	ret = kobject_init_and_add(&hive->kobj,
383 			&amdgpu_xgmi_hive_type,
384 			&adev->dev->kobj,
385 			"%s", "xgmi_hive_info");
386 	if (ret) {
387 		dev_err(adev->dev, "XGMI: failed initializing kobject for xgmi hive\n");
388 		kobject_put(&hive->kobj);
389 		kfree(hive);
390 		hive = NULL;
391 		goto pro_end;
392 	}
393 
394 	/**
395 	 * Avoid recreating reset domain when hive is reconstructed for the case
396 	 * of reset the devices in the XGMI hive during probe for SRIOV
397 	 * See https://www.spinics.net/lists/amd-gfx/msg58836.html
398 	 */
399 	if (adev->reset_domain->type != XGMI_HIVE) {
400 		hive->reset_domain = amdgpu_reset_create_reset_domain(XGMI_HIVE, "amdgpu-reset-hive");
401 			if (!hive->reset_domain) {
402 				dev_err(adev->dev, "XGMI: failed initializing reset domain for xgmi hive\n");
403 				ret = -ENOMEM;
404 				kobject_put(&hive->kobj);
405 				kfree(hive);
406 				hive = NULL;
407 				goto pro_end;
408 			}
409 	} else {
410 		amdgpu_reset_get_reset_domain(adev->reset_domain);
411 		hive->reset_domain = adev->reset_domain;
412 	}
413 
414 	hive->hive_id = adev->gmc.xgmi.hive_id;
415 	INIT_LIST_HEAD(&hive->device_list);
416 	INIT_LIST_HEAD(&hive->node);
417 	mutex_init(&hive->hive_lock);
418 	atomic_set(&hive->number_devices, 0);
419 	task_barrier_init(&hive->tb);
420 	hive->pstate = AMDGPU_XGMI_PSTATE_UNKNOWN;
421 	hive->hi_req_gpu = NULL;
422 
423 	/*
424 	 * hive pstate on boot is high in vega20 so we have to go to low
425 	 * pstate on after boot.
426 	 */
427 	hive->hi_req_count = AMDGPU_MAX_XGMI_DEVICE_PER_HIVE;
428 	list_add_tail(&hive->node, &xgmi_hive_list);
429 
430 pro_end:
431 	if (hive)
432 		kobject_get(&hive->kobj);
433 	mutex_unlock(&xgmi_mutex);
434 	return hive;
435 }
436 
437 void amdgpu_put_xgmi_hive(struct amdgpu_hive_info *hive)
438 {
439 	if (hive)
440 		kobject_put(&hive->kobj);
441 }
442 
443 int amdgpu_xgmi_set_pstate(struct amdgpu_device *adev, int pstate)
444 {
445 	int ret = 0;
446 	struct amdgpu_hive_info *hive;
447 	struct amdgpu_device *request_adev;
448 	bool is_hi_req = pstate == AMDGPU_XGMI_PSTATE_MAX_VEGA20;
449 	bool init_low;
450 
451 	hive = amdgpu_get_xgmi_hive(adev);
452 	if (!hive)
453 		return 0;
454 
455 	request_adev = hive->hi_req_gpu ? hive->hi_req_gpu : adev;
456 	init_low = hive->pstate == AMDGPU_XGMI_PSTATE_UNKNOWN;
457 	amdgpu_put_xgmi_hive(hive);
458 	/* fw bug so temporarily disable pstate switching */
459 	return 0;
460 
461 	if (!hive || adev->asic_type != CHIP_VEGA20)
462 		return 0;
463 
464 	mutex_lock(&hive->hive_lock);
465 
466 	if (is_hi_req)
467 		hive->hi_req_count++;
468 	else
469 		hive->hi_req_count--;
470 
471 	/*
472 	 * Vega20 only needs single peer to request pstate high for the hive to
473 	 * go high but all peers must request pstate low for the hive to go low
474 	 */
475 	if (hive->pstate == pstate ||
476 			(!is_hi_req && hive->hi_req_count && !init_low))
477 		goto out;
478 
479 	dev_dbg(request_adev->dev, "Set xgmi pstate %d.\n", pstate);
480 
481 	ret = amdgpu_dpm_set_xgmi_pstate(request_adev, pstate);
482 	if (ret) {
483 		dev_err(request_adev->dev,
484 			"XGMI: Set pstate failure on device %llx, hive %llx, ret %d",
485 			request_adev->gmc.xgmi.node_id,
486 			request_adev->gmc.xgmi.hive_id, ret);
487 		goto out;
488 	}
489 
490 	if (init_low)
491 		hive->pstate = hive->hi_req_count ?
492 					hive->pstate : AMDGPU_XGMI_PSTATE_MIN;
493 	else {
494 		hive->pstate = pstate;
495 		hive->hi_req_gpu = pstate != AMDGPU_XGMI_PSTATE_MIN ?
496 							adev : NULL;
497 	}
498 out:
499 	mutex_unlock(&hive->hive_lock);
500 	return ret;
501 }
502 
503 int amdgpu_xgmi_update_topology(struct amdgpu_hive_info *hive, struct amdgpu_device *adev)
504 {
505 	int ret;
506 
507 	if (amdgpu_sriov_vf(adev))
508 		return 0;
509 
510 	/* Each psp need to set the latest topology */
511 	ret = psp_xgmi_set_topology_info(&adev->psp,
512 					 atomic_read(&hive->number_devices),
513 					 &adev->psp.xgmi_context.top_info);
514 	if (ret)
515 		dev_err(adev->dev,
516 			"XGMI: Set topology failure on device %llx, hive %llx, ret %d",
517 			adev->gmc.xgmi.node_id,
518 			adev->gmc.xgmi.hive_id, ret);
519 
520 	return ret;
521 }
522 
523 
524 /*
525  * NOTE psp_xgmi_node_info.num_hops layout is as follows:
526  * num_hops[7:6] = link type (0 = xGMI2, 1 = xGMI3, 2/3 = reserved)
527  * num_hops[5:3] = reserved
528  * num_hops[2:0] = number of hops
529  */
530 int amdgpu_xgmi_get_hops_count(struct amdgpu_device *adev,
531 		struct amdgpu_device *peer_adev)
532 {
533 	struct psp_xgmi_topology_info *top = &adev->psp.xgmi_context.top_info;
534 	uint8_t num_hops_mask = 0x7;
535 	int i;
536 
537 	for (i = 0 ; i < top->num_nodes; ++i)
538 		if (top->nodes[i].node_id == peer_adev->gmc.xgmi.node_id)
539 			return top->nodes[i].num_hops & num_hops_mask;
540 	return	-EINVAL;
541 }
542 
543 int amdgpu_xgmi_get_num_links(struct amdgpu_device *adev,
544 		struct amdgpu_device *peer_adev)
545 {
546 	struct psp_xgmi_topology_info *top = &adev->psp.xgmi_context.top_info;
547 	int i;
548 
549 	for (i = 0 ; i < top->num_nodes; ++i)
550 		if (top->nodes[i].node_id == peer_adev->gmc.xgmi.node_id)
551 			return top->nodes[i].num_links;
552 	return	-EINVAL;
553 }
554 
555 /*
556  * Devices that support extended data require the entire hive to initialize with
557  * the shared memory buffer flag set.
558  *
559  * Hive locks and conditions apply - see amdgpu_xgmi_add_device
560  */
561 static int amdgpu_xgmi_initialize_hive_get_data_partition(struct amdgpu_hive_info *hive,
562 							bool set_extended_data)
563 {
564 	struct amdgpu_device *tmp_adev;
565 	int ret;
566 
567 	list_for_each_entry(tmp_adev, &hive->device_list, gmc.xgmi.head) {
568 		ret = psp_xgmi_initialize(&tmp_adev->psp, set_extended_data, false);
569 		if (ret) {
570 			dev_err(tmp_adev->dev,
571 				"XGMI: Failed to initialize xgmi session for data partition %i\n",
572 				set_extended_data);
573 			return ret;
574 		}
575 
576 	}
577 
578 	return 0;
579 }
580 
581 int amdgpu_xgmi_add_device(struct amdgpu_device *adev)
582 {
583 	struct psp_xgmi_topology_info *top_info;
584 	struct amdgpu_hive_info *hive;
585 	struct amdgpu_xgmi	*entry;
586 	struct amdgpu_device *tmp_adev = NULL;
587 
588 	int count = 0, ret = 0;
589 
590 	if (!adev->gmc.xgmi.supported)
591 		return 0;
592 
593 	if (!adev->gmc.xgmi.pending_reset &&
594 	    amdgpu_device_ip_get_ip_block(adev, AMD_IP_BLOCK_TYPE_PSP)) {
595 		ret = psp_xgmi_initialize(&adev->psp, false, true);
596 		if (ret) {
597 			dev_err(adev->dev,
598 				"XGMI: Failed to initialize xgmi session\n");
599 			return ret;
600 		}
601 
602 		ret = psp_xgmi_get_hive_id(&adev->psp, &adev->gmc.xgmi.hive_id);
603 		if (ret) {
604 			dev_err(adev->dev,
605 				"XGMI: Failed to get hive id\n");
606 			return ret;
607 		}
608 
609 		ret = psp_xgmi_get_node_id(&adev->psp, &adev->gmc.xgmi.node_id);
610 		if (ret) {
611 			dev_err(adev->dev,
612 				"XGMI: Failed to get node id\n");
613 			return ret;
614 		}
615 	} else {
616 		adev->gmc.xgmi.hive_id = 16;
617 		adev->gmc.xgmi.node_id = adev->gmc.xgmi.physical_node_id + 16;
618 	}
619 
620 	hive = amdgpu_get_xgmi_hive(adev);
621 	if (!hive) {
622 		ret = -EINVAL;
623 		dev_err(adev->dev,
624 			"XGMI: node 0x%llx, can not match hive 0x%llx in the hive list.\n",
625 			adev->gmc.xgmi.node_id, adev->gmc.xgmi.hive_id);
626 		goto exit;
627 	}
628 	mutex_lock(&hive->hive_lock);
629 
630 	top_info = &adev->psp.xgmi_context.top_info;
631 
632 	list_add_tail(&adev->gmc.xgmi.head, &hive->device_list);
633 	list_for_each_entry(entry, &hive->device_list, head)
634 		top_info->nodes[count++].node_id = entry->node_id;
635 	top_info->num_nodes = count;
636 	atomic_set(&hive->number_devices, count);
637 
638 	task_barrier_add_task(&hive->tb);
639 
640 	if (!adev->gmc.xgmi.pending_reset &&
641 	    amdgpu_device_ip_get_ip_block(adev, AMD_IP_BLOCK_TYPE_PSP)) {
642 		list_for_each_entry(tmp_adev, &hive->device_list, gmc.xgmi.head) {
643 			/* update node list for other device in the hive */
644 			if (tmp_adev != adev) {
645 				top_info = &tmp_adev->psp.xgmi_context.top_info;
646 				top_info->nodes[count - 1].node_id =
647 					adev->gmc.xgmi.node_id;
648 				top_info->num_nodes = count;
649 			}
650 			ret = amdgpu_xgmi_update_topology(hive, tmp_adev);
651 			if (ret)
652 				goto exit_unlock;
653 		}
654 
655 		/* get latest topology info for each device from psp */
656 		list_for_each_entry(tmp_adev, &hive->device_list, gmc.xgmi.head) {
657 			ret = psp_xgmi_get_topology_info(&tmp_adev->psp, count,
658 					&tmp_adev->psp.xgmi_context.top_info, false);
659 			if (ret) {
660 				dev_err(tmp_adev->dev,
661 					"XGMI: Get topology failure on device %llx, hive %llx, ret %d",
662 					tmp_adev->gmc.xgmi.node_id,
663 					tmp_adev->gmc.xgmi.hive_id, ret);
664 				/* To do : continue with some node failed or disable the whole hive */
665 				goto exit_unlock;
666 			}
667 		}
668 
669 		/* get topology again for hives that support extended data */
670 		if (adev->psp.xgmi_context.supports_extended_data) {
671 
672 			/* initialize the hive to get extended data.  */
673 			ret = amdgpu_xgmi_initialize_hive_get_data_partition(hive, true);
674 			if (ret)
675 				goto exit_unlock;
676 
677 			/* get the extended data. */
678 			list_for_each_entry(tmp_adev, &hive->device_list, gmc.xgmi.head) {
679 				ret = psp_xgmi_get_topology_info(&tmp_adev->psp, count,
680 						&tmp_adev->psp.xgmi_context.top_info, true);
681 				if (ret) {
682 					dev_err(tmp_adev->dev,
683 						"XGMI: Get topology for extended data failure on device %llx, hive %llx, ret %d",
684 						tmp_adev->gmc.xgmi.node_id,
685 						tmp_adev->gmc.xgmi.hive_id, ret);
686 					goto exit_unlock;
687 				}
688 			}
689 
690 			/* initialize the hive to get non-extended data for the next round. */
691 			ret = amdgpu_xgmi_initialize_hive_get_data_partition(hive, false);
692 			if (ret)
693 				goto exit_unlock;
694 
695 		}
696 	}
697 
698 	if (!ret && !adev->gmc.xgmi.pending_reset)
699 		ret = amdgpu_xgmi_sysfs_add_dev_info(adev, hive);
700 
701 exit_unlock:
702 	mutex_unlock(&hive->hive_lock);
703 exit:
704 	if (!ret) {
705 		adev->hive = hive;
706 		dev_info(adev->dev, "XGMI: Add node %d, hive 0x%llx.\n",
707 			 adev->gmc.xgmi.physical_node_id, adev->gmc.xgmi.hive_id);
708 	} else {
709 		amdgpu_put_xgmi_hive(hive);
710 		dev_err(adev->dev, "XGMI: Failed to add node %d, hive 0x%llx ret: %d\n",
711 			adev->gmc.xgmi.physical_node_id, adev->gmc.xgmi.hive_id,
712 			ret);
713 	}
714 
715 	return ret;
716 }
717 
718 int amdgpu_xgmi_remove_device(struct amdgpu_device *adev)
719 {
720 	struct amdgpu_hive_info *hive = adev->hive;
721 
722 	if (!adev->gmc.xgmi.supported)
723 		return -EINVAL;
724 
725 	if (!hive)
726 		return -EINVAL;
727 
728 	mutex_lock(&hive->hive_lock);
729 	task_barrier_rem_task(&hive->tb);
730 	amdgpu_xgmi_sysfs_rem_dev_info(adev, hive);
731 	if (hive->hi_req_gpu == adev)
732 		hive->hi_req_gpu = NULL;
733 	list_del(&adev->gmc.xgmi.head);
734 	mutex_unlock(&hive->hive_lock);
735 
736 	amdgpu_put_xgmi_hive(hive);
737 	adev->hive = NULL;
738 
739 	if (atomic_dec_return(&hive->number_devices) == 0) {
740 		/* Remove the hive from global hive list */
741 		mutex_lock(&xgmi_mutex);
742 		list_del(&hive->node);
743 		mutex_unlock(&xgmi_mutex);
744 
745 		amdgpu_put_xgmi_hive(hive);
746 	}
747 
748 	return 0;
749 }
750 
751 static int amdgpu_xgmi_ras_late_init(struct amdgpu_device *adev, struct ras_common_if *ras_block)
752 {
753 	if (!adev->gmc.xgmi.supported ||
754 	    adev->gmc.xgmi.num_physical_nodes == 0)
755 		return 0;
756 
757 	adev->gmc.xgmi.ras->ras_block.hw_ops->reset_ras_error_count(adev);
758 
759 	return amdgpu_ras_block_late_init(adev, ras_block);
760 }
761 
762 uint64_t amdgpu_xgmi_get_relative_phy_addr(struct amdgpu_device *adev,
763 					   uint64_t addr)
764 {
765 	struct amdgpu_xgmi *xgmi = &adev->gmc.xgmi;
766 	return (addr + xgmi->physical_node_id * xgmi->node_segment_size);
767 }
768 
769 static void pcs_clear_status(struct amdgpu_device *adev, uint32_t pcs_status_reg)
770 {
771 	WREG32_PCIE(pcs_status_reg, 0xFFFFFFFF);
772 	WREG32_PCIE(pcs_status_reg, 0);
773 }
774 
775 static void amdgpu_xgmi_reset_ras_error_count(struct amdgpu_device *adev)
776 {
777 	uint32_t i;
778 
779 	switch (adev->asic_type) {
780 	case CHIP_ARCTURUS:
781 		for (i = 0; i < ARRAY_SIZE(xgmi_pcs_err_status_reg_arct); i++)
782 			pcs_clear_status(adev,
783 					 xgmi_pcs_err_status_reg_arct[i]);
784 		break;
785 	case CHIP_VEGA20:
786 		for (i = 0; i < ARRAY_SIZE(xgmi_pcs_err_status_reg_vg20); i++)
787 			pcs_clear_status(adev,
788 					 xgmi_pcs_err_status_reg_vg20[i]);
789 		break;
790 	case CHIP_ALDEBARAN:
791 		for (i = 0; i < ARRAY_SIZE(xgmi3x16_pcs_err_status_reg_aldebaran); i++)
792 			pcs_clear_status(adev,
793 					 xgmi3x16_pcs_err_status_reg_aldebaran[i]);
794 		for (i = 0; i < ARRAY_SIZE(walf_pcs_err_status_reg_aldebaran); i++)
795 			pcs_clear_status(adev,
796 					 walf_pcs_err_status_reg_aldebaran[i]);
797 		break;
798 	default:
799 		break;
800 	}
801 }
802 
803 static int amdgpu_xgmi_query_pcs_error_status(struct amdgpu_device *adev,
804 					      uint32_t value,
805 					      uint32_t *ue_count,
806 					      uint32_t *ce_count,
807 					      bool is_xgmi_pcs)
808 {
809 	int i;
810 	int ue_cnt;
811 
812 	if (is_xgmi_pcs) {
813 		/* query xgmi pcs error status,
814 		 * only ue is supported */
815 		for (i = 0; i < ARRAY_SIZE(xgmi_pcs_ras_fields); i ++) {
816 			ue_cnt = (value &
817 				  xgmi_pcs_ras_fields[i].pcs_err_mask) >>
818 				  xgmi_pcs_ras_fields[i].pcs_err_shift;
819 			if (ue_cnt) {
820 				dev_info(adev->dev, "%s detected\n",
821 					 xgmi_pcs_ras_fields[i].err_name);
822 				*ue_count += ue_cnt;
823 			}
824 		}
825 	} else {
826 		/* query wafl pcs error status,
827 		 * only ue is supported */
828 		for (i = 0; i < ARRAY_SIZE(wafl_pcs_ras_fields); i++) {
829 			ue_cnt = (value &
830 				  wafl_pcs_ras_fields[i].pcs_err_mask) >>
831 				  wafl_pcs_ras_fields[i].pcs_err_shift;
832 			if (ue_cnt) {
833 				dev_info(adev->dev, "%s detected\n",
834 					 wafl_pcs_ras_fields[i].err_name);
835 				*ue_count += ue_cnt;
836 			}
837 		}
838 	}
839 
840 	return 0;
841 }
842 
843 static void amdgpu_xgmi_query_ras_error_count(struct amdgpu_device *adev,
844 					     void *ras_error_status)
845 {
846 	struct ras_err_data *err_data = (struct ras_err_data *)ras_error_status;
847 	int i;
848 	uint32_t data;
849 	uint32_t ue_cnt = 0, ce_cnt = 0;
850 
851 	if (!amdgpu_ras_is_supported(adev, AMDGPU_RAS_BLOCK__XGMI_WAFL))
852 		return ;
853 
854 	err_data->ue_count = 0;
855 	err_data->ce_count = 0;
856 
857 	switch (adev->asic_type) {
858 	case CHIP_ARCTURUS:
859 		/* check xgmi pcs error */
860 		for (i = 0; i < ARRAY_SIZE(xgmi_pcs_err_status_reg_arct); i++) {
861 			data = RREG32_PCIE(xgmi_pcs_err_status_reg_arct[i]);
862 			if (data)
863 				amdgpu_xgmi_query_pcs_error_status(adev,
864 						data, &ue_cnt, &ce_cnt, true);
865 		}
866 		/* check wafl pcs error */
867 		for (i = 0; i < ARRAY_SIZE(wafl_pcs_err_status_reg_arct); i++) {
868 			data = RREG32_PCIE(wafl_pcs_err_status_reg_arct[i]);
869 			if (data)
870 				amdgpu_xgmi_query_pcs_error_status(adev,
871 						data, &ue_cnt, &ce_cnt, false);
872 		}
873 		break;
874 	case CHIP_VEGA20:
875 		/* check xgmi pcs error */
876 		for (i = 0; i < ARRAY_SIZE(xgmi_pcs_err_status_reg_vg20); i++) {
877 			data = RREG32_PCIE(xgmi_pcs_err_status_reg_vg20[i]);
878 			if (data)
879 				amdgpu_xgmi_query_pcs_error_status(adev,
880 						data, &ue_cnt, &ce_cnt, true);
881 		}
882 		/* check wafl pcs error */
883 		for (i = 0; i < ARRAY_SIZE(wafl_pcs_err_status_reg_vg20); i++) {
884 			data = RREG32_PCIE(wafl_pcs_err_status_reg_vg20[i]);
885 			if (data)
886 				amdgpu_xgmi_query_pcs_error_status(adev,
887 						data, &ue_cnt, &ce_cnt, false);
888 		}
889 		break;
890 	case CHIP_ALDEBARAN:
891 		/* check xgmi3x16 pcs error */
892 		for (i = 0; i < ARRAY_SIZE(xgmi3x16_pcs_err_status_reg_aldebaran); i++) {
893 			data = RREG32_PCIE(xgmi3x16_pcs_err_status_reg_aldebaran[i]);
894 			if (data)
895 				amdgpu_xgmi_query_pcs_error_status(adev,
896 						data, &ue_cnt, &ce_cnt, true);
897 		}
898 		/* check wafl pcs error */
899 		for (i = 0; i < ARRAY_SIZE(walf_pcs_err_status_reg_aldebaran); i++) {
900 			data = RREG32_PCIE(walf_pcs_err_status_reg_aldebaran[i]);
901 			if (data)
902 				amdgpu_xgmi_query_pcs_error_status(adev,
903 						data, &ue_cnt, &ce_cnt, false);
904 		}
905 		break;
906 	default:
907 		dev_warn(adev->dev, "XGMI RAS error query not supported");
908 		break;
909 	}
910 
911 	adev->gmc.xgmi.ras->ras_block.hw_ops->reset_ras_error_count(adev);
912 
913 	err_data->ue_count += ue_cnt;
914 	err_data->ce_count += ce_cnt;
915 }
916 
917 /* Trigger XGMI/WAFL error */
918 static int amdgpu_ras_error_inject_xgmi(struct amdgpu_device *adev,  void *inject_if)
919 {
920 	int ret = 0;
921 	struct ta_ras_trigger_error_input *block_info =
922 				(struct ta_ras_trigger_error_input *)inject_if;
923 
924 	if (amdgpu_dpm_set_df_cstate(adev, DF_CSTATE_DISALLOW))
925 		dev_warn(adev->dev, "Failed to disallow df cstate");
926 
927 	if (amdgpu_dpm_allow_xgmi_power_down(adev, false))
928 		dev_warn(adev->dev, "Failed to disallow XGMI power down");
929 
930 	ret = psp_ras_trigger_error(&adev->psp, block_info);
931 
932 	if (amdgpu_ras_intr_triggered())
933 		return ret;
934 
935 	if (amdgpu_dpm_allow_xgmi_power_down(adev, true))
936 		dev_warn(adev->dev, "Failed to allow XGMI power down");
937 
938 	if (amdgpu_dpm_set_df_cstate(adev, DF_CSTATE_ALLOW))
939 		dev_warn(adev->dev, "Failed to allow df cstate");
940 
941 	return ret;
942 }
943 
944 struct amdgpu_ras_block_hw_ops  xgmi_ras_hw_ops = {
945 	.query_ras_error_count = amdgpu_xgmi_query_ras_error_count,
946 	.reset_ras_error_count = amdgpu_xgmi_reset_ras_error_count,
947 	.ras_error_inject = amdgpu_ras_error_inject_xgmi,
948 };
949 
950 struct amdgpu_xgmi_ras xgmi_ras = {
951 	.ras_block = {
952 		.ras_comm = {
953 			.name = "xgmi_wafl",
954 			.block = AMDGPU_RAS_BLOCK__XGMI_WAFL,
955 			.type = AMDGPU_RAS_ERROR__MULTI_UNCORRECTABLE,
956 		},
957 		.hw_ops = &xgmi_ras_hw_ops,
958 		.ras_late_init = amdgpu_xgmi_ras_late_init,
959 	},
960 };
961