1/*
2 * Copyright 2019 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#include <linux/module.h>
23#include <linux/fdtable.h>
24#include <linux/uaccess.h>
25#include <linux/firmware.h>
26#include "amdgpu.h"
27#include "amdgpu_amdkfd.h"
28#include "amdgpu_amdkfd_arcturus.h"
29#include "amdgpu_reset.h"
30#include "sdma0/sdma0_4_2_2_offset.h"
31#include "sdma0/sdma0_4_2_2_sh_mask.h"
32#include "sdma1/sdma1_4_2_2_offset.h"
33#include "sdma1/sdma1_4_2_2_sh_mask.h"
34#include "sdma2/sdma2_4_2_2_offset.h"
35#include "sdma2/sdma2_4_2_2_sh_mask.h"
36#include "sdma3/sdma3_4_2_2_offset.h"
37#include "sdma3/sdma3_4_2_2_sh_mask.h"
38#include "sdma4/sdma4_4_2_2_offset.h"
39#include "sdma4/sdma4_4_2_2_sh_mask.h"
40#include "sdma5/sdma5_4_2_2_offset.h"
41#include "sdma5/sdma5_4_2_2_sh_mask.h"
42#include "sdma6/sdma6_4_2_2_offset.h"
43#include "sdma6/sdma6_4_2_2_sh_mask.h"
44#include "sdma7/sdma7_4_2_2_offset.h"
45#include "sdma7/sdma7_4_2_2_sh_mask.h"
46#include "v9_structs.h"
47#include "soc15.h"
48#include "soc15d.h"
49#include "amdgpu_amdkfd_gfx_v9.h"
50#include "gfxhub_v1_0.h"
51#include "mmhub_v9_4.h"
52#include "gc/gc_9_0_offset.h"
53#include "gc/gc_9_0_sh_mask.h"
54
55#define HQD_N_REGS 56
56#define DUMP_REG(addr) do {				\
57		if (WARN_ON_ONCE(i >= HQD_N_REGS))	\
58			break;				\
59		(*dump)[i][0] = (addr) << 2;		\
60		(*dump)[i++][1] = RREG32(addr);		\
61	} while (0)
62
63static inline struct v9_sdma_mqd *get_sdma_mqd(void *mqd)
64{
65	return (struct v9_sdma_mqd *)mqd;
66}
67
68static uint32_t get_sdma_rlc_reg_offset(struct amdgpu_device *adev,
69				unsigned int engine_id,
70				unsigned int queue_id)
71{
72	uint32_t sdma_engine_reg_base = 0;
73	uint32_t sdma_rlc_reg_offset;
74
75	switch (engine_id) {
76	default:
77		dev_warn(adev->dev,
78			 "Invalid sdma engine id (%d), using engine id 0\n",
79			 engine_id);
80		fallthrough;
81	case 0:
82		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA0, 0,
83				mmSDMA0_RLC0_RB_CNTL) - mmSDMA0_RLC0_RB_CNTL;
84		break;
85	case 1:
86		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA1, 0,
87				mmSDMA1_RLC0_RB_CNTL) - mmSDMA1_RLC0_RB_CNTL;
88		break;
89	case 2:
90		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA2, 0,
91				mmSDMA2_RLC0_RB_CNTL) - mmSDMA2_RLC0_RB_CNTL;
92		break;
93	case 3:
94		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA3, 0,
95				mmSDMA3_RLC0_RB_CNTL) - mmSDMA3_RLC0_RB_CNTL;
96		break;
97	case 4:
98		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA4, 0,
99				mmSDMA4_RLC0_RB_CNTL) - mmSDMA4_RLC0_RB_CNTL;
100		break;
101	case 5:
102		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA5, 0,
103				mmSDMA5_RLC0_RB_CNTL) - mmSDMA5_RLC0_RB_CNTL;
104		break;
105	case 6:
106		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA6, 0,
107				mmSDMA6_RLC0_RB_CNTL) - mmSDMA6_RLC0_RB_CNTL;
108		break;
109	case 7:
110		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA7, 0,
111				mmSDMA7_RLC0_RB_CNTL) - mmSDMA7_RLC0_RB_CNTL;
112		break;
113	}
114
115	sdma_rlc_reg_offset = sdma_engine_reg_base
116		+ queue_id * (mmSDMA0_RLC1_RB_CNTL - mmSDMA0_RLC0_RB_CNTL);
117
118	pr_debug("RLC register offset for SDMA%d RLC%d: 0x%x\n", engine_id,
119			queue_id, sdma_rlc_reg_offset);
120
121	return sdma_rlc_reg_offset;
122}
123
124int kgd_arcturus_hqd_sdma_load(struct amdgpu_device *adev, void *mqd,
125			     uint32_t __user *wptr, struct mm_struct *mm)
126{
127	struct v9_sdma_mqd *m;
128	uint32_t sdma_rlc_reg_offset;
129	unsigned long end_jiffies;
130	uint32_t data;
131	uint64_t data64;
132	uint64_t __user *wptr64 = (uint64_t __user *)wptr;
133
134	m = get_sdma_mqd(mqd);
135	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
136					    m->sdma_queue_id);
137
138	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL,
139		m->sdmax_rlcx_rb_cntl & (~SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK));
140
141	end_jiffies = msecs_to_jiffies(2000) + jiffies;
142	while (true) {
143		data = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_CONTEXT_STATUS);
144		if (data & SDMA0_RLC0_CONTEXT_STATUS__IDLE_MASK)
145			break;
146		if (time_after(jiffies, end_jiffies)) {
147			pr_err("SDMA RLC not idle in %s\n", __func__);
148			return -ETIME;
149		}
150		usleep_range(500, 1000);
151	}
152
153	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL_OFFSET,
154	       m->sdmax_rlcx_doorbell_offset);
155
156	data = REG_SET_FIELD(m->sdmax_rlcx_doorbell, SDMA0_RLC0_DOORBELL,
157			     ENABLE, 1);
158	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL, data);
159	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR,
160				m->sdmax_rlcx_rb_rptr);
161	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_HI,
162				m->sdmax_rlcx_rb_rptr_hi);
163
164	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_MINOR_PTR_UPDATE, 1);
165	if (read_user_wptr(mm, wptr64, data64)) {
166		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR,
167		       lower_32_bits(data64));
168		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR_HI,
169		       upper_32_bits(data64));
170	} else {
171		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR,
172		       m->sdmax_rlcx_rb_rptr);
173		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR_HI,
174		       m->sdmax_rlcx_rb_rptr_hi);
175	}
176	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_MINOR_PTR_UPDATE, 0);
177
178	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_BASE, m->sdmax_rlcx_rb_base);
179	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_BASE_HI,
180			m->sdmax_rlcx_rb_base_hi);
181	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_ADDR_LO,
182			m->sdmax_rlcx_rb_rptr_addr_lo);
183	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_ADDR_HI,
184			m->sdmax_rlcx_rb_rptr_addr_hi);
185
186	data = REG_SET_FIELD(m->sdmax_rlcx_rb_cntl, SDMA0_RLC0_RB_CNTL,
187			     RB_ENABLE, 1);
188	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL, data);
189
190	return 0;
191}
192
193int kgd_arcturus_hqd_sdma_dump(struct amdgpu_device *adev,
194			     uint32_t engine_id, uint32_t queue_id,
195			     uint32_t (**dump)[2], uint32_t *n_regs)
196{
197	uint32_t sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev,
198			engine_id, queue_id);
199	uint32_t i = 0, reg;
200#undef HQD_N_REGS
201#define HQD_N_REGS (19+6+7+10)
202
203	*dump = kmalloc_array(HQD_N_REGS, sizeof(**dump), GFP_KERNEL);
204	if (*dump == NULL)
205		return -ENOMEM;
206
207	for (reg = mmSDMA0_RLC0_RB_CNTL; reg <= mmSDMA0_RLC0_DOORBELL; reg++)
208		DUMP_REG(sdma_rlc_reg_offset + reg);
209	for (reg = mmSDMA0_RLC0_STATUS; reg <= mmSDMA0_RLC0_CSA_ADDR_HI; reg++)
210		DUMP_REG(sdma_rlc_reg_offset + reg);
211	for (reg = mmSDMA0_RLC0_IB_SUB_REMAIN;
212	     reg <= mmSDMA0_RLC0_MINOR_PTR_UPDATE; reg++)
213		DUMP_REG(sdma_rlc_reg_offset + reg);
214	for (reg = mmSDMA0_RLC0_MIDCMD_DATA0;
215	     reg <= mmSDMA0_RLC0_MIDCMD_CNTL; reg++)
216		DUMP_REG(sdma_rlc_reg_offset + reg);
217
218	WARN_ON_ONCE(i != HQD_N_REGS);
219	*n_regs = i;
220
221	return 0;
222}
223
224bool kgd_arcturus_hqd_sdma_is_occupied(struct amdgpu_device *adev,
225				void *mqd)
226{
227	struct v9_sdma_mqd *m;
228	uint32_t sdma_rlc_reg_offset;
229	uint32_t sdma_rlc_rb_cntl;
230
231	m = get_sdma_mqd(mqd);
232	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
233					    m->sdma_queue_id);
234
235	sdma_rlc_rb_cntl = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL);
236
237	if (sdma_rlc_rb_cntl & SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK)
238		return true;
239
240	return false;
241}
242
243int kgd_arcturus_hqd_sdma_destroy(struct amdgpu_device *adev, void *mqd,
244				unsigned int utimeout)
245{
246	struct v9_sdma_mqd *m;
247	uint32_t sdma_rlc_reg_offset;
248	uint32_t temp;
249	unsigned long end_jiffies = (utimeout * HZ / 1000) + jiffies;
250
251	m = get_sdma_mqd(mqd);
252	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
253					    m->sdma_queue_id);
254
255	temp = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL);
256	temp = temp & ~SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK;
257	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL, temp);
258
259	while (true) {
260		temp = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_CONTEXT_STATUS);
261		if (temp & SDMA0_RLC0_CONTEXT_STATUS__IDLE_MASK)
262			break;
263		if (time_after(jiffies, end_jiffies)) {
264			pr_err("SDMA RLC not idle in %s\n", __func__);
265			return -ETIME;
266		}
267		usleep_range(500, 1000);
268	}
269
270	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL, 0);
271	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL,
272		RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL) |
273		SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK);
274
275	m->sdmax_rlcx_rb_rptr = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR);
276	m->sdmax_rlcx_rb_rptr_hi =
277		RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_HI);
278
279	return 0;
280}
281
282/*
283 * Helper used to suspend/resume gfx pipe for image post process work to set
284 * barrier behaviour.
285 */
286static int suspend_resume_compute_scheduler(struct amdgpu_device *adev, bool suspend)
287{
288	int i, r = 0;
289
290	for (i = 0; i < adev->gfx.num_compute_rings; i++) {
291		struct amdgpu_ring *ring = &adev->gfx.compute_ring[i];
292
293		if (!amdgpu_ring_sched_ready(ring))
294			continue;
295
296		/* stop secheduler and drain ring. */
297		if (suspend) {
298			drm_sched_stop(&ring->sched, NULL);
299			r = amdgpu_fence_wait_empty(ring);
300			if (r)
301				goto out;
302		} else {
303			drm_sched_start(&ring->sched, false);
304		}
305	}
306
307out:
308	/* return on resume or failure to drain rings. */
309	if (!suspend || r)
310		return r;
311
312	return amdgpu_device_ip_wait_for_idle(adev, AMD_IP_BLOCK_TYPE_GFX);
313}
314
315static void set_barrier_auto_waitcnt(struct amdgpu_device *adev, bool enable_waitcnt)
316{
317	uint32_t data;
318
319	WRITE_ONCE(adev->barrier_has_auto_waitcnt, enable_waitcnt);
320
321	if (!down_read_trylock(&adev->reset_domain->sem))
322		return;
323
324	amdgpu_amdkfd_suspend(adev, false);
325
326	if (suspend_resume_compute_scheduler(adev, true))
327		goto out;
328
329	data = RREG32(SOC15_REG_OFFSET(GC, 0, mmSQ_CONFIG));
330	data = REG_SET_FIELD(data, SQ_CONFIG, DISABLE_BARRIER_WAITCNT,
331						!enable_waitcnt);
332	WREG32(SOC15_REG_OFFSET(GC, 0, mmSQ_CONFIG), data);
333
334out:
335	suspend_resume_compute_scheduler(adev, false);
336
337	amdgpu_amdkfd_resume(adev, false);
338
339	up_read(&adev->reset_domain->sem);
340}
341
342/*
343 * restore_dbg_registers is ignored here but is a general interface requirement
344 * for devices that support GFXOFF and where the RLC save/restore list
345 * does not support hw registers for debugging i.e. the driver has to manually
346 * initialize the debug mode registers after it has disabled GFX off during the
347 * debug session.
348 */
349static uint32_t kgd_arcturus_enable_debug_trap(struct amdgpu_device *adev,
350				bool restore_dbg_registers,
351				uint32_t vmid)
352{
353	mutex_lock(&adev->grbm_idx_mutex);
354
355	kgd_gfx_v9_set_wave_launch_stall(adev, vmid, true);
356
357	set_barrier_auto_waitcnt(adev, true);
358
359	WREG32(SOC15_REG_OFFSET(GC, 0, mmSPI_GDBG_TRAP_MASK), 0);
360
361	kgd_gfx_v9_set_wave_launch_stall(adev, vmid, false);
362
363	mutex_unlock(&adev->grbm_idx_mutex);
364
365	return 0;
366}
367
368/*
369 * keep_trap_enabled is ignored here but is a general interface requirement
370 * for devices that support multi-process debugging where the performance
371 * overhead from trap temporary setup needs to be bypassed when the debug
372 * session has ended.
373 */
374static uint32_t kgd_arcturus_disable_debug_trap(struct amdgpu_device *adev,
375					bool keep_trap_enabled,
376					uint32_t vmid)
377{
378
379	mutex_lock(&adev->grbm_idx_mutex);
380
381	kgd_gfx_v9_set_wave_launch_stall(adev, vmid, true);
382
383	set_barrier_auto_waitcnt(adev, false);
384
385	WREG32(SOC15_REG_OFFSET(GC, 0, mmSPI_GDBG_TRAP_MASK), 0);
386
387	kgd_gfx_v9_set_wave_launch_stall(adev, vmid, false);
388
389	mutex_unlock(&adev->grbm_idx_mutex);
390
391	return 0;
392}
393const struct kfd2kgd_calls arcturus_kfd2kgd = {
394	.program_sh_mem_settings = kgd_gfx_v9_program_sh_mem_settings,
395	.set_pasid_vmid_mapping = kgd_gfx_v9_set_pasid_vmid_mapping,
396	.init_interrupts = kgd_gfx_v9_init_interrupts,
397	.hqd_load = kgd_gfx_v9_hqd_load,
398	.hiq_mqd_load = kgd_gfx_v9_hiq_mqd_load,
399	.hqd_sdma_load = kgd_arcturus_hqd_sdma_load,
400	.hqd_dump = kgd_gfx_v9_hqd_dump,
401	.hqd_sdma_dump = kgd_arcturus_hqd_sdma_dump,
402	.hqd_is_occupied = kgd_gfx_v9_hqd_is_occupied,
403	.hqd_sdma_is_occupied = kgd_arcturus_hqd_sdma_is_occupied,
404	.hqd_destroy = kgd_gfx_v9_hqd_destroy,
405	.hqd_sdma_destroy = kgd_arcturus_hqd_sdma_destroy,
406	.wave_control_execute = kgd_gfx_v9_wave_control_execute,
407	.get_atc_vmid_pasid_mapping_info =
408				kgd_gfx_v9_get_atc_vmid_pasid_mapping_info,
409	.set_vm_context_page_table_base =
410				kgd_gfx_v9_set_vm_context_page_table_base,
411	.enable_debug_trap = kgd_arcturus_enable_debug_trap,
412	.disable_debug_trap = kgd_arcturus_disable_debug_trap,
413	.validate_trap_override_request = kgd_gfx_v9_validate_trap_override_request,
414	.set_wave_launch_trap_override = kgd_gfx_v9_set_wave_launch_trap_override,
415	.set_wave_launch_mode = kgd_gfx_v9_set_wave_launch_mode,
416	.set_address_watch = kgd_gfx_v9_set_address_watch,
417	.clear_address_watch = kgd_gfx_v9_clear_address_watch,
418	.get_iq_wait_times = kgd_gfx_v9_get_iq_wait_times,
419	.build_grace_period_packet_info = kgd_gfx_v9_build_grace_period_packet_info,
420	.get_cu_occupancy = kgd_gfx_v9_get_cu_occupancy,
421	.program_trap_handler_settings = kgd_gfx_v9_program_trap_handler_settings
422};
423