Linux Audio

Check our new training course

Loading...
Note: File does not exist in v4.17.
  1// SPDX-License-Identifier: MIT
  2/*
  3 * Copyright 2023 Advanced Micro Devices, Inc.
  4 *
  5 * Permission is hereby granted, free of charge, to any person obtaining a
  6 * copy of this software and associated documentation files (the "Software"),
  7 * to deal in the Software without restriction, including without limitation
  8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
  9 * and/or sell copies of the Software, and to permit persons to whom the
 10 * Software is furnished to do so, subject to the following conditions:
 11 *
 12 * The above copyright notice and this permission notice shall be included in
 13 * all copies or substantial portions of the Software.
 14 *
 15 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 16 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 17 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 18 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 19 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 20 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 21 * OTHER DEALINGS IN THE SOFTWARE.
 22 *
 23 */
 24
 25#include <linux/firmware.h>
 26#include <drm/drm_exec.h>
 27
 28#include "amdgpu.h"
 29#include "amdgpu_umsch_mm.h"
 30#include "umsch_mm_v4_0.h"
 31
 32struct umsch_mm_test_ctx_data {
 33	uint8_t process_csa[PAGE_SIZE];
 34	uint8_t vpe_ctx_csa[PAGE_SIZE];
 35	uint8_t vcn_ctx_csa[PAGE_SIZE];
 36};
 37
 38struct umsch_mm_test_mqd_data {
 39	uint8_t vpe_mqd[PAGE_SIZE];
 40	uint8_t vcn_mqd[PAGE_SIZE];
 41};
 42
 43struct umsch_mm_test_ring_data {
 44	uint8_t vpe_ring[PAGE_SIZE];
 45	uint8_t vpe_ib[PAGE_SIZE];
 46	uint8_t vcn_ring[PAGE_SIZE];
 47	uint8_t vcn_ib[PAGE_SIZE];
 48};
 49
 50struct umsch_mm_test_queue_info {
 51	uint64_t mqd_addr;
 52	uint64_t csa_addr;
 53	uint32_t doorbell_offset_0;
 54	uint32_t doorbell_offset_1;
 55	enum UMSCH_SWIP_ENGINE_TYPE engine;
 56};
 57
 58struct umsch_mm_test {
 59	struct amdgpu_bo	*ctx_data_obj;
 60	uint64_t		ctx_data_gpu_addr;
 61	uint32_t		*ctx_data_cpu_addr;
 62
 63	struct amdgpu_bo	*mqd_data_obj;
 64	uint64_t		mqd_data_gpu_addr;
 65	uint32_t		*mqd_data_cpu_addr;
 66
 67	struct amdgpu_bo	*ring_data_obj;
 68	uint64_t		ring_data_gpu_addr;
 69	uint32_t		*ring_data_cpu_addr;
 70
 71
 72	struct amdgpu_vm	*vm;
 73	struct amdgpu_bo_va	*bo_va;
 74	uint32_t		pasid;
 75	uint32_t		vm_cntx_cntl;
 76	uint32_t		num_queues;
 77};
 78
 79static int map_ring_data(struct amdgpu_device *adev, struct amdgpu_vm *vm,
 80			  struct amdgpu_bo *bo, struct amdgpu_bo_va **bo_va,
 81			  uint64_t addr, uint32_t size)
 82{
 83	struct amdgpu_sync sync;
 84	struct drm_exec exec;
 85	int r;
 86
 87	amdgpu_sync_create(&sync);
 88
 89	drm_exec_init(&exec, 0, 0);
 90	drm_exec_until_all_locked(&exec) {
 91		r = drm_exec_lock_obj(&exec, &bo->tbo.base);
 92		drm_exec_retry_on_contention(&exec);
 93		if (unlikely(r))
 94			goto error_fini_exec;
 95
 96		r = amdgpu_vm_lock_pd(vm, &exec, 0);
 97		drm_exec_retry_on_contention(&exec);
 98		if (unlikely(r))
 99			goto error_fini_exec;
100	}
101
102	*bo_va = amdgpu_vm_bo_add(adev, vm, bo);
103	if (!*bo_va) {
104		r = -ENOMEM;
105		goto error_fini_exec;
106	}
107
108	r = amdgpu_vm_bo_map(adev, *bo_va, addr, 0, size,
109			     AMDGPU_PTE_READABLE | AMDGPU_PTE_WRITEABLE |
110			     AMDGPU_PTE_EXECUTABLE);
111
112	if (r)
113		goto error_del_bo_va;
114
115
116	r = amdgpu_vm_bo_update(adev, *bo_va, false);
117	if (r)
118		goto error_del_bo_va;
119
120	amdgpu_sync_fence(&sync, (*bo_va)->last_pt_update);
121
122	r = amdgpu_vm_update_pdes(adev, vm, false);
123	if (r)
124		goto error_del_bo_va;
125
126	amdgpu_sync_fence(&sync, vm->last_update);
127
128	amdgpu_sync_wait(&sync, false);
129	drm_exec_fini(&exec);
130
131	amdgpu_sync_free(&sync);
132
133	return 0;
134
135error_del_bo_va:
136	amdgpu_vm_bo_del(adev, *bo_va);
137	amdgpu_sync_free(&sync);
138
139error_fini_exec:
140	drm_exec_fini(&exec);
141	amdgpu_sync_free(&sync);
142	return r;
143}
144
145static int unmap_ring_data(struct amdgpu_device *adev, struct amdgpu_vm *vm,
146			    struct amdgpu_bo *bo, struct amdgpu_bo_va *bo_va,
147			    uint64_t addr)
148{
149	struct drm_exec exec;
150	long r;
151
152	drm_exec_init(&exec, 0, 0);
153	drm_exec_until_all_locked(&exec) {
154		r = drm_exec_lock_obj(&exec, &bo->tbo.base);
155		drm_exec_retry_on_contention(&exec);
156		if (unlikely(r))
157			goto out_unlock;
158
159		r = amdgpu_vm_lock_pd(vm, &exec, 0);
160		drm_exec_retry_on_contention(&exec);
161		if (unlikely(r))
162			goto out_unlock;
163	}
164
165
166	r = amdgpu_vm_bo_unmap(adev, bo_va, addr);
167	if (r)
168		goto out_unlock;
169
170	amdgpu_vm_bo_del(adev, bo_va);
171
172out_unlock:
173	drm_exec_fini(&exec);
174
175	return r;
176}
177
178static void setup_vpe_queue(struct amdgpu_device *adev,
179			    struct umsch_mm_test *test,
180			    struct umsch_mm_test_queue_info *qinfo)
181{
182	struct MQD_INFO *mqd = (struct MQD_INFO *)test->mqd_data_cpu_addr;
183	uint64_t ring_gpu_addr = test->ring_data_gpu_addr;
184
185	mqd->rb_base_lo = (ring_gpu_addr >> 8);
186	mqd->rb_base_hi = (ring_gpu_addr >> 40);
187	mqd->rb_size = PAGE_SIZE / 4;
188	mqd->wptr_val = 0;
189	mqd->rptr_val = 0;
190	mqd->unmapped = 1;
191
192	if (adev->vpe.collaborate_mode)
193		memcpy(++mqd, test->mqd_data_cpu_addr, sizeof(struct MQD_INFO));
194
195	qinfo->mqd_addr = test->mqd_data_gpu_addr;
196	qinfo->csa_addr = test->ctx_data_gpu_addr +
197		offsetof(struct umsch_mm_test_ctx_data, vpe_ctx_csa);
198	qinfo->doorbell_offset_0 = 0;
199	qinfo->doorbell_offset_1 = 0;
200}
201
202static void setup_vcn_queue(struct amdgpu_device *adev,
203			    struct umsch_mm_test *test,
204			    struct umsch_mm_test_queue_info *qinfo)
205{
206}
207
208static int add_test_queue(struct amdgpu_device *adev,
209			  struct umsch_mm_test *test,
210			  struct umsch_mm_test_queue_info *qinfo)
211{
212	struct umsch_mm_add_queue_input queue_input = {};
213	int r;
214
215	queue_input.process_id = test->pasid;
216	queue_input.page_table_base_addr = amdgpu_gmc_pd_addr(test->vm->root.bo);
217
218	queue_input.process_va_start = 0;
219	queue_input.process_va_end = (adev->vm_manager.max_pfn - 1) << AMDGPU_GPU_PAGE_SHIFT;
220
221	queue_input.process_quantum = 100000; /* 10ms */
222	queue_input.process_csa_addr = test->ctx_data_gpu_addr +
223				       offsetof(struct umsch_mm_test_ctx_data, process_csa);
224
225	queue_input.context_quantum = 10000; /* 1ms */
226	queue_input.context_csa_addr = qinfo->csa_addr;
227
228	queue_input.inprocess_context_priority = CONTEXT_PRIORITY_LEVEL_NORMAL;
229	queue_input.context_global_priority_level = CONTEXT_PRIORITY_LEVEL_NORMAL;
230	queue_input.doorbell_offset_0 = qinfo->doorbell_offset_0;
231	queue_input.doorbell_offset_1 = qinfo->doorbell_offset_1;
232
233	queue_input.engine_type = qinfo->engine;
234	queue_input.mqd_addr = qinfo->mqd_addr;
235	queue_input.vm_context_cntl = test->vm_cntx_cntl;
236
237	amdgpu_umsch_mm_lock(&adev->umsch_mm);
238	r = adev->umsch_mm.funcs->add_queue(&adev->umsch_mm, &queue_input);
239	amdgpu_umsch_mm_unlock(&adev->umsch_mm);
240	if (r)
241		return r;
242
243	return 0;
244}
245
246static int remove_test_queue(struct amdgpu_device *adev,
247			     struct umsch_mm_test *test,
248			     struct umsch_mm_test_queue_info *qinfo)
249{
250	struct umsch_mm_remove_queue_input queue_input = {};
251	int r;
252
253	queue_input.doorbell_offset_0 = qinfo->doorbell_offset_0;
254	queue_input.doorbell_offset_1 = qinfo->doorbell_offset_1;
255	queue_input.context_csa_addr = qinfo->csa_addr;
256
257	amdgpu_umsch_mm_lock(&adev->umsch_mm);
258	r = adev->umsch_mm.funcs->remove_queue(&adev->umsch_mm, &queue_input);
259	amdgpu_umsch_mm_unlock(&adev->umsch_mm);
260	if (r)
261		return r;
262
263	return 0;
264}
265
266static int submit_vpe_queue(struct amdgpu_device *adev, struct umsch_mm_test *test)
267{
268	struct MQD_INFO *mqd = (struct MQD_INFO *)test->mqd_data_cpu_addr;
269	uint32_t *ring = test->ring_data_cpu_addr +
270		offsetof(struct umsch_mm_test_ring_data, vpe_ring) / 4;
271	uint32_t *ib = test->ring_data_cpu_addr +
272		offsetof(struct umsch_mm_test_ring_data, vpe_ib) / 4;
273	uint64_t ib_gpu_addr = test->ring_data_gpu_addr +
274		offsetof(struct umsch_mm_test_ring_data, vpe_ib);
275	uint32_t *fence = ib + 2048 / 4;
276	uint64_t fence_gpu_addr = ib_gpu_addr + 2048;
277	const uint32_t test_pattern = 0xdeadbeef;
278	int i;
279
280	ib[0] = VPE_CMD_HEADER(VPE_CMD_OPCODE_FENCE, 0);
281	ib[1] = lower_32_bits(fence_gpu_addr);
282	ib[2] = upper_32_bits(fence_gpu_addr);
283	ib[3] = test_pattern;
284
285	ring[0] = VPE_CMD_HEADER(VPE_CMD_OPCODE_INDIRECT, 0);
286	ring[1] = (ib_gpu_addr & 0xffffffe0);
287	ring[2] = upper_32_bits(ib_gpu_addr);
288	ring[3] = 4;
289	ring[4] = 0;
290	ring[5] = 0;
291
292	mqd->wptr_val = (6 << 2);
293	if (adev->vpe.collaborate_mode)
294		(++mqd)->wptr_val = (6 << 2);
295
296	WDOORBELL32(adev->umsch_mm.agdb_index[CONTEXT_PRIORITY_LEVEL_NORMAL], mqd->wptr_val);
297
298	for (i = 0; i < adev->usec_timeout; i++) {
299		if (*fence == test_pattern)
300			return 0;
301		udelay(1);
302	}
303
304	dev_err(adev->dev, "vpe queue submission timeout\n");
305
306	return -ETIMEDOUT;
307}
308
309static int submit_vcn_queue(struct amdgpu_device *adev, struct umsch_mm_test *test)
310{
311	return 0;
312}
313
314static int setup_umsch_mm_test(struct amdgpu_device *adev,
315			  struct umsch_mm_test *test)
316{
317	struct amdgpu_vmhub *hub = &adev->vmhub[AMDGPU_MMHUB0(0)];
318	int r;
319
320	test->vm_cntx_cntl = hub->vm_cntx_cntl;
321
322	test->vm = kzalloc(sizeof(*test->vm), GFP_KERNEL);
323	if (!test->vm) {
324		r = -ENOMEM;
325		return r;
326	}
327
328	r = amdgpu_vm_init(adev, test->vm, -1);
329	if (r)
330		goto error_free_vm;
331
332	r = amdgpu_pasid_alloc(16);
333	if (r < 0)
334		goto error_fini_vm;
335	test->pasid = r;
336
337	r = amdgpu_bo_create_kernel(adev, sizeof(struct umsch_mm_test_ctx_data),
338				    PAGE_SIZE, AMDGPU_GEM_DOMAIN_GTT,
339				    &test->ctx_data_obj,
340				    &test->ctx_data_gpu_addr,
341				    (void **)&test->ctx_data_cpu_addr);
342	if (r)
343		goto error_free_pasid;
344
345	memset(test->ctx_data_cpu_addr, 0, sizeof(struct umsch_mm_test_ctx_data));
346
347	r = amdgpu_bo_create_kernel(adev, PAGE_SIZE,
348				    PAGE_SIZE, AMDGPU_GEM_DOMAIN_GTT,
349				    &test->mqd_data_obj,
350				    &test->mqd_data_gpu_addr,
351				    (void **)&test->mqd_data_cpu_addr);
352	if (r)
353		goto error_free_ctx_data_obj;
354
355	memset(test->mqd_data_cpu_addr, 0, PAGE_SIZE);
356
357	r = amdgpu_bo_create_kernel(adev, sizeof(struct umsch_mm_test_ring_data),
358				    PAGE_SIZE, AMDGPU_GEM_DOMAIN_GTT,
359				    &test->ring_data_obj,
360				    NULL,
361				    (void **)&test->ring_data_cpu_addr);
362	if (r)
363		goto error_free_mqd_data_obj;
364
365	memset(test->ring_data_cpu_addr, 0, sizeof(struct umsch_mm_test_ring_data));
366
367	test->ring_data_gpu_addr = AMDGPU_VA_RESERVED_BOTTOM;
368	r = map_ring_data(adev, test->vm, test->ring_data_obj, &test->bo_va,
369			  test->ring_data_gpu_addr, sizeof(struct umsch_mm_test_ring_data));
370	if (r)
371		goto error_free_ring_data_obj;
372
373	return 0;
374
375error_free_ring_data_obj:
376	amdgpu_bo_free_kernel(&test->ring_data_obj, NULL,
377			      (void **)&test->ring_data_cpu_addr);
378error_free_mqd_data_obj:
379	amdgpu_bo_free_kernel(&test->mqd_data_obj, &test->mqd_data_gpu_addr,
380			      (void **)&test->mqd_data_cpu_addr);
381error_free_ctx_data_obj:
382	amdgpu_bo_free_kernel(&test->ctx_data_obj, &test->ctx_data_gpu_addr,
383			      (void **)&test->ctx_data_cpu_addr);
384error_free_pasid:
385	amdgpu_pasid_free(test->pasid);
386error_fini_vm:
387	amdgpu_vm_fini(adev, test->vm);
388error_free_vm:
389	kfree(test->vm);
390
391	return r;
392}
393
394static void cleanup_umsch_mm_test(struct amdgpu_device *adev,
395				  struct umsch_mm_test *test)
396{
397	unmap_ring_data(adev, test->vm, test->ring_data_obj,
398			test->bo_va, test->ring_data_gpu_addr);
399	amdgpu_bo_free_kernel(&test->mqd_data_obj, &test->mqd_data_gpu_addr,
400			      (void **)&test->mqd_data_cpu_addr);
401	amdgpu_bo_free_kernel(&test->ring_data_obj, NULL,
402			      (void **)&test->ring_data_cpu_addr);
403	amdgpu_bo_free_kernel(&test->ctx_data_obj, &test->ctx_data_gpu_addr,
404			       (void **)&test->ctx_data_cpu_addr);
405	amdgpu_pasid_free(test->pasid);
406	amdgpu_vm_fini(adev, test->vm);
407	kfree(test->vm);
408}
409
410static int setup_test_queues(struct amdgpu_device *adev,
411			     struct umsch_mm_test *test,
412			     struct umsch_mm_test_queue_info *qinfo)
413{
414	int i, r;
415
416	for (i = 0; i < test->num_queues; i++) {
417		if (qinfo[i].engine == UMSCH_SWIP_ENGINE_TYPE_VPE)
418			setup_vpe_queue(adev, test, &qinfo[i]);
419		else
420			setup_vcn_queue(adev, test, &qinfo[i]);
421
422		r = add_test_queue(adev, test, &qinfo[i]);
423		if (r)
424			return r;
425	}
426
427	return 0;
428}
429
430static int submit_test_queues(struct amdgpu_device *adev,
431			      struct umsch_mm_test *test,
432			      struct umsch_mm_test_queue_info *qinfo)
433{
434	int i, r;
435
436	for (i = 0; i < test->num_queues; i++) {
437		if (qinfo[i].engine == UMSCH_SWIP_ENGINE_TYPE_VPE)
438			r = submit_vpe_queue(adev, test);
439		else
440			r = submit_vcn_queue(adev, test);
441		if (r)
442			return r;
443	}
444
445	return 0;
446}
447
448static void cleanup_test_queues(struct amdgpu_device *adev,
449			      struct umsch_mm_test *test,
450			      struct umsch_mm_test_queue_info *qinfo)
451{
452	int i;
453
454	for (i = 0; i < test->num_queues; i++)
455		remove_test_queue(adev, test, &qinfo[i]);
456}
457
458static int umsch_mm_test(struct amdgpu_device *adev)
459{
460	struct umsch_mm_test_queue_info qinfo[] = {
461		{ .engine = UMSCH_SWIP_ENGINE_TYPE_VPE },
462	};
463	struct umsch_mm_test test = { .num_queues = ARRAY_SIZE(qinfo) };
464	int r;
465
466	r = setup_umsch_mm_test(adev, &test);
467	if (r)
468		return r;
469
470	r = setup_test_queues(adev, &test, qinfo);
471	if (r)
472		goto cleanup;
473
474	r = submit_test_queues(adev, &test, qinfo);
475	if (r)
476		goto cleanup;
477
478	cleanup_test_queues(adev, &test, qinfo);
479	cleanup_umsch_mm_test(adev, &test);
480
481	return 0;
482
483cleanup:
484	cleanup_test_queues(adev, &test, qinfo);
485	cleanup_umsch_mm_test(adev, &test);
486	return r;
487}
488
489int amdgpu_umsch_mm_submit_pkt(struct amdgpu_umsch_mm *umsch, void *pkt, int ndws)
490{
491	struct amdgpu_ring *ring = &umsch->ring;
492
493	if (amdgpu_ring_alloc(ring, ndws))
494		return -ENOMEM;
495
496	amdgpu_ring_write_multiple(ring, pkt, ndws);
497	amdgpu_ring_commit(ring);
498
499	return 0;
500}
501
502int amdgpu_umsch_mm_query_fence(struct amdgpu_umsch_mm *umsch)
503{
504	struct amdgpu_ring *ring = &umsch->ring;
505	struct amdgpu_device *adev = ring->adev;
506	int r;
507
508	r = amdgpu_fence_wait_polling(ring, ring->fence_drv.sync_seq, adev->usec_timeout);
509	if (r < 1) {
510		dev_err(adev->dev, "ring umsch timeout, emitted fence %u\n",
511			ring->fence_drv.sync_seq);
512		return -ETIMEDOUT;
513	}
514
515	return 0;
516}
517
518static void umsch_mm_ring_set_wptr(struct amdgpu_ring *ring)
519{
520	struct amdgpu_umsch_mm *umsch = (struct amdgpu_umsch_mm *)ring;
521	struct amdgpu_device *adev = ring->adev;
522
523	if (ring->use_doorbell)
524		WDOORBELL32(ring->doorbell_index, ring->wptr << 2);
525	else
526		WREG32(umsch->rb_wptr, ring->wptr << 2);
527}
528
529static u64 umsch_mm_ring_get_rptr(struct amdgpu_ring *ring)
530{
531	struct amdgpu_umsch_mm *umsch = (struct amdgpu_umsch_mm *)ring;
532	struct amdgpu_device *adev = ring->adev;
533
534	return RREG32(umsch->rb_rptr);
535}
536
537static u64 umsch_mm_ring_get_wptr(struct amdgpu_ring *ring)
538{
539	struct amdgpu_umsch_mm *umsch = (struct amdgpu_umsch_mm *)ring;
540	struct amdgpu_device *adev = ring->adev;
541
542	return RREG32(umsch->rb_wptr);
543}
544
545static const struct amdgpu_ring_funcs umsch_v4_0_ring_funcs = {
546	.type = AMDGPU_RING_TYPE_UMSCH_MM,
547	.align_mask = 0,
548	.nop = 0,
549	.support_64bit_ptrs = false,
550	.get_rptr = umsch_mm_ring_get_rptr,
551	.get_wptr = umsch_mm_ring_get_wptr,
552	.set_wptr = umsch_mm_ring_set_wptr,
553	.insert_nop = amdgpu_ring_insert_nop,
554};
555
556int amdgpu_umsch_mm_ring_init(struct amdgpu_umsch_mm *umsch)
557{
558	struct amdgpu_device *adev = container_of(umsch, struct amdgpu_device, umsch_mm);
559	struct amdgpu_ring *ring = &umsch->ring;
560
561	ring->vm_hub = AMDGPU_MMHUB0(0);
562	ring->use_doorbell = true;
563	ring->no_scheduler = true;
564	ring->doorbell_index = (AMDGPU_NAVI10_DOORBELL64_VCN0_1 << 1) + 6;
565
566	snprintf(ring->name, sizeof(ring->name), "umsch");
567
568	return amdgpu_ring_init(adev, ring, 1024, NULL, 0, AMDGPU_RING_PRIO_DEFAULT, NULL);
569}
570
571int amdgpu_umsch_mm_init_microcode(struct amdgpu_umsch_mm *umsch)
572{
573	const struct umsch_mm_firmware_header_v1_0 *umsch_mm_hdr;
574	struct amdgpu_device *adev = umsch->ring.adev;
575	const char *fw_name = NULL;
576	int r;
577
578	switch (amdgpu_ip_version(adev, VCN_HWIP, 0)) {
579	case IP_VERSION(4, 0, 5):
580	case IP_VERSION(4, 0, 6):
581		fw_name = "amdgpu/umsch_mm_4_0_0.bin";
582		break;
583	default:
584		break;
585	}
586
587	r = amdgpu_ucode_request(adev, &adev->umsch_mm.fw, fw_name);
588	if (r) {
589		release_firmware(adev->umsch_mm.fw);
590		adev->umsch_mm.fw = NULL;
591		return r;
592	}
593
594	umsch_mm_hdr = (const struct umsch_mm_firmware_header_v1_0 *)adev->umsch_mm.fw->data;
595
596	adev->umsch_mm.ucode_size = le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_size_bytes);
597	adev->umsch_mm.data_size = le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_data_size_bytes);
598
599	adev->umsch_mm.irq_start_addr =
600		le32_to_cpu(umsch_mm_hdr->umsch_mm_irq_start_addr_lo) |
601		((uint64_t)(le32_to_cpu(umsch_mm_hdr->umsch_mm_irq_start_addr_hi)) << 32);
602	adev->umsch_mm.uc_start_addr =
603		le32_to_cpu(umsch_mm_hdr->umsch_mm_uc_start_addr_lo) |
604		((uint64_t)(le32_to_cpu(umsch_mm_hdr->umsch_mm_uc_start_addr_hi)) << 32);
605	adev->umsch_mm.data_start_addr =
606		le32_to_cpu(umsch_mm_hdr->umsch_mm_data_start_addr_lo) |
607		((uint64_t)(le32_to_cpu(umsch_mm_hdr->umsch_mm_data_start_addr_hi)) << 32);
608
609	if (adev->firmware.load_type == AMDGPU_FW_LOAD_PSP) {
610		struct amdgpu_firmware_info *info;
611
612		info = &adev->firmware.ucode[AMDGPU_UCODE_ID_UMSCH_MM_UCODE];
613		info->ucode_id = AMDGPU_UCODE_ID_UMSCH_MM_UCODE;
614		info->fw = adev->umsch_mm.fw;
615		adev->firmware.fw_size +=
616			ALIGN(le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_size_bytes), PAGE_SIZE);
617
618		info = &adev->firmware.ucode[AMDGPU_UCODE_ID_UMSCH_MM_DATA];
619		info->ucode_id = AMDGPU_UCODE_ID_UMSCH_MM_DATA;
620		info->fw = adev->umsch_mm.fw;
621		adev->firmware.fw_size +=
622			ALIGN(le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_data_size_bytes), PAGE_SIZE);
623	}
624
625	return 0;
626}
627
628int amdgpu_umsch_mm_allocate_ucode_buffer(struct amdgpu_umsch_mm *umsch)
629{
630	const struct umsch_mm_firmware_header_v1_0 *umsch_mm_hdr;
631	struct amdgpu_device *adev = umsch->ring.adev;
632	const __le32 *fw_data;
633	uint32_t fw_size;
634	int r;
635
636	umsch_mm_hdr = (const struct umsch_mm_firmware_header_v1_0 *)
637		       adev->umsch_mm.fw->data;
638
639	fw_data = (const __le32 *)(adev->umsch_mm.fw->data +
640		  le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_offset_bytes));
641	fw_size = le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_size_bytes);
642
643	r = amdgpu_bo_create_reserved(adev, fw_size,
644				      4 * 1024, AMDGPU_GEM_DOMAIN_VRAM,
645				      &adev->umsch_mm.ucode_fw_obj,
646				      &adev->umsch_mm.ucode_fw_gpu_addr,
647				      (void **)&adev->umsch_mm.ucode_fw_ptr);
648	if (r) {
649		dev_err(adev->dev, "(%d) failed to create umsch_mm fw ucode bo\n", r);
650		return r;
651	}
652
653	memcpy(adev->umsch_mm.ucode_fw_ptr, fw_data, fw_size);
654
655	amdgpu_bo_kunmap(adev->umsch_mm.ucode_fw_obj);
656	amdgpu_bo_unreserve(adev->umsch_mm.ucode_fw_obj);
657	return 0;
658}
659
660int amdgpu_umsch_mm_allocate_ucode_data_buffer(struct amdgpu_umsch_mm *umsch)
661{
662	const struct umsch_mm_firmware_header_v1_0 *umsch_mm_hdr;
663	struct amdgpu_device *adev = umsch->ring.adev;
664	const __le32 *fw_data;
665	uint32_t fw_size;
666	int r;
667
668	umsch_mm_hdr = (const struct umsch_mm_firmware_header_v1_0 *)
669		       adev->umsch_mm.fw->data;
670
671	fw_data = (const __le32 *)(adev->umsch_mm.fw->data +
672		  le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_data_offset_bytes));
673	fw_size = le32_to_cpu(umsch_mm_hdr->umsch_mm_ucode_data_size_bytes);
674
675	r = amdgpu_bo_create_reserved(adev, fw_size,
676				      64 * 1024, AMDGPU_GEM_DOMAIN_VRAM,
677				      &adev->umsch_mm.data_fw_obj,
678				      &adev->umsch_mm.data_fw_gpu_addr,
679				      (void **)&adev->umsch_mm.data_fw_ptr);
680	if (r) {
681		dev_err(adev->dev, "(%d) failed to create umsch_mm fw data bo\n", r);
682		return r;
683	}
684
685	memcpy(adev->umsch_mm.data_fw_ptr, fw_data, fw_size);
686
687	amdgpu_bo_kunmap(adev->umsch_mm.data_fw_obj);
688	amdgpu_bo_unreserve(adev->umsch_mm.data_fw_obj);
689	return 0;
690}
691
692int amdgpu_umsch_mm_psp_execute_cmd_buf(struct amdgpu_umsch_mm *umsch)
693{
694	struct amdgpu_device *adev = umsch->ring.adev;
695	struct amdgpu_firmware_info ucode = {
696		.ucode_id = AMDGPU_UCODE_ID_UMSCH_MM_CMD_BUFFER,
697		.mc_addr = adev->umsch_mm.cmd_buf_gpu_addr,
698		.ucode_size = ((uintptr_t)adev->umsch_mm.cmd_buf_curr_ptr -
699			      (uintptr_t)adev->umsch_mm.cmd_buf_ptr),
700	};
701
702	return psp_execute_ip_fw_load(&adev->psp, &ucode);
703}
704
705static void umsch_mm_agdb_index_init(struct amdgpu_device *adev)
706{
707	uint32_t umsch_mm_agdb_start;
708	int i;
709
710	umsch_mm_agdb_start = adev->doorbell_index.max_assignment + 1;
711	umsch_mm_agdb_start = roundup(umsch_mm_agdb_start, 1024);
712	umsch_mm_agdb_start += (AMDGPU_NAVI10_DOORBELL64_VCN0_1 << 1);
713
714	for (i = 0; i < CONTEXT_PRIORITY_NUM_LEVELS; i++)
715		adev->umsch_mm.agdb_index[i] = umsch_mm_agdb_start + i;
716}
717
718static int umsch_mm_init(struct amdgpu_device *adev)
719{
720	int r;
721
722	adev->umsch_mm.vmid_mask_mm_vpe = 0xf00;
723	adev->umsch_mm.engine_mask = (1 << UMSCH_SWIP_ENGINE_TYPE_VPE);
724	adev->umsch_mm.vpe_hqd_mask = 0xfe;
725
726	r = amdgpu_device_wb_get(adev, &adev->umsch_mm.wb_index);
727	if (r) {
728		dev_err(adev->dev, "failed to alloc wb for umsch: %d\n", r);
729		return r;
730	}
731
732	adev->umsch_mm.sch_ctx_gpu_addr = adev->wb.gpu_addr +
733					  (adev->umsch_mm.wb_index * 4);
734
735	r = amdgpu_bo_create_kernel(adev, PAGE_SIZE, PAGE_SIZE,
736				    AMDGPU_GEM_DOMAIN_GTT,
737				    &adev->umsch_mm.cmd_buf_obj,
738				    &adev->umsch_mm.cmd_buf_gpu_addr,
739				    (void **)&adev->umsch_mm.cmd_buf_ptr);
740	if (r) {
741		dev_err(adev->dev, "failed to allocate cmdbuf bo %d\n", r);
742		amdgpu_device_wb_free(adev, adev->umsch_mm.wb_index);
743		return r;
744	}
745
746	mutex_init(&adev->umsch_mm.mutex_hidden);
747
748	umsch_mm_agdb_index_init(adev);
749
750	return 0;
751}
752
753
754static int umsch_mm_early_init(void *handle)
755{
756	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
757
758	switch (amdgpu_ip_version(adev, VCN_HWIP, 0)) {
759	case IP_VERSION(4, 0, 5):
760	case IP_VERSION(4, 0, 6):
761		umsch_mm_v4_0_set_funcs(&adev->umsch_mm);
762		break;
763	default:
764		return -EINVAL;
765	}
766
767	adev->umsch_mm.ring.funcs = &umsch_v4_0_ring_funcs;
768	umsch_mm_set_regs(&adev->umsch_mm);
769
770	return 0;
771}
772
773static int umsch_mm_late_init(void *handle)
774{
775	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
776
777	if (amdgpu_in_reset(adev) || adev->in_s0ix || adev->in_suspend)
778		return 0;
779
780	return umsch_mm_test(adev);
781}
782
783static int umsch_mm_sw_init(void *handle)
784{
785	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
786	int r;
787
788	r = umsch_mm_init(adev);
789	if (r)
790		return r;
791
792	r = umsch_mm_ring_init(&adev->umsch_mm);
793	if (r)
794		return r;
795
796	r = umsch_mm_init_microcode(&adev->umsch_mm);
797	if (r)
798		return r;
799
800	return 0;
801}
802
803static int umsch_mm_sw_fini(void *handle)
804{
805	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
806
807	release_firmware(adev->umsch_mm.fw);
808	adev->umsch_mm.fw = NULL;
809
810	amdgpu_ring_fini(&adev->umsch_mm.ring);
811
812	mutex_destroy(&adev->umsch_mm.mutex_hidden);
813
814	amdgpu_bo_free_kernel(&adev->umsch_mm.cmd_buf_obj,
815			      &adev->umsch_mm.cmd_buf_gpu_addr,
816			      (void **)&adev->umsch_mm.cmd_buf_ptr);
817
818	amdgpu_device_wb_free(adev, adev->umsch_mm.wb_index);
819
820	return 0;
821}
822
823static int umsch_mm_hw_init(void *handle)
824{
825	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
826	int r;
827
828	r = umsch_mm_load_microcode(&adev->umsch_mm);
829	if (r)
830		return r;
831
832	umsch_mm_ring_start(&adev->umsch_mm);
833
834	r = umsch_mm_set_hw_resources(&adev->umsch_mm);
835	if (r)
836		return r;
837
838	return 0;
839}
840
841static int umsch_mm_hw_fini(void *handle)
842{
843	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
844
845	umsch_mm_ring_stop(&adev->umsch_mm);
846
847	amdgpu_bo_free_kernel(&adev->umsch_mm.data_fw_obj,
848			      &adev->umsch_mm.data_fw_gpu_addr,
849			      (void **)&adev->umsch_mm.data_fw_ptr);
850
851	amdgpu_bo_free_kernel(&adev->umsch_mm.ucode_fw_obj,
852			      &adev->umsch_mm.ucode_fw_gpu_addr,
853			      (void **)&adev->umsch_mm.ucode_fw_ptr);
854	return 0;
855}
856
857static int umsch_mm_suspend(void *handle)
858{
859	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
860
861	return umsch_mm_hw_fini(adev);
862}
863
864static int umsch_mm_resume(void *handle)
865{
866	struct amdgpu_device *adev = (struct amdgpu_device *)handle;
867
868	return umsch_mm_hw_init(adev);
869}
870
871static const struct amd_ip_funcs umsch_mm_v4_0_ip_funcs = {
872	.name = "umsch_mm_v4_0",
873	.early_init = umsch_mm_early_init,
874	.late_init = umsch_mm_late_init,
875	.sw_init = umsch_mm_sw_init,
876	.sw_fini = umsch_mm_sw_fini,
877	.hw_init = umsch_mm_hw_init,
878	.hw_fini = umsch_mm_hw_fini,
879	.suspend = umsch_mm_suspend,
880	.resume = umsch_mm_resume,
881};
882
883const struct amdgpu_ip_block_version umsch_mm_v4_0_ip_block = {
884	.type = AMD_IP_BLOCK_TYPE_UMSCH_MM,
885	.major = 4,
886	.minor = 0,
887	.rev = 0,
888	.funcs = &umsch_mm_v4_0_ip_funcs,
889};