1/*
2 * Copyright 2022 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#include "amdgpu.h"
24#include "soc15.h"
25
26#include "soc15_common.h"
27#include "amdgpu_reg_state.h"
28#include "amdgpu_xcp.h"
29#include "gfx_v9_4_3.h"
30#include "gfxhub_v1_2.h"
31#include "sdma_v4_4_2.h"
32
33#define XCP_INST_MASK(num_inst, xcp_id)                                        \
34	(num_inst ? GENMASK(num_inst - 1, 0) << (xcp_id * num_inst) : 0)
35
36#define AMDGPU_XCP_OPS_KFD	(1 << 0)
37
38void aqua_vanjaram_doorbell_index_init(struct amdgpu_device *adev)
39{
40	int i;
41
42	adev->doorbell_index.kiq = AMDGPU_DOORBELL_LAYOUT1_KIQ_START;
43
44	adev->doorbell_index.mec_ring0 = AMDGPU_DOORBELL_LAYOUT1_MEC_RING_START;
45
46	adev->doorbell_index.userqueue_start = AMDGPU_DOORBELL_LAYOUT1_USERQUEUE_START;
47	adev->doorbell_index.userqueue_end = AMDGPU_DOORBELL_LAYOUT1_USERQUEUE_END;
48	adev->doorbell_index.xcc_doorbell_range = AMDGPU_DOORBELL_LAYOUT1_XCC_RANGE;
49
50	adev->doorbell_index.sdma_doorbell_range = 20;
51	for (i = 0; i < adev->sdma.num_instances; i++)
52		adev->doorbell_index.sdma_engine[i] =
53			AMDGPU_DOORBELL_LAYOUT1_sDMA_ENGINE_START +
54			i * (adev->doorbell_index.sdma_doorbell_range >> 1);
55
56	adev->doorbell_index.ih = AMDGPU_DOORBELL_LAYOUT1_IH;
57	adev->doorbell_index.vcn.vcn_ring0_1 = AMDGPU_DOORBELL_LAYOUT1_VCN_START;
58
59	adev->doorbell_index.first_non_cp = AMDGPU_DOORBELL_LAYOUT1_FIRST_NON_CP;
60	adev->doorbell_index.last_non_cp = AMDGPU_DOORBELL_LAYOUT1_LAST_NON_CP;
61
62	adev->doorbell_index.max_assignment = AMDGPU_DOORBELL_LAYOUT1_MAX_ASSIGNMENT << 1;
63}
64
65static bool aqua_vanjaram_xcp_vcn_shared(struct amdgpu_device *adev)
66{
67	return (adev->xcp_mgr->num_xcps > adev->vcn.num_vcn_inst);
68}
69
70static void aqua_vanjaram_set_xcp_id(struct amdgpu_device *adev,
71			     uint32_t inst_idx, struct amdgpu_ring *ring)
72{
73	int xcp_id;
74	enum AMDGPU_XCP_IP_BLOCK ip_blk;
75	uint32_t inst_mask;
76
77	ring->xcp_id = AMDGPU_XCP_NO_PARTITION;
78	if (adev->xcp_mgr->mode == AMDGPU_XCP_MODE_NONE)
79		return;
80
81	inst_mask = 1 << inst_idx;
82
83	switch (ring->funcs->type) {
84	case AMDGPU_HW_IP_GFX:
85	case AMDGPU_RING_TYPE_COMPUTE:
86	case AMDGPU_RING_TYPE_KIQ:
87		ip_blk = AMDGPU_XCP_GFX;
88		break;
89	case AMDGPU_RING_TYPE_SDMA:
90		ip_blk = AMDGPU_XCP_SDMA;
91		break;
92	case AMDGPU_RING_TYPE_VCN_ENC:
93	case AMDGPU_RING_TYPE_VCN_JPEG:
94		ip_blk = AMDGPU_XCP_VCN;
95		if (aqua_vanjaram_xcp_vcn_shared(adev))
96			inst_mask = 1 << (inst_idx * 2);
97		break;
98	default:
99		DRM_ERROR("Not support ring type %d!", ring->funcs->type);
100		return;
101	}
102
103	for (xcp_id = 0; xcp_id < adev->xcp_mgr->num_xcps; xcp_id++) {
104		if (adev->xcp_mgr->xcp[xcp_id].ip[ip_blk].inst_mask & inst_mask) {
105			ring->xcp_id = xcp_id;
106			break;
107		}
108	}
109}
110
111static void aqua_vanjaram_xcp_gpu_sched_update(
112		struct amdgpu_device *adev,
113		struct amdgpu_ring *ring,
114		unsigned int sel_xcp_id)
115{
116	unsigned int *num_gpu_sched;
117
118	num_gpu_sched = &adev->xcp_mgr->xcp[sel_xcp_id]
119			.gpu_sched[ring->funcs->type][ring->hw_prio].num_scheds;
120	adev->xcp_mgr->xcp[sel_xcp_id].gpu_sched[ring->funcs->type][ring->hw_prio]
121			.sched[(*num_gpu_sched)++] = &ring->sched;
122	DRM_DEBUG("%s :[%d] gpu_sched[%d][%d] = %d", ring->name,
123			sel_xcp_id, ring->funcs->type,
124			ring->hw_prio, *num_gpu_sched);
125}
126
127static int aqua_vanjaram_xcp_sched_list_update(
128		struct amdgpu_device *adev)
129{
130	struct amdgpu_ring *ring;
131	int i;
132
133	for (i = 0; i < MAX_XCP; i++) {
134		atomic_set(&adev->xcp_mgr->xcp[i].ref_cnt, 0);
135		memset(adev->xcp_mgr->xcp[i].gpu_sched, 0, sizeof(adev->xcp_mgr->xcp->gpu_sched));
136	}
137
138	if (adev->xcp_mgr->mode == AMDGPU_XCP_MODE_NONE)
139		return 0;
140
141	for (i = 0; i < AMDGPU_MAX_RINGS; i++) {
142		ring = adev->rings[i];
143		if (!ring || !ring->sched.ready || ring->no_scheduler)
144			continue;
145
146		aqua_vanjaram_xcp_gpu_sched_update(adev, ring, ring->xcp_id);
147
148		/* VCN may be shared by two partitions under CPX MODE in certain
149		 * configs.
150		 */
151		if ((ring->funcs->type == AMDGPU_RING_TYPE_VCN_ENC ||
152		     ring->funcs->type == AMDGPU_RING_TYPE_VCN_JPEG) &&
153		    aqua_vanjaram_xcp_vcn_shared(adev))
154			aqua_vanjaram_xcp_gpu_sched_update(adev, ring, ring->xcp_id + 1);
155	}
156
157	return 0;
158}
159
160static int aqua_vanjaram_update_partition_sched_list(struct amdgpu_device *adev)
161{
162	int i;
163
164	for (i = 0; i < adev->num_rings; i++) {
165		struct amdgpu_ring *ring = adev->rings[i];
166
167		if (ring->funcs->type == AMDGPU_RING_TYPE_COMPUTE ||
168			ring->funcs->type == AMDGPU_RING_TYPE_KIQ)
169			aqua_vanjaram_set_xcp_id(adev, ring->xcc_id, ring);
170		else
171			aqua_vanjaram_set_xcp_id(adev, ring->me, ring);
172	}
173
174	return aqua_vanjaram_xcp_sched_list_update(adev);
175}
176
177static int aqua_vanjaram_select_scheds(
178		struct amdgpu_device *adev,
179		u32 hw_ip,
180		u32 hw_prio,
181		struct amdgpu_fpriv *fpriv,
182		unsigned int *num_scheds,
183		struct drm_gpu_scheduler ***scheds)
184{
185	u32 sel_xcp_id;
186	int i;
187
188	if (fpriv->xcp_id == AMDGPU_XCP_NO_PARTITION) {
189		u32 least_ref_cnt = ~0;
190
191		fpriv->xcp_id = 0;
192		for (i = 0; i < adev->xcp_mgr->num_xcps; i++) {
193			u32 total_ref_cnt;
194
195			total_ref_cnt = atomic_read(&adev->xcp_mgr->xcp[i].ref_cnt);
196			if (total_ref_cnt < least_ref_cnt) {
197				fpriv->xcp_id = i;
198				least_ref_cnt = total_ref_cnt;
199			}
200		}
201	}
202	sel_xcp_id = fpriv->xcp_id;
203
204	if (adev->xcp_mgr->xcp[sel_xcp_id].gpu_sched[hw_ip][hw_prio].num_scheds) {
205		*num_scheds = adev->xcp_mgr->xcp[fpriv->xcp_id].gpu_sched[hw_ip][hw_prio].num_scheds;
206		*scheds = adev->xcp_mgr->xcp[fpriv->xcp_id].gpu_sched[hw_ip][hw_prio].sched;
207		atomic_inc(&adev->xcp_mgr->xcp[sel_xcp_id].ref_cnt);
208		DRM_DEBUG("Selected partition #%d", sel_xcp_id);
209	} else {
210		DRM_ERROR("Failed to schedule partition #%d.", sel_xcp_id);
211		return -ENOENT;
212	}
213
214	return 0;
215}
216
217static int8_t aqua_vanjaram_logical_to_dev_inst(struct amdgpu_device *adev,
218					 enum amd_hw_ip_block_type block,
219					 int8_t inst)
220{
221	int8_t dev_inst;
222
223	switch (block) {
224	case GC_HWIP:
225	case SDMA0_HWIP:
226	/* Both JPEG and VCN as JPEG is only alias of VCN */
227	case VCN_HWIP:
228		dev_inst = adev->ip_map.dev_inst[block][inst];
229		break;
230	default:
231		/* For rest of the IPs, no look up required.
232		 * Assume 'logical instance == physical instance' for all configs. */
233		dev_inst = inst;
234		break;
235	}
236
237	return dev_inst;
238}
239
240static uint32_t aqua_vanjaram_logical_to_dev_mask(struct amdgpu_device *adev,
241					 enum amd_hw_ip_block_type block,
242					 uint32_t mask)
243{
244	uint32_t dev_mask = 0;
245	int8_t log_inst, dev_inst;
246
247	while (mask) {
248		log_inst = ffs(mask) - 1;
249		dev_inst = aqua_vanjaram_logical_to_dev_inst(adev, block, log_inst);
250		dev_mask |= (1 << dev_inst);
251		mask &= ~(1 << log_inst);
252	}
253
254	return dev_mask;
255}
256
257static void aqua_vanjaram_populate_ip_map(struct amdgpu_device *adev,
258					  enum amd_hw_ip_block_type ip_block,
259					  uint32_t inst_mask)
260{
261	int l = 0, i;
262
263	while (inst_mask) {
264		i = ffs(inst_mask) - 1;
265		adev->ip_map.dev_inst[ip_block][l++] = i;
266		inst_mask &= ~(1 << i);
267	}
268	for (; l < HWIP_MAX_INSTANCE; l++)
269		adev->ip_map.dev_inst[ip_block][l] = -1;
270}
271
272void aqua_vanjaram_ip_map_init(struct amdgpu_device *adev)
273{
274	u32 ip_map[][2] = {
275		{ GC_HWIP, adev->gfx.xcc_mask },
276		{ SDMA0_HWIP, adev->sdma.sdma_mask },
277		{ VCN_HWIP, adev->vcn.inst_mask },
278	};
279	int i;
280
281	for (i = 0; i < ARRAY_SIZE(ip_map); ++i)
282		aqua_vanjaram_populate_ip_map(adev, ip_map[i][0], ip_map[i][1]);
283
284	adev->ip_map.logical_to_dev_inst = aqua_vanjaram_logical_to_dev_inst;
285	adev->ip_map.logical_to_dev_mask = aqua_vanjaram_logical_to_dev_mask;
286}
287
288/* Fixed pattern for smn addressing on different AIDs:
289 *   bit[34]: indicate cross AID access
290 *   bit[33:32]: indicate target AID id
291 * AID id range is 0 ~ 3 as maximum AID number is 4.
292 */
293u64 aqua_vanjaram_encode_ext_smn_addressing(int ext_id)
294{
295	u64 ext_offset;
296
297	/* local routing and bit[34:32] will be zeros */
298	if (ext_id == 0)
299		return 0;
300
301	/* Initiated from host, accessing to all non-zero aids are cross traffic */
302	ext_offset = ((u64)(ext_id & 0x3) << 32) | (1ULL << 34);
303
304	return ext_offset;
305}
306
307static int aqua_vanjaram_query_partition_mode(struct amdgpu_xcp_mgr *xcp_mgr)
308{
309	enum amdgpu_gfx_partition mode = AMDGPU_UNKNOWN_COMPUTE_PARTITION_MODE;
310	struct amdgpu_device *adev = xcp_mgr->adev;
311
312	if (adev->nbio.funcs->get_compute_partition_mode)
313		mode = adev->nbio.funcs->get_compute_partition_mode(adev);
314
315	return mode;
316}
317
318static int __aqua_vanjaram_get_xcc_per_xcp(struct amdgpu_xcp_mgr *xcp_mgr, int mode)
319{
320	int num_xcc, num_xcc_per_xcp = 0;
321
322	num_xcc = NUM_XCC(xcp_mgr->adev->gfx.xcc_mask);
323
324	switch (mode) {
325	case AMDGPU_SPX_PARTITION_MODE:
326		num_xcc_per_xcp = num_xcc;
327		break;
328	case AMDGPU_DPX_PARTITION_MODE:
329		num_xcc_per_xcp = num_xcc / 2;
330		break;
331	case AMDGPU_TPX_PARTITION_MODE:
332		num_xcc_per_xcp = num_xcc / 3;
333		break;
334	case AMDGPU_QPX_PARTITION_MODE:
335		num_xcc_per_xcp = num_xcc / 4;
336		break;
337	case AMDGPU_CPX_PARTITION_MODE:
338		num_xcc_per_xcp = 1;
339		break;
340	}
341
342	return num_xcc_per_xcp;
343}
344
345static int __aqua_vanjaram_get_xcp_ip_info(struct amdgpu_xcp_mgr *xcp_mgr, int xcp_id,
346				    enum AMDGPU_XCP_IP_BLOCK ip_id,
347				    struct amdgpu_xcp_ip *ip)
348{
349	struct amdgpu_device *adev = xcp_mgr->adev;
350	int num_xcc_xcp, num_sdma_xcp, num_vcn_xcp;
351	int num_sdma, num_vcn;
352
353	num_sdma = adev->sdma.num_instances;
354	num_vcn = adev->vcn.num_vcn_inst;
355
356	switch (xcp_mgr->mode) {
357	case AMDGPU_SPX_PARTITION_MODE:
358		num_sdma_xcp = num_sdma;
359		num_vcn_xcp = num_vcn;
360		break;
361	case AMDGPU_DPX_PARTITION_MODE:
362		num_sdma_xcp = num_sdma / 2;
363		num_vcn_xcp = num_vcn / 2;
364		break;
365	case AMDGPU_TPX_PARTITION_MODE:
366		num_sdma_xcp = num_sdma / 3;
367		num_vcn_xcp = num_vcn / 3;
368		break;
369	case AMDGPU_QPX_PARTITION_MODE:
370		num_sdma_xcp = num_sdma / 4;
371		num_vcn_xcp = num_vcn / 4;
372		break;
373	case AMDGPU_CPX_PARTITION_MODE:
374		num_sdma_xcp = 2;
375		num_vcn_xcp = num_vcn ? 1 : 0;
376		break;
377	default:
378		return -EINVAL;
379	}
380
381	num_xcc_xcp = adev->gfx.num_xcc_per_xcp;
382
383	switch (ip_id) {
384	case AMDGPU_XCP_GFXHUB:
385		ip->inst_mask = XCP_INST_MASK(num_xcc_xcp, xcp_id);
386		ip->ip_funcs = &gfxhub_v1_2_xcp_funcs;
387		break;
388	case AMDGPU_XCP_GFX:
389		ip->inst_mask = XCP_INST_MASK(num_xcc_xcp, xcp_id);
390		ip->ip_funcs = &gfx_v9_4_3_xcp_funcs;
391		break;
392	case AMDGPU_XCP_SDMA:
393		ip->inst_mask = XCP_INST_MASK(num_sdma_xcp, xcp_id);
394		ip->ip_funcs = &sdma_v4_4_2_xcp_funcs;
395		break;
396	case AMDGPU_XCP_VCN:
397		ip->inst_mask = XCP_INST_MASK(num_vcn_xcp, xcp_id);
398		/* TODO : Assign IP funcs */
399		break;
400	default:
401		return -EINVAL;
402	}
403
404	ip->ip_id = ip_id;
405
406	return 0;
407}
408
409static enum amdgpu_gfx_partition
410__aqua_vanjaram_get_auto_mode(struct amdgpu_xcp_mgr *xcp_mgr)
411{
412	struct amdgpu_device *adev = xcp_mgr->adev;
413	int num_xcc;
414
415	num_xcc = NUM_XCC(xcp_mgr->adev->gfx.xcc_mask);
416
417	if (adev->gmc.num_mem_partitions == 1)
418		return AMDGPU_SPX_PARTITION_MODE;
419
420	if (adev->gmc.num_mem_partitions == num_xcc)
421		return AMDGPU_CPX_PARTITION_MODE;
422
423	if (adev->gmc.num_mem_partitions == num_xcc / 2)
424		return (adev->flags & AMD_IS_APU) ? AMDGPU_TPX_PARTITION_MODE :
425						    AMDGPU_QPX_PARTITION_MODE;
426
427	if (adev->gmc.num_mem_partitions == 2 && !(adev->flags & AMD_IS_APU))
428		return AMDGPU_DPX_PARTITION_MODE;
429
430	return AMDGPU_UNKNOWN_COMPUTE_PARTITION_MODE;
431}
432
433static bool __aqua_vanjaram_is_valid_mode(struct amdgpu_xcp_mgr *xcp_mgr,
434					  enum amdgpu_gfx_partition mode)
435{
436	struct amdgpu_device *adev = xcp_mgr->adev;
437	int num_xcc, num_xccs_per_xcp;
438
439	num_xcc = NUM_XCC(adev->gfx.xcc_mask);
440	switch (mode) {
441	case AMDGPU_SPX_PARTITION_MODE:
442		return adev->gmc.num_mem_partitions == 1 && num_xcc > 0;
443	case AMDGPU_DPX_PARTITION_MODE:
444		return adev->gmc.num_mem_partitions != 8 && (num_xcc % 4) == 0;
445	case AMDGPU_TPX_PARTITION_MODE:
446		return (adev->gmc.num_mem_partitions == 1 ||
447			adev->gmc.num_mem_partitions == 3) &&
448		       ((num_xcc % 3) == 0);
449	case AMDGPU_QPX_PARTITION_MODE:
450		num_xccs_per_xcp = num_xcc / 4;
451		return (adev->gmc.num_mem_partitions == 1 ||
452			adev->gmc.num_mem_partitions == 4) &&
453		       (num_xccs_per_xcp >= 2);
454	case AMDGPU_CPX_PARTITION_MODE:
455		return ((num_xcc > 1) &&
456		       (adev->gmc.num_mem_partitions == 1 || adev->gmc.num_mem_partitions == 4) &&
457		       (num_xcc % adev->gmc.num_mem_partitions) == 0);
458	default:
459		return false;
460	}
461
462	return false;
463}
464
465static int __aqua_vanjaram_pre_partition_switch(struct amdgpu_xcp_mgr *xcp_mgr, u32 flags)
466{
467	/* TODO:
468	 * Stop user queues and threads, and make sure GPU is empty of work.
469	 */
470
471	if (flags & AMDGPU_XCP_OPS_KFD)
472		amdgpu_amdkfd_device_fini_sw(xcp_mgr->adev);
473
474	return 0;
475}
476
477static int __aqua_vanjaram_post_partition_switch(struct amdgpu_xcp_mgr *xcp_mgr, u32 flags)
478{
479	int ret = 0;
480
481	if (flags & AMDGPU_XCP_OPS_KFD) {
482		amdgpu_amdkfd_device_probe(xcp_mgr->adev);
483		amdgpu_amdkfd_device_init(xcp_mgr->adev);
484		/* If KFD init failed, return failure */
485		if (!xcp_mgr->adev->kfd.init_complete)
486			ret = -EIO;
487	}
488
489	return ret;
490}
491
492static int aqua_vanjaram_switch_partition_mode(struct amdgpu_xcp_mgr *xcp_mgr,
493					       int mode, int *num_xcps)
494{
495	int num_xcc_per_xcp, num_xcc, ret;
496	struct amdgpu_device *adev;
497	u32 flags = 0;
498
499	adev = xcp_mgr->adev;
500	num_xcc = NUM_XCC(adev->gfx.xcc_mask);
501
502	if (mode == AMDGPU_AUTO_COMPUTE_PARTITION_MODE) {
503		mode = __aqua_vanjaram_get_auto_mode(xcp_mgr);
504	} else if (!__aqua_vanjaram_is_valid_mode(xcp_mgr, mode)) {
505		dev_err(adev->dev,
506			"Invalid compute partition mode requested, requested: %s, available memory partitions: %d",
507			amdgpu_gfx_compute_mode_desc(mode), adev->gmc.num_mem_partitions);
508		return -EINVAL;
509	}
510
511	if (adev->kfd.init_complete && !amdgpu_in_reset(adev))
512		flags |= AMDGPU_XCP_OPS_KFD;
513
514	if (flags & AMDGPU_XCP_OPS_KFD) {
515		ret = amdgpu_amdkfd_check_and_lock_kfd(adev);
516		if (ret)
517			goto out;
518	}
519
520	ret = __aqua_vanjaram_pre_partition_switch(xcp_mgr, flags);
521	if (ret)
522		goto unlock;
523
524	num_xcc_per_xcp = __aqua_vanjaram_get_xcc_per_xcp(xcp_mgr, mode);
525	if (adev->gfx.funcs->switch_partition_mode)
526		adev->gfx.funcs->switch_partition_mode(xcp_mgr->adev,
527						       num_xcc_per_xcp);
528
529	/* Init info about new xcps */
530	*num_xcps = num_xcc / num_xcc_per_xcp;
531	amdgpu_xcp_init(xcp_mgr, *num_xcps, mode);
532
533	ret = __aqua_vanjaram_post_partition_switch(xcp_mgr, flags);
534unlock:
535	if (flags & AMDGPU_XCP_OPS_KFD)
536		amdgpu_amdkfd_unlock_kfd(adev);
537out:
538	return ret;
539}
540
541static int __aqua_vanjaram_get_xcp_mem_id(struct amdgpu_device *adev,
542					  int xcc_id, uint8_t *mem_id)
543{
544	/* memory/spatial modes validation check is already done */
545	*mem_id = xcc_id / adev->gfx.num_xcc_per_xcp;
546	*mem_id /= adev->xcp_mgr->num_xcp_per_mem_partition;
547
548	return 0;
549}
550
551static int aqua_vanjaram_get_xcp_mem_id(struct amdgpu_xcp_mgr *xcp_mgr,
552					struct amdgpu_xcp *xcp, uint8_t *mem_id)
553{
554	struct amdgpu_numa_info numa_info;
555	struct amdgpu_device *adev;
556	uint32_t xcc_mask;
557	int r, i, xcc_id;
558
559	adev = xcp_mgr->adev;
560	/* TODO: BIOS is not returning the right info now
561	 * Check on this later
562	 */
563	/*
564	if (adev->gmc.gmc_funcs->query_mem_partition_mode)
565		mode = adev->gmc.gmc_funcs->query_mem_partition_mode(adev);
566	*/
567	if (adev->gmc.num_mem_partitions == 1) {
568		/* Only one range */
569		*mem_id = 0;
570		return 0;
571	}
572
573	r = amdgpu_xcp_get_inst_details(xcp, AMDGPU_XCP_GFX, &xcc_mask);
574	if (r || !xcc_mask)
575		return -EINVAL;
576
577	xcc_id = ffs(xcc_mask) - 1;
578	if (!adev->gmc.is_app_apu)
579		return __aqua_vanjaram_get_xcp_mem_id(adev, xcc_id, mem_id);
580
581	r = amdgpu_acpi_get_mem_info(adev, xcc_id, &numa_info);
582
583	if (r)
584		return r;
585
586	r = -EINVAL;
587	for (i = 0; i < adev->gmc.num_mem_partitions; ++i) {
588		if (adev->gmc.mem_partitions[i].numa.node == numa_info.nid) {
589			*mem_id = i;
590			r = 0;
591			break;
592		}
593	}
594
595	return r;
596}
597
598static int aqua_vanjaram_get_xcp_ip_details(struct amdgpu_xcp_mgr *xcp_mgr, int xcp_id,
599				     enum AMDGPU_XCP_IP_BLOCK ip_id,
600				     struct amdgpu_xcp_ip *ip)
601{
602	if (!ip)
603		return -EINVAL;
604
605	return __aqua_vanjaram_get_xcp_ip_info(xcp_mgr, xcp_id, ip_id, ip);
606}
607
608struct amdgpu_xcp_mgr_funcs aqua_vanjaram_xcp_funcs = {
609	.switch_partition_mode = &aqua_vanjaram_switch_partition_mode,
610	.query_partition_mode = &aqua_vanjaram_query_partition_mode,
611	.get_ip_details = &aqua_vanjaram_get_xcp_ip_details,
612	.get_xcp_mem_id = &aqua_vanjaram_get_xcp_mem_id,
613	.select_scheds = &aqua_vanjaram_select_scheds,
614	.update_partition_sched_list = &aqua_vanjaram_update_partition_sched_list
615};
616
617static int aqua_vanjaram_xcp_mgr_init(struct amdgpu_device *adev)
618{
619	int ret;
620
621	ret = amdgpu_xcp_mgr_init(adev, AMDGPU_UNKNOWN_COMPUTE_PARTITION_MODE, 1,
622				  &aqua_vanjaram_xcp_funcs);
623	if (ret)
624		return ret;
625
626	/* TODO: Default memory node affinity init */
627
628	return ret;
629}
630
631int aqua_vanjaram_init_soc_config(struct amdgpu_device *adev)
632{
633	u32 mask, inst_mask = adev->sdma.sdma_mask;
634	int ret, i;
635
636	/* generally 1 AID supports 4 instances */
637	adev->sdma.num_inst_per_aid = 4;
638	adev->sdma.num_instances = NUM_SDMA(adev->sdma.sdma_mask);
639
640	adev->aid_mask = i = 1;
641	inst_mask >>= adev->sdma.num_inst_per_aid;
642
643	for (mask = (1 << adev->sdma.num_inst_per_aid) - 1; inst_mask;
644	     inst_mask >>= adev->sdma.num_inst_per_aid, ++i) {
645		if ((inst_mask & mask) == mask)
646			adev->aid_mask |= (1 << i);
647	}
648
649	/* Harvest config is not used for aqua vanjaram. VCN and JPEGs will be
650	 * addressed based on logical instance ids.
651	 */
652	adev->vcn.harvest_config = 0;
653	adev->vcn.num_inst_per_aid = 1;
654	adev->vcn.num_vcn_inst = hweight32(adev->vcn.inst_mask);
655	adev->jpeg.harvest_config = 0;
656	adev->jpeg.num_inst_per_aid = 1;
657	adev->jpeg.num_jpeg_inst = hweight32(adev->jpeg.inst_mask);
658
659	ret = aqua_vanjaram_xcp_mgr_init(adev);
660	if (ret)
661		return ret;
662
663	aqua_vanjaram_ip_map_init(adev);
664
665	return 0;
666}
667
668static void aqua_read_smn(struct amdgpu_device *adev,
669			  struct amdgpu_smn_reg_data *regdata,
670			  uint64_t smn_addr)
671{
672	regdata->addr = smn_addr;
673	regdata->value = RREG32_PCIE(smn_addr);
674}
675
676struct aqua_reg_list {
677	uint64_t start_addr;
678	uint32_t num_regs;
679	uint32_t incrx;
680};
681
682#define DW_ADDR_INCR	4
683
684static void aqua_read_smn_ext(struct amdgpu_device *adev,
685			      struct amdgpu_smn_reg_data *regdata,
686			      uint64_t smn_addr, int i)
687{
688	regdata->addr =
689		smn_addr + adev->asic_funcs->encode_ext_smn_addressing(i);
690	regdata->value = RREG32_PCIE_EXT(regdata->addr);
691}
692
693#define smnreg_0x1A340218	0x1A340218
694#define smnreg_0x1A3402E4	0x1A3402E4
695#define smnreg_0x1A340294	0x1A340294
696#define smreg_0x1A380088	0x1A380088
697
698#define NUM_PCIE_SMN_REGS	14
699
700static struct aqua_reg_list pcie_reg_addrs[] = {
701	{ smnreg_0x1A340218, 1, 0 },
702	{ smnreg_0x1A3402E4, 1, 0 },
703	{ smnreg_0x1A340294, 6, DW_ADDR_INCR },
704	{ smreg_0x1A380088, 6, DW_ADDR_INCR },
705};
706
707static ssize_t aqua_vanjaram_read_pcie_state(struct amdgpu_device *adev,
708					     void *buf, size_t max_size)
709{
710	struct amdgpu_reg_state_pcie_v1_0 *pcie_reg_state;
711	uint32_t start_addr, incrx, num_regs, szbuf;
712	struct amdgpu_regs_pcie_v1_0 *pcie_regs;
713	struct amdgpu_smn_reg_data *reg_data;
714	struct pci_dev *us_pdev, *ds_pdev;
715	int aer_cap, r, n;
716
717	if (!buf || !max_size)
718		return -EINVAL;
719
720	pcie_reg_state = (struct amdgpu_reg_state_pcie_v1_0 *)buf;
721
722	szbuf = sizeof(*pcie_reg_state) +
723		amdgpu_reginst_size(1, sizeof(*pcie_regs), NUM_PCIE_SMN_REGS);
724	/* Only one instance of pcie regs */
725	if (max_size < szbuf)
726		return -EOVERFLOW;
727
728	pcie_regs = (struct amdgpu_regs_pcie_v1_0 *)((uint8_t *)buf +
729						     sizeof(*pcie_reg_state));
730	pcie_regs->inst_header.instance = 0;
731	pcie_regs->inst_header.state = AMDGPU_INST_S_OK;
732	pcie_regs->inst_header.num_smn_regs = NUM_PCIE_SMN_REGS;
733
734	reg_data = pcie_regs->smn_reg_values;
735
736	for (r = 0; r < ARRAY_SIZE(pcie_reg_addrs); r++) {
737		start_addr = pcie_reg_addrs[r].start_addr;
738		incrx = pcie_reg_addrs[r].incrx;
739		num_regs = pcie_reg_addrs[r].num_regs;
740		for (n = 0; n < num_regs; n++) {
741			aqua_read_smn(adev, reg_data, start_addr + n * incrx);
742			++reg_data;
743		}
744	}
745
746	ds_pdev = pci_upstream_bridge(adev->pdev);
747	us_pdev = pci_upstream_bridge(ds_pdev);
748
749	pcie_capability_read_word(us_pdev, PCI_EXP_DEVSTA,
750				  &pcie_regs->device_status);
751	pcie_capability_read_word(us_pdev, PCI_EXP_LNKSTA,
752				  &pcie_regs->link_status);
753
754	aer_cap = pci_find_ext_capability(us_pdev, PCI_EXT_CAP_ID_ERR);
755	if (aer_cap) {
756		pci_read_config_dword(us_pdev, aer_cap + PCI_ERR_COR_STATUS,
757				      &pcie_regs->pcie_corr_err_status);
758		pci_read_config_dword(us_pdev, aer_cap + PCI_ERR_UNCOR_STATUS,
759				      &pcie_regs->pcie_uncorr_err_status);
760	}
761
762	pci_read_config_dword(us_pdev, PCI_PRIMARY_BUS,
763			      &pcie_regs->sub_bus_number_latency);
764
765	pcie_reg_state->common_header.structure_size = szbuf;
766	pcie_reg_state->common_header.format_revision = 1;
767	pcie_reg_state->common_header.content_revision = 0;
768	pcie_reg_state->common_header.state_type = AMDGPU_REG_STATE_TYPE_PCIE;
769	pcie_reg_state->common_header.num_instances = 1;
770
771	return pcie_reg_state->common_header.structure_size;
772}
773
774#define smnreg_0x11A00050	0x11A00050
775#define smnreg_0x11A00180	0x11A00180
776#define smnreg_0x11A00070	0x11A00070
777#define smnreg_0x11A00200	0x11A00200
778#define smnreg_0x11A0020C	0x11A0020C
779#define smnreg_0x11A00210	0x11A00210
780#define smnreg_0x11A00108	0x11A00108
781
782#define XGMI_LINK_REG(smnreg, l) ((smnreg) | (l << 20))
783
784#define NUM_XGMI_SMN_REGS 25
785
786static struct aqua_reg_list xgmi_reg_addrs[] = {
787	{ smnreg_0x11A00050, 1, 0 },
788	{ smnreg_0x11A00180, 16, DW_ADDR_INCR },
789	{ smnreg_0x11A00070, 4, DW_ADDR_INCR },
790	{ smnreg_0x11A00200, 1, 0 },
791	{ smnreg_0x11A0020C, 1, 0 },
792	{ smnreg_0x11A00210, 1, 0 },
793	{ smnreg_0x11A00108, 1, 0 },
794};
795
796static ssize_t aqua_vanjaram_read_xgmi_state(struct amdgpu_device *adev,
797					     void *buf, size_t max_size)
798{
799	struct amdgpu_reg_state_xgmi_v1_0 *xgmi_reg_state;
800	uint32_t start_addr, incrx, num_regs, szbuf;
801	struct amdgpu_regs_xgmi_v1_0 *xgmi_regs;
802	struct amdgpu_smn_reg_data *reg_data;
803	const int max_xgmi_instances = 8;
804	int inst = 0, i, j, r, n;
805	const int xgmi_inst = 2;
806	void *p;
807
808	if (!buf || !max_size)
809		return -EINVAL;
810
811	xgmi_reg_state = (struct amdgpu_reg_state_xgmi_v1_0 *)buf;
812
813	szbuf = sizeof(*xgmi_reg_state) +
814		amdgpu_reginst_size(max_xgmi_instances, sizeof(*xgmi_regs),
815				    NUM_XGMI_SMN_REGS);
816	/* Only one instance of pcie regs */
817	if (max_size < szbuf)
818		return -EOVERFLOW;
819
820	p = &xgmi_reg_state->xgmi_state_regs[0];
821	for_each_inst(i, adev->aid_mask) {
822		for (j = 0; j < xgmi_inst; ++j) {
823			xgmi_regs = (struct amdgpu_regs_xgmi_v1_0 *)p;
824			xgmi_regs->inst_header.instance = inst++;
825
826			xgmi_regs->inst_header.state = AMDGPU_INST_S_OK;
827			xgmi_regs->inst_header.num_smn_regs = NUM_XGMI_SMN_REGS;
828
829			reg_data = xgmi_regs->smn_reg_values;
830
831			for (r = 0; r < ARRAY_SIZE(xgmi_reg_addrs); r++) {
832				start_addr = xgmi_reg_addrs[r].start_addr;
833				incrx = xgmi_reg_addrs[r].incrx;
834				num_regs = xgmi_reg_addrs[r].num_regs;
835
836				for (n = 0; n < num_regs; n++) {
837					aqua_read_smn_ext(
838						adev, reg_data,
839						XGMI_LINK_REG(start_addr, j) +
840							n * incrx,
841						i);
842					++reg_data;
843				}
844			}
845			p = reg_data;
846		}
847	}
848
849	xgmi_reg_state->common_header.structure_size = szbuf;
850	xgmi_reg_state->common_header.format_revision = 1;
851	xgmi_reg_state->common_header.content_revision = 0;
852	xgmi_reg_state->common_header.state_type = AMDGPU_REG_STATE_TYPE_XGMI;
853	xgmi_reg_state->common_header.num_instances = max_xgmi_instances;
854
855	return xgmi_reg_state->common_header.structure_size;
856}
857
858#define smnreg_0x11C00070	0x11C00070
859#define smnreg_0x11C00210	0x11C00210
860
861static struct aqua_reg_list wafl_reg_addrs[] = {
862	{ smnreg_0x11C00070, 4, DW_ADDR_INCR },
863	{ smnreg_0x11C00210, 1, 0 },
864};
865
866#define WAFL_LINK_REG(smnreg, l) ((smnreg) | (l << 20))
867
868#define NUM_WAFL_SMN_REGS 5
869
870static ssize_t aqua_vanjaram_read_wafl_state(struct amdgpu_device *adev,
871					     void *buf, size_t max_size)
872{
873	struct amdgpu_reg_state_wafl_v1_0 *wafl_reg_state;
874	uint32_t start_addr, incrx, num_regs, szbuf;
875	struct amdgpu_regs_wafl_v1_0 *wafl_regs;
876	struct amdgpu_smn_reg_data *reg_data;
877	const int max_wafl_instances = 8;
878	int inst = 0, i, j, r, n;
879	const int wafl_inst = 2;
880	void *p;
881
882	if (!buf || !max_size)
883		return -EINVAL;
884
885	wafl_reg_state = (struct amdgpu_reg_state_wafl_v1_0 *)buf;
886
887	szbuf = sizeof(*wafl_reg_state) +
888		amdgpu_reginst_size(max_wafl_instances, sizeof(*wafl_regs),
889				    NUM_WAFL_SMN_REGS);
890
891	if (max_size < szbuf)
892		return -EOVERFLOW;
893
894	p = &wafl_reg_state->wafl_state_regs[0];
895	for_each_inst(i, adev->aid_mask) {
896		for (j = 0; j < wafl_inst; ++j) {
897			wafl_regs = (struct amdgpu_regs_wafl_v1_0 *)p;
898			wafl_regs->inst_header.instance = inst++;
899
900			wafl_regs->inst_header.state = AMDGPU_INST_S_OK;
901			wafl_regs->inst_header.num_smn_regs = NUM_WAFL_SMN_REGS;
902
903			reg_data = wafl_regs->smn_reg_values;
904
905			for (r = 0; r < ARRAY_SIZE(wafl_reg_addrs); r++) {
906				start_addr = wafl_reg_addrs[r].start_addr;
907				incrx = wafl_reg_addrs[r].incrx;
908				num_regs = wafl_reg_addrs[r].num_regs;
909				for (n = 0; n < num_regs; n++) {
910					aqua_read_smn_ext(
911						adev, reg_data,
912						WAFL_LINK_REG(start_addr, j) +
913							n * incrx,
914						i);
915					++reg_data;
916				}
917			}
918			p = reg_data;
919		}
920	}
921
922	wafl_reg_state->common_header.structure_size = szbuf;
923	wafl_reg_state->common_header.format_revision = 1;
924	wafl_reg_state->common_header.content_revision = 0;
925	wafl_reg_state->common_header.state_type = AMDGPU_REG_STATE_TYPE_WAFL;
926	wafl_reg_state->common_header.num_instances = max_wafl_instances;
927
928	return wafl_reg_state->common_header.structure_size;
929}
930
931#define smnreg_0x1B311060 0x1B311060
932#define smnreg_0x1B411060 0x1B411060
933#define smnreg_0x1B511060 0x1B511060
934#define smnreg_0x1B611060 0x1B611060
935
936#define smnreg_0x1C307120 0x1C307120
937#define smnreg_0x1C317120 0x1C317120
938
939#define smnreg_0x1C320830 0x1C320830
940#define smnreg_0x1C380830 0x1C380830
941#define smnreg_0x1C3D0830 0x1C3D0830
942#define smnreg_0x1C420830 0x1C420830
943
944#define smnreg_0x1C320100 0x1C320100
945#define smnreg_0x1C380100 0x1C380100
946#define smnreg_0x1C3D0100 0x1C3D0100
947#define smnreg_0x1C420100 0x1C420100
948
949#define smnreg_0x1B310500 0x1B310500
950#define smnreg_0x1C300400 0x1C300400
951
952#define USR_CAKE_INCR 0x11000
953#define USR_LINK_INCR 0x100000
954#define USR_CP_INCR 0x10000
955
956#define NUM_USR_SMN_REGS	20
957
958struct aqua_reg_list usr_reg_addrs[] = {
959	{ smnreg_0x1B311060, 4, DW_ADDR_INCR },
960	{ smnreg_0x1B411060, 4, DW_ADDR_INCR },
961	{ smnreg_0x1B511060, 4, DW_ADDR_INCR },
962	{ smnreg_0x1B611060, 4, DW_ADDR_INCR },
963	{ smnreg_0x1C307120, 2, DW_ADDR_INCR },
964	{ smnreg_0x1C317120, 2, DW_ADDR_INCR },
965};
966
967#define NUM_USR1_SMN_REGS	46
968struct aqua_reg_list usr1_reg_addrs[] = {
969	{ smnreg_0x1C320830, 6, USR_CAKE_INCR },
970	{ smnreg_0x1C380830, 5, USR_CAKE_INCR },
971	{ smnreg_0x1C3D0830, 5, USR_CAKE_INCR },
972	{ smnreg_0x1C420830, 4, USR_CAKE_INCR },
973	{ smnreg_0x1C320100, 6, USR_CAKE_INCR },
974	{ smnreg_0x1C380100, 5, USR_CAKE_INCR },
975	{ smnreg_0x1C3D0100, 5, USR_CAKE_INCR },
976	{ smnreg_0x1C420100, 4, USR_CAKE_INCR },
977	{ smnreg_0x1B310500, 4, USR_LINK_INCR },
978	{ smnreg_0x1C300400, 2, USR_CP_INCR },
979};
980
981static ssize_t aqua_vanjaram_read_usr_state(struct amdgpu_device *adev,
982					    void *buf, size_t max_size,
983					    int reg_state)
984{
985	uint32_t start_addr, incrx, num_regs, szbuf, num_smn;
986	struct amdgpu_reg_state_usr_v1_0 *usr_reg_state;
987	struct amdgpu_regs_usr_v1_0 *usr_regs;
988	struct amdgpu_smn_reg_data *reg_data;
989	const int max_usr_instances = 4;
990	struct aqua_reg_list *reg_addrs;
991	int inst = 0, i, n, r, arr_size;
992	void *p;
993
994	if (!buf || !max_size)
995		return -EINVAL;
996
997	switch (reg_state) {
998	case AMDGPU_REG_STATE_TYPE_USR:
999		arr_size = ARRAY_SIZE(usr_reg_addrs);
1000		reg_addrs = usr_reg_addrs;
1001		num_smn = NUM_USR_SMN_REGS;
1002		break;
1003	case AMDGPU_REG_STATE_TYPE_USR_1:
1004		arr_size = ARRAY_SIZE(usr1_reg_addrs);
1005		reg_addrs = usr1_reg_addrs;
1006		num_smn = NUM_USR1_SMN_REGS;
1007		break;
1008	default:
1009		return -EINVAL;
1010	}
1011
1012	usr_reg_state = (struct amdgpu_reg_state_usr_v1_0 *)buf;
1013
1014	szbuf = sizeof(*usr_reg_state) + amdgpu_reginst_size(max_usr_instances,
1015							     sizeof(*usr_regs),
1016							     num_smn);
1017	if (max_size < szbuf)
1018		return -EOVERFLOW;
1019
1020	p = &usr_reg_state->usr_state_regs[0];
1021	for_each_inst(i, adev->aid_mask) {
1022		usr_regs = (struct amdgpu_regs_usr_v1_0 *)p;
1023		usr_regs->inst_header.instance = inst++;
1024		usr_regs->inst_header.state = AMDGPU_INST_S_OK;
1025		usr_regs->inst_header.num_smn_regs = num_smn;
1026		reg_data = usr_regs->smn_reg_values;
1027
1028		for (r = 0; r < arr_size; r++) {
1029			start_addr = reg_addrs[r].start_addr;
1030			incrx = reg_addrs[r].incrx;
1031			num_regs = reg_addrs[r].num_regs;
1032			for (n = 0; n < num_regs; n++) {
1033				aqua_read_smn_ext(adev, reg_data,
1034						  start_addr + n * incrx, i);
1035				reg_data++;
1036			}
1037		}
1038		p = reg_data;
1039	}
1040
1041	usr_reg_state->common_header.structure_size = szbuf;
1042	usr_reg_state->common_header.format_revision = 1;
1043	usr_reg_state->common_header.content_revision = 0;
1044	usr_reg_state->common_header.state_type = AMDGPU_REG_STATE_TYPE_USR;
1045	usr_reg_state->common_header.num_instances = max_usr_instances;
1046
1047	return usr_reg_state->common_header.structure_size;
1048}
1049
1050ssize_t aqua_vanjaram_get_reg_state(struct amdgpu_device *adev,
1051				    enum amdgpu_reg_state reg_state, void *buf,
1052				    size_t max_size)
1053{
1054	ssize_t size;
1055
1056	switch (reg_state) {
1057	case AMDGPU_REG_STATE_TYPE_PCIE:
1058		size = aqua_vanjaram_read_pcie_state(adev, buf, max_size);
1059		break;
1060	case AMDGPU_REG_STATE_TYPE_XGMI:
1061		size = aqua_vanjaram_read_xgmi_state(adev, buf, max_size);
1062		break;
1063	case AMDGPU_REG_STATE_TYPE_WAFL:
1064		size = aqua_vanjaram_read_wafl_state(adev, buf, max_size);
1065		break;
1066	case AMDGPU_REG_STATE_TYPE_USR:
1067		size = aqua_vanjaram_read_usr_state(adev, buf, max_size,
1068						    AMDGPU_REG_STATE_TYPE_USR);
1069		break;
1070	case AMDGPU_REG_STATE_TYPE_USR_1:
1071		size = aqua_vanjaram_read_usr_state(
1072			adev, buf, max_size, AMDGPU_REG_STATE_TYPE_USR_1);
1073		break;
1074	default:
1075		return -EINVAL;
1076	}
1077
1078	return size;
1079}
1080