162306a36Sopenharmony_ci/*
262306a36Sopenharmony_ci * Copyright 2016 Advanced Micro Devices, Inc.
362306a36Sopenharmony_ci *
462306a36Sopenharmony_ci * Permission is hereby granted, free of charge, to any person obtaining a
562306a36Sopenharmony_ci * copy of this software and associated documentation files (the "Software"),
662306a36Sopenharmony_ci * to deal in the Software without restriction, including without limitation
762306a36Sopenharmony_ci * the rights to use, copy, modify, merge, publish, distribute, sublicense,
862306a36Sopenharmony_ci * and/or sell copies of the Software, and to permit persons to whom the
962306a36Sopenharmony_ci * Software is furnished to do so, subject to the following conditions:
1062306a36Sopenharmony_ci *
1162306a36Sopenharmony_ci * The above copyright notice and this permission notice shall be included in
1262306a36Sopenharmony_ci * all copies or substantial portions of the Software.
1362306a36Sopenharmony_ci *
1462306a36Sopenharmony_ci * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
1562306a36Sopenharmony_ci * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
1662306a36Sopenharmony_ci * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
1762306a36Sopenharmony_ci * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
1862306a36Sopenharmony_ci * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
1962306a36Sopenharmony_ci * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
2062306a36Sopenharmony_ci * OTHER DEALINGS IN THE SOFTWARE.
2162306a36Sopenharmony_ci
2262306a36Sopenharmony_ci * * Author: Monk.liu@amd.com
2362306a36Sopenharmony_ci */
2462306a36Sopenharmony_ci
2562306a36Sopenharmony_ci#include <drm/drm_exec.h>
2662306a36Sopenharmony_ci
2762306a36Sopenharmony_ci#include "amdgpu.h"
2862306a36Sopenharmony_ci
2962306a36Sopenharmony_ciuint64_t amdgpu_csa_vaddr(struct amdgpu_device *adev)
3062306a36Sopenharmony_ci{
3162306a36Sopenharmony_ci	uint64_t addr = adev->vm_manager.max_pfn << AMDGPU_GPU_PAGE_SHIFT;
3262306a36Sopenharmony_ci
3362306a36Sopenharmony_ci	addr -= AMDGPU_VA_RESERVED_SIZE;
3462306a36Sopenharmony_ci	addr = amdgpu_gmc_sign_extend(addr);
3562306a36Sopenharmony_ci
3662306a36Sopenharmony_ci	return addr;
3762306a36Sopenharmony_ci}
3862306a36Sopenharmony_ci
3962306a36Sopenharmony_ciint amdgpu_allocate_static_csa(struct amdgpu_device *adev, struct amdgpu_bo **bo,
4062306a36Sopenharmony_ci				u32 domain, uint32_t size)
4162306a36Sopenharmony_ci{
4262306a36Sopenharmony_ci	void *ptr;
4362306a36Sopenharmony_ci
4462306a36Sopenharmony_ci	amdgpu_bo_create_kernel(adev, size, PAGE_SIZE,
4562306a36Sopenharmony_ci				domain, bo,
4662306a36Sopenharmony_ci				NULL, &ptr);
4762306a36Sopenharmony_ci	if (!*bo)
4862306a36Sopenharmony_ci		return -ENOMEM;
4962306a36Sopenharmony_ci
5062306a36Sopenharmony_ci	memset(ptr, 0, size);
5162306a36Sopenharmony_ci	adev->virt.csa_cpu_addr = ptr;
5262306a36Sopenharmony_ci	return 0;
5362306a36Sopenharmony_ci}
5462306a36Sopenharmony_ci
5562306a36Sopenharmony_civoid amdgpu_free_static_csa(struct amdgpu_bo **bo)
5662306a36Sopenharmony_ci{
5762306a36Sopenharmony_ci	amdgpu_bo_free_kernel(bo, NULL, NULL);
5862306a36Sopenharmony_ci}
5962306a36Sopenharmony_ci
6062306a36Sopenharmony_ci/*
6162306a36Sopenharmony_ci * amdgpu_map_static_csa should be called during amdgpu_vm_init
6262306a36Sopenharmony_ci * it maps virtual address amdgpu_csa_vaddr() to this VM, and each command
6362306a36Sopenharmony_ci * submission of GFX should use this virtual address within META_DATA init
6462306a36Sopenharmony_ci * package to support SRIOV gfx preemption.
6562306a36Sopenharmony_ci */
6662306a36Sopenharmony_ciint amdgpu_map_static_csa(struct amdgpu_device *adev, struct amdgpu_vm *vm,
6762306a36Sopenharmony_ci			  struct amdgpu_bo *bo, struct amdgpu_bo_va **bo_va,
6862306a36Sopenharmony_ci			  uint64_t csa_addr, uint32_t size)
6962306a36Sopenharmony_ci{
7062306a36Sopenharmony_ci	struct drm_exec exec;
7162306a36Sopenharmony_ci	int r;
7262306a36Sopenharmony_ci
7362306a36Sopenharmony_ci	drm_exec_init(&exec, DRM_EXEC_INTERRUPTIBLE_WAIT);
7462306a36Sopenharmony_ci	drm_exec_until_all_locked(&exec) {
7562306a36Sopenharmony_ci		r = amdgpu_vm_lock_pd(vm, &exec, 0);
7662306a36Sopenharmony_ci		if (likely(!r))
7762306a36Sopenharmony_ci			r = drm_exec_lock_obj(&exec, &bo->tbo.base);
7862306a36Sopenharmony_ci		drm_exec_retry_on_contention(&exec);
7962306a36Sopenharmony_ci		if (unlikely(r)) {
8062306a36Sopenharmony_ci			DRM_ERROR("failed to reserve CSA,PD BOs: err=%d\n", r);
8162306a36Sopenharmony_ci			goto error;
8262306a36Sopenharmony_ci		}
8362306a36Sopenharmony_ci	}
8462306a36Sopenharmony_ci
8562306a36Sopenharmony_ci	*bo_va = amdgpu_vm_bo_add(adev, vm, bo);
8662306a36Sopenharmony_ci	if (!*bo_va) {
8762306a36Sopenharmony_ci		r = -ENOMEM;
8862306a36Sopenharmony_ci		goto error;
8962306a36Sopenharmony_ci	}
9062306a36Sopenharmony_ci
9162306a36Sopenharmony_ci	r = amdgpu_vm_bo_map(adev, *bo_va, csa_addr, 0, size,
9262306a36Sopenharmony_ci			     AMDGPU_PTE_READABLE | AMDGPU_PTE_WRITEABLE |
9362306a36Sopenharmony_ci			     AMDGPU_PTE_EXECUTABLE);
9462306a36Sopenharmony_ci
9562306a36Sopenharmony_ci	if (r) {
9662306a36Sopenharmony_ci		DRM_ERROR("failed to do bo_map on static CSA, err=%d\n", r);
9762306a36Sopenharmony_ci		amdgpu_vm_bo_del(adev, *bo_va);
9862306a36Sopenharmony_ci		goto error;
9962306a36Sopenharmony_ci	}
10062306a36Sopenharmony_ci
10162306a36Sopenharmony_cierror:
10262306a36Sopenharmony_ci	drm_exec_fini(&exec);
10362306a36Sopenharmony_ci	return r;
10462306a36Sopenharmony_ci}
10562306a36Sopenharmony_ci
10662306a36Sopenharmony_ciint amdgpu_unmap_static_csa(struct amdgpu_device *adev, struct amdgpu_vm *vm,
10762306a36Sopenharmony_ci			    struct amdgpu_bo *bo, struct amdgpu_bo_va *bo_va,
10862306a36Sopenharmony_ci			    uint64_t csa_addr)
10962306a36Sopenharmony_ci{
11062306a36Sopenharmony_ci	struct drm_exec exec;
11162306a36Sopenharmony_ci	int r;
11262306a36Sopenharmony_ci
11362306a36Sopenharmony_ci	drm_exec_init(&exec, DRM_EXEC_INTERRUPTIBLE_WAIT);
11462306a36Sopenharmony_ci	drm_exec_until_all_locked(&exec) {
11562306a36Sopenharmony_ci		r = amdgpu_vm_lock_pd(vm, &exec, 0);
11662306a36Sopenharmony_ci		if (likely(!r))
11762306a36Sopenharmony_ci			r = drm_exec_lock_obj(&exec, &bo->tbo.base);
11862306a36Sopenharmony_ci		drm_exec_retry_on_contention(&exec);
11962306a36Sopenharmony_ci		if (unlikely(r)) {
12062306a36Sopenharmony_ci			DRM_ERROR("failed to reserve CSA,PD BOs: err=%d\n", r);
12162306a36Sopenharmony_ci			goto error;
12262306a36Sopenharmony_ci		}
12362306a36Sopenharmony_ci	}
12462306a36Sopenharmony_ci
12562306a36Sopenharmony_ci	r = amdgpu_vm_bo_unmap(adev, bo_va, csa_addr);
12662306a36Sopenharmony_ci	if (r) {
12762306a36Sopenharmony_ci		DRM_ERROR("failed to do bo_unmap on static CSA, err=%d\n", r);
12862306a36Sopenharmony_ci		goto error;
12962306a36Sopenharmony_ci	}
13062306a36Sopenharmony_ci
13162306a36Sopenharmony_ci	amdgpu_vm_bo_del(adev, bo_va);
13262306a36Sopenharmony_ci
13362306a36Sopenharmony_cierror:
13462306a36Sopenharmony_ci	drm_exec_fini(&exec);
13562306a36Sopenharmony_ci	return r;
13662306a36Sopenharmony_ci}
137