16dd09027SChristian König /*
26dd09027SChristian König  * Copyright 2019 Advanced Micro Devices, Inc.
36dd09027SChristian König  *
46dd09027SChristian König  * Permission is hereby granted, free of charge, to any person obtaining a
56dd09027SChristian König  * copy of this software and associated documentation files (the "Software"),
66dd09027SChristian König  * to deal in the Software without restriction, including without limitation
76dd09027SChristian König  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
86dd09027SChristian König  * and/or sell copies of the Software, and to permit persons to whom the
96dd09027SChristian König  * Software is furnished to do so, subject to the following conditions:
106dd09027SChristian König  *
116dd09027SChristian König  * The above copyright notice and this permission notice shall be included in
126dd09027SChristian König  * all copies or substantial portions of the Software.
136dd09027SChristian König  *
146dd09027SChristian König  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
156dd09027SChristian König  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
166dd09027SChristian König  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
176dd09027SChristian König  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
186dd09027SChristian König  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
196dd09027SChristian König  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
206dd09027SChristian König  * OTHER DEALINGS IN THE SOFTWARE.
216dd09027SChristian König  */
226dd09027SChristian König 
236dd09027SChristian König #include "amdgpu_vm.h"
246dd09027SChristian König #include "amdgpu_job.h"
256dd09027SChristian König #include "amdgpu_object.h"
266dd09027SChristian König #include "amdgpu_trace.h"
276dd09027SChristian König 
286dd09027SChristian König #define AMDGPU_VM_SDMA_MIN_NUM_DW	256u
296dd09027SChristian König #define AMDGPU_VM_SDMA_MAX_NUM_DW	(16u * 1024u)
306dd09027SChristian König 
316dd09027SChristian König /**
32*ecf96b52SChristian König  * amdgpu_vm_sdma_map_table - make sure new PDs/PTs are GTT mapped
33*ecf96b52SChristian König  *
34*ecf96b52SChristian König  * @table: newly allocated or validated PD/PT
35*ecf96b52SChristian König  */
36*ecf96b52SChristian König static int amdgpu_vm_sdma_map_table(struct amdgpu_bo *table)
37*ecf96b52SChristian König {
38*ecf96b52SChristian König 	int r;
39*ecf96b52SChristian König 
40*ecf96b52SChristian König 	r = amdgpu_ttm_alloc_gart(&table->tbo);
41*ecf96b52SChristian König 	if (r)
42*ecf96b52SChristian König 		return r;
43*ecf96b52SChristian König 
44*ecf96b52SChristian König 	if (table->shadow)
45*ecf96b52SChristian König 		r = amdgpu_ttm_alloc_gart(&table->shadow->tbo);
46*ecf96b52SChristian König 
47*ecf96b52SChristian König 	return r;
48*ecf96b52SChristian König }
49*ecf96b52SChristian König 
50*ecf96b52SChristian König /**
516dd09027SChristian König  * amdgpu_vm_sdma_prepare - prepare SDMA command submission
526dd09027SChristian König  *
536dd09027SChristian König  * @p: see amdgpu_vm_update_params definition
546dd09027SChristian König  * @owner: owner we need to sync to
556dd09027SChristian König  * @exclusive: exclusive move fence we need to sync to
566dd09027SChristian König  *
576dd09027SChristian König  * Returns:
586dd09027SChristian König  * Negativ errno, 0 for success.
596dd09027SChristian König  */
606dd09027SChristian König static int amdgpu_vm_sdma_prepare(struct amdgpu_vm_update_params *p,
616dd09027SChristian König 				  void *owner, struct dma_fence *exclusive)
626dd09027SChristian König {
636dd09027SChristian König 	struct amdgpu_bo *root = p->vm->root.base.bo;
646dd09027SChristian König 	unsigned int ndw = AMDGPU_VM_SDMA_MIN_NUM_DW;
656dd09027SChristian König 	int r;
666dd09027SChristian König 
676dd09027SChristian König 	r = amdgpu_job_alloc_with_ib(p->adev, ndw * 4, &p->job);
686dd09027SChristian König 	if (r)
696dd09027SChristian König 		return r;
706dd09027SChristian König 
716dd09027SChristian König 	r = amdgpu_sync_fence(p->adev, &p->job->sync, exclusive, false);
726dd09027SChristian König 	if (r)
736dd09027SChristian König 		return r;
746dd09027SChristian König 
756dd09027SChristian König 	r = amdgpu_sync_resv(p->adev, &p->job->sync, root->tbo.resv,
766dd09027SChristian König 			     owner, false);
776dd09027SChristian König 	if (r)
786dd09027SChristian König 		return r;
796dd09027SChristian König 
806dd09027SChristian König 	p->num_dw_left = ndw;
816dd09027SChristian König 	p->ib = &p->job->ibs[0];
826dd09027SChristian König 	return 0;
836dd09027SChristian König }
846dd09027SChristian König 
856dd09027SChristian König /**
866dd09027SChristian König  * amdgpu_vm_sdma_commit - commit SDMA command submission
876dd09027SChristian König  *
886dd09027SChristian König  * @p: see amdgpu_vm_update_params definition
896dd09027SChristian König  * @fence: resulting fence
906dd09027SChristian König  *
916dd09027SChristian König  * Returns:
926dd09027SChristian König  * Negativ errno, 0 for success.
936dd09027SChristian König  */
946dd09027SChristian König static int amdgpu_vm_sdma_commit(struct amdgpu_vm_update_params *p,
956dd09027SChristian König 				 struct dma_fence **fence)
966dd09027SChristian König {
976dd09027SChristian König 	struct amdgpu_bo *root = p->vm->root.base.bo;
986dd09027SChristian König 	struct amdgpu_ring *ring;
996dd09027SChristian König 	struct dma_fence *f;
1006dd09027SChristian König 	int r;
1016dd09027SChristian König 
1026dd09027SChristian König 	ring = container_of(p->vm->entity.rq->sched, struct amdgpu_ring, sched);
1036dd09027SChristian König 
1046dd09027SChristian König 	WARN_ON(p->ib->length_dw == 0);
1056dd09027SChristian König 	amdgpu_ring_pad_ib(ring, p->ib);
1066dd09027SChristian König 	WARN_ON(p->ib->length_dw > p->num_dw_left);
1076dd09027SChristian König 	r = amdgpu_job_submit(p->job, &p->vm->entity,
1086dd09027SChristian König 			      AMDGPU_FENCE_OWNER_VM, &f);
1096dd09027SChristian König 	if (r)
1106dd09027SChristian König 		goto error;
1116dd09027SChristian König 
1126dd09027SChristian König 	amdgpu_bo_fence(root, f, true);
1136dd09027SChristian König 	if (fence)
1146dd09027SChristian König 		swap(*fence, f);
1156dd09027SChristian König 	dma_fence_put(f);
1166dd09027SChristian König 	return 0;
1176dd09027SChristian König 
1186dd09027SChristian König error:
1196dd09027SChristian König 	amdgpu_job_free(p->job);
1206dd09027SChristian König 	return r;
1216dd09027SChristian König }
1226dd09027SChristian König 
1236dd09027SChristian König 
1246dd09027SChristian König /**
1256dd09027SChristian König  * amdgpu_vm_sdma_copy_ptes - copy the PTEs from mapping
1266dd09027SChristian König  *
1276dd09027SChristian König  * @p: see amdgpu_vm_update_params definition
1286dd09027SChristian König  * @bo: PD/PT to update
1296dd09027SChristian König  * @pe: addr of the page entry
1306dd09027SChristian König  * @count: number of page entries to copy
1316dd09027SChristian König  *
1326dd09027SChristian König  * Traces the parameters and calls the DMA function to copy the PTEs.
1336dd09027SChristian König  */
1346dd09027SChristian König static void amdgpu_vm_sdma_copy_ptes(struct amdgpu_vm_update_params *p,
1356dd09027SChristian König 				     struct amdgpu_bo *bo, uint64_t pe,
1366dd09027SChristian König 				     unsigned count)
1376dd09027SChristian König {
1386dd09027SChristian König 	uint64_t src = p->ib->gpu_addr;
1396dd09027SChristian König 
1406dd09027SChristian König 	src += p->num_dw_left * 4;
1416dd09027SChristian König 
1426dd09027SChristian König 	pe += amdgpu_bo_gpu_offset(bo);
1436dd09027SChristian König 	trace_amdgpu_vm_copy_ptes(pe, src, count);
1446dd09027SChristian König 
1456dd09027SChristian König 	amdgpu_vm_copy_pte(p->adev, p->ib, pe, src, count);
1466dd09027SChristian König }
1476dd09027SChristian König 
1486dd09027SChristian König /**
1496dd09027SChristian König  * amdgpu_vm_sdma_set_ptes - helper to call the right asic function
1506dd09027SChristian König  *
1516dd09027SChristian König  * @p: see amdgpu_vm_update_params definition
1526dd09027SChristian König  * @bo: PD/PT to update
1536dd09027SChristian König  * @pe: addr of the page entry
1546dd09027SChristian König  * @addr: dst addr to write into pe
1556dd09027SChristian König  * @count: number of page entries to update
1566dd09027SChristian König  * @incr: increase next addr by incr bytes
1576dd09027SChristian König  * @flags: hw access flags
1586dd09027SChristian König  *
1596dd09027SChristian König  * Traces the parameters and calls the right asic functions
1606dd09027SChristian König  * to setup the page table using the DMA.
1616dd09027SChristian König  */
1626dd09027SChristian König static void amdgpu_vm_sdma_set_ptes(struct amdgpu_vm_update_params *p,
1636dd09027SChristian König 				    struct amdgpu_bo *bo, uint64_t pe,
1646dd09027SChristian König 				    uint64_t addr, unsigned count,
1656dd09027SChristian König 				    uint32_t incr, uint64_t flags)
1666dd09027SChristian König {
1676dd09027SChristian König 	pe += amdgpu_bo_gpu_offset(bo);
1686dd09027SChristian König 	trace_amdgpu_vm_set_ptes(pe, addr, count, incr, flags);
1696dd09027SChristian König 	if (count < 3) {
1706dd09027SChristian König 		amdgpu_vm_write_pte(p->adev, p->ib, pe, addr | flags,
1716dd09027SChristian König 				    count, incr);
1726dd09027SChristian König 	} else {
1736dd09027SChristian König 		amdgpu_vm_set_pte_pde(p->adev, p->ib, pe, addr,
1746dd09027SChristian König 				      count, incr, flags);
1756dd09027SChristian König 	}
1766dd09027SChristian König }
1776dd09027SChristian König 
1786dd09027SChristian König /**
1796dd09027SChristian König  * amdgpu_vm_sdma_update - execute VM update
1806dd09027SChristian König  *
1816dd09027SChristian König  * @p: see amdgpu_vm_update_params definition
1826dd09027SChristian König  * @bo: PD/PT to update
1836dd09027SChristian König  * @pe: addr of the page entry
1846dd09027SChristian König  * @addr: dst addr to write into pe
1856dd09027SChristian König  * @count: number of page entries to update
1866dd09027SChristian König  * @incr: increase next addr by incr bytes
1876dd09027SChristian König  * @flags: hw access flags
1886dd09027SChristian König  *
1896dd09027SChristian König  * Reserve space in the IB, setup mapping buffer on demand and write commands to
1906dd09027SChristian König  * the IB.
1916dd09027SChristian König  */
1926dd09027SChristian König static int amdgpu_vm_sdma_update(struct amdgpu_vm_update_params *p,
1936dd09027SChristian König 				 struct amdgpu_bo *bo, uint64_t pe,
1946dd09027SChristian König 				 uint64_t addr, unsigned count, uint32_t incr,
1956dd09027SChristian König 				 uint64_t flags)
1966dd09027SChristian König {
1976dd09027SChristian König 	unsigned int i, ndw, nptes;
1986dd09027SChristian König 	uint64_t *pte;
1996dd09027SChristian König 	int r;
2006dd09027SChristian König 
2016dd09027SChristian König 	do {
2026dd09027SChristian König 		ndw = p->num_dw_left;
2036dd09027SChristian König 		ndw -= p->ib->length_dw;
2046dd09027SChristian König 
2056dd09027SChristian König 		if (ndw < 32) {
2066dd09027SChristian König 			r = amdgpu_vm_sdma_commit(p, NULL);
2076dd09027SChristian König 			if (r)
2086dd09027SChristian König 				return r;
2096dd09027SChristian König 
2106dd09027SChristian König 			/* estimate how many dw we need */
2116dd09027SChristian König 			ndw = 32;
2126dd09027SChristian König 			if (p->pages_addr)
2136dd09027SChristian König 				ndw += count * 2;
2146dd09027SChristian König 			ndw = max(ndw, AMDGPU_VM_SDMA_MIN_NUM_DW);
2156dd09027SChristian König 			ndw = min(ndw, AMDGPU_VM_SDMA_MAX_NUM_DW);
2166dd09027SChristian König 
2176dd09027SChristian König 			r = amdgpu_job_alloc_with_ib(p->adev, ndw * 4, &p->job);
2186dd09027SChristian König 			if (r)
2196dd09027SChristian König 				return r;
2206dd09027SChristian König 
2216dd09027SChristian König 			p->num_dw_left = ndw;
2226dd09027SChristian König 			p->ib = &p->job->ibs[0];
2236dd09027SChristian König 		}
2246dd09027SChristian König 
2256dd09027SChristian König 		if (!p->pages_addr) {
2266dd09027SChristian König 			/* set page commands needed */
2276dd09027SChristian König 			if (bo->shadow)
2286dd09027SChristian König 				amdgpu_vm_sdma_set_ptes(p, bo->shadow, pe, addr,
2296dd09027SChristian König 							count, incr, flags);
2306dd09027SChristian König 			amdgpu_vm_sdma_set_ptes(p, bo, pe, addr, count,
2316dd09027SChristian König 						incr, flags);
2326dd09027SChristian König 			return 0;
2336dd09027SChristian König 		}
2346dd09027SChristian König 
2356dd09027SChristian König 		/* copy commands needed */
2366dd09027SChristian König 		ndw -= p->adev->vm_manager.vm_pte_funcs->copy_pte_num_dw *
2376dd09027SChristian König 			(bo->shadow ? 2 : 1);
2386dd09027SChristian König 
2396dd09027SChristian König 		/* for padding */
2406dd09027SChristian König 		ndw -= 7;
2416dd09027SChristian König 
2426dd09027SChristian König 		nptes = min(count, ndw / 2);
2436dd09027SChristian König 
2446dd09027SChristian König 		/* Put the PTEs at the end of the IB. */
2456dd09027SChristian König 		p->num_dw_left -= nptes * 2;
2466dd09027SChristian König 		pte = (uint64_t *)&(p->ib->ptr[p->num_dw_left]);
2476dd09027SChristian König 		for (i = 0; i < nptes; ++i, addr += incr) {
2486dd09027SChristian König 			pte[i] = amdgpu_vm_map_gart(p->pages_addr, addr);
2496dd09027SChristian König 			pte[i] |= flags;
2506dd09027SChristian König 		}
2516dd09027SChristian König 
2526dd09027SChristian König 		if (bo->shadow)
2536dd09027SChristian König 			amdgpu_vm_sdma_copy_ptes(p, bo->shadow, pe, nptes);
2546dd09027SChristian König 		amdgpu_vm_sdma_copy_ptes(p, bo, pe, nptes);
2556dd09027SChristian König 
2566dd09027SChristian König 		pe += nptes * 8;
2576dd09027SChristian König 		count -= nptes;
2586dd09027SChristian König 	} while (count);
2596dd09027SChristian König 
2606dd09027SChristian König 	return 0;
2616dd09027SChristian König }
2626dd09027SChristian König 
2636dd09027SChristian König const struct amdgpu_vm_update_funcs amdgpu_vm_sdma_funcs = {
264*ecf96b52SChristian König 	.map_table = amdgpu_vm_sdma_map_table,
2656dd09027SChristian König 	.prepare = amdgpu_vm_sdma_prepare,
2666dd09027SChristian König 	.update = amdgpu_vm_sdma_update,
2676dd09027SChristian König 	.commit = amdgpu_vm_sdma_commit
2686dd09027SChristian König };
269