Patch series "improve use_mm / unuse_mm", v2. This series improves the use_mm / unuse_mm interface by better documenting the assumptions, and my taking the set_fs manipulations spread over the callers into the core API. This patch (of 3): Use the proper API instead. Link: http://lkml.kernel.org/r/20200404094101.672954-1-hch@lst.de These helpers are only for use with kernel threads, and I will tie them more into the kthread infrastructure going forward. Also move the prototypes to kthread.h - mmu_context.h was a little weird to start with as it otherwise contains very low-level MM bits. Signed-off-by: Christoph Hellwig <hch@lst.de> Signed-off-by: Andrew Morton <akpm@linux-foundation.org> Tested-by: Jens Axboe <axboe@kernel.dk> Reviewed-by: Jens Axboe <axboe@kernel.dk> Acked-by: Felix Kuehling <Felix.Kuehling@amd.com> Cc: Alex Deucher <alexander.deucher@amd.com> Cc: Al Viro <viro@zeniv.linux.org.uk> Cc: Felipe Balbi <balbi@kernel.org> Cc: Jason Wang <jasowang@redhat.com> Cc: "Michael S. Tsirkin" <mst@redhat.com> Cc: Zhenyu Wang <zhenyuw@linux.intel.com> Cc: Zhi Wang <zhi.a.wang@intel.com> Cc: Greg Kroah-Hartman <gregkh@linuxfoundation.org> Link: http://lkml.kernel.org/r/20200404094101.672954-1-hch@lst.de Link: http://lkml.kernel.org/r/20200416053158.586887-1-hch@lst.de Link: http://lkml.kernel.org/r/20200404094101.672954-5-hch@lst.de Signed-off-by: Linus Torvalds <torvalds@linux-foundation.org>
		
			
				
	
	
		
			324 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
			
		
		
	
	
			324 lines
		
	
	
		
			10 KiB
		
	
	
	
		
			C
		
	
	
	
	
	
/*
 | 
						|
 * Copyright 2019 Advanced Micro Devices, Inc.
 | 
						|
 *
 | 
						|
 * Permission is hereby granted, free of charge, to any person obtaining a
 | 
						|
 * copy of this software and associated documentation files (the "Software"),
 | 
						|
 * to deal in the Software without restriction, including without limitation
 | 
						|
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 | 
						|
 * and/or sell copies of the Software, and to permit persons to whom the
 | 
						|
 * Software is furnished to do so, subject to the following conditions:
 | 
						|
 *
 | 
						|
 * The above copyright notice and this permission notice shall be included in
 | 
						|
 * all copies or substantial portions of the Software.
 | 
						|
 *
 | 
						|
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 | 
						|
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 | 
						|
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 | 
						|
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 | 
						|
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 | 
						|
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 | 
						|
 * OTHER DEALINGS IN THE SOFTWARE.
 | 
						|
 */
 | 
						|
#include <linux/module.h>
 | 
						|
#include <linux/fdtable.h>
 | 
						|
#include <linux/uaccess.h>
 | 
						|
#include <linux/firmware.h>
 | 
						|
#include "amdgpu.h"
 | 
						|
#include "amdgpu_amdkfd.h"
 | 
						|
#include "sdma0/sdma0_4_2_2_offset.h"
 | 
						|
#include "sdma0/sdma0_4_2_2_sh_mask.h"
 | 
						|
#include "sdma1/sdma1_4_2_2_offset.h"
 | 
						|
#include "sdma1/sdma1_4_2_2_sh_mask.h"
 | 
						|
#include "sdma2/sdma2_4_2_2_offset.h"
 | 
						|
#include "sdma2/sdma2_4_2_2_sh_mask.h"
 | 
						|
#include "sdma3/sdma3_4_2_2_offset.h"
 | 
						|
#include "sdma3/sdma3_4_2_2_sh_mask.h"
 | 
						|
#include "sdma4/sdma4_4_2_2_offset.h"
 | 
						|
#include "sdma4/sdma4_4_2_2_sh_mask.h"
 | 
						|
#include "sdma5/sdma5_4_2_2_offset.h"
 | 
						|
#include "sdma5/sdma5_4_2_2_sh_mask.h"
 | 
						|
#include "sdma6/sdma6_4_2_2_offset.h"
 | 
						|
#include "sdma6/sdma6_4_2_2_sh_mask.h"
 | 
						|
#include "sdma7/sdma7_4_2_2_offset.h"
 | 
						|
#include "sdma7/sdma7_4_2_2_sh_mask.h"
 | 
						|
#include "v9_structs.h"
 | 
						|
#include "soc15.h"
 | 
						|
#include "soc15d.h"
 | 
						|
#include "amdgpu_amdkfd_gfx_v9.h"
 | 
						|
#include "gfxhub_v1_0.h"
 | 
						|
#include "mmhub_v9_4.h"
 | 
						|
 | 
						|
#define HQD_N_REGS 56
 | 
						|
#define DUMP_REG(addr) do {				\
 | 
						|
		if (WARN_ON_ONCE(i >= HQD_N_REGS))	\
 | 
						|
			break;				\
 | 
						|
		(*dump)[i][0] = (addr) << 2;		\
 | 
						|
		(*dump)[i++][1] = RREG32(addr);		\
 | 
						|
	} while (0)
 | 
						|
 | 
						|
static inline struct amdgpu_device *get_amdgpu_device(struct kgd_dev *kgd)
 | 
						|
{
 | 
						|
	return (struct amdgpu_device *)kgd;
 | 
						|
}
 | 
						|
 | 
						|
static inline struct v9_sdma_mqd *get_sdma_mqd(void *mqd)
 | 
						|
{
 | 
						|
	return (struct v9_sdma_mqd *)mqd;
 | 
						|
}
 | 
						|
 | 
						|
static uint32_t get_sdma_rlc_reg_offset(struct amdgpu_device *adev,
 | 
						|
				unsigned int engine_id,
 | 
						|
				unsigned int queue_id)
 | 
						|
{
 | 
						|
	uint32_t sdma_engine_reg_base = 0;
 | 
						|
	uint32_t sdma_rlc_reg_offset;
 | 
						|
 | 
						|
	switch (engine_id) {
 | 
						|
	default:
 | 
						|
		dev_warn(adev->dev,
 | 
						|
			 "Invalid sdma engine id (%d), using engine id 0\n",
 | 
						|
			 engine_id);
 | 
						|
		fallthrough;
 | 
						|
	case 0:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA0, 0,
 | 
						|
				mmSDMA0_RLC0_RB_CNTL) - mmSDMA0_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 1:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA1, 0,
 | 
						|
				mmSDMA1_RLC0_RB_CNTL) - mmSDMA1_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 2:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA2, 0,
 | 
						|
				mmSDMA2_RLC0_RB_CNTL) - mmSDMA2_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 3:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA3, 0,
 | 
						|
				mmSDMA3_RLC0_RB_CNTL) - mmSDMA3_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 4:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA4, 0,
 | 
						|
				mmSDMA4_RLC0_RB_CNTL) - mmSDMA4_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 5:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA5, 0,
 | 
						|
				mmSDMA5_RLC0_RB_CNTL) - mmSDMA5_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 6:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA6, 0,
 | 
						|
				mmSDMA6_RLC0_RB_CNTL) - mmSDMA6_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	case 7:
 | 
						|
		sdma_engine_reg_base = SOC15_REG_OFFSET(SDMA7, 0,
 | 
						|
				mmSDMA7_RLC0_RB_CNTL) - mmSDMA7_RLC0_RB_CNTL;
 | 
						|
		break;
 | 
						|
	}
 | 
						|
 | 
						|
	sdma_rlc_reg_offset = sdma_engine_reg_base
 | 
						|
		+ queue_id * (mmSDMA0_RLC1_RB_CNTL - mmSDMA0_RLC0_RB_CNTL);
 | 
						|
 | 
						|
	pr_debug("RLC register offset for SDMA%d RLC%d: 0x%x\n", engine_id,
 | 
						|
			queue_id, sdma_rlc_reg_offset);
 | 
						|
 | 
						|
	return sdma_rlc_reg_offset;
 | 
						|
}
 | 
						|
 | 
						|
static int kgd_hqd_sdma_load(struct kgd_dev *kgd, void *mqd,
 | 
						|
			     uint32_t __user *wptr, struct mm_struct *mm)
 | 
						|
{
 | 
						|
	struct amdgpu_device *adev = get_amdgpu_device(kgd);
 | 
						|
	struct v9_sdma_mqd *m;
 | 
						|
	uint32_t sdma_rlc_reg_offset;
 | 
						|
	unsigned long end_jiffies;
 | 
						|
	uint32_t data;
 | 
						|
	uint64_t data64;
 | 
						|
	uint64_t __user *wptr64 = (uint64_t __user *)wptr;
 | 
						|
 | 
						|
	m = get_sdma_mqd(mqd);
 | 
						|
	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
 | 
						|
					    m->sdma_queue_id);
 | 
						|
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL,
 | 
						|
		m->sdmax_rlcx_rb_cntl & (~SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK));
 | 
						|
 | 
						|
	end_jiffies = msecs_to_jiffies(2000) + jiffies;
 | 
						|
	while (true) {
 | 
						|
		data = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_CONTEXT_STATUS);
 | 
						|
		if (data & SDMA0_RLC0_CONTEXT_STATUS__IDLE_MASK)
 | 
						|
			break;
 | 
						|
		if (time_after(jiffies, end_jiffies)) {
 | 
						|
			pr_err("SDMA RLC not idle in %s\n", __func__);
 | 
						|
			return -ETIME;
 | 
						|
		}
 | 
						|
		usleep_range(500, 1000);
 | 
						|
	}
 | 
						|
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL_OFFSET,
 | 
						|
	       m->sdmax_rlcx_doorbell_offset);
 | 
						|
 | 
						|
	data = REG_SET_FIELD(m->sdmax_rlcx_doorbell, SDMA0_RLC0_DOORBELL,
 | 
						|
			     ENABLE, 1);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL, data);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR,
 | 
						|
				m->sdmax_rlcx_rb_rptr);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_HI,
 | 
						|
				m->sdmax_rlcx_rb_rptr_hi);
 | 
						|
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_MINOR_PTR_UPDATE, 1);
 | 
						|
	if (read_user_wptr(mm, wptr64, data64)) {
 | 
						|
		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR,
 | 
						|
		       lower_32_bits(data64));
 | 
						|
		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR_HI,
 | 
						|
		       upper_32_bits(data64));
 | 
						|
	} else {
 | 
						|
		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR,
 | 
						|
		       m->sdmax_rlcx_rb_rptr);
 | 
						|
		WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_WPTR_HI,
 | 
						|
		       m->sdmax_rlcx_rb_rptr_hi);
 | 
						|
	}
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_MINOR_PTR_UPDATE, 0);
 | 
						|
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_BASE, m->sdmax_rlcx_rb_base);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_BASE_HI,
 | 
						|
			m->sdmax_rlcx_rb_base_hi);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_ADDR_LO,
 | 
						|
			m->sdmax_rlcx_rb_rptr_addr_lo);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_ADDR_HI,
 | 
						|
			m->sdmax_rlcx_rb_rptr_addr_hi);
 | 
						|
 | 
						|
	data = REG_SET_FIELD(m->sdmax_rlcx_rb_cntl, SDMA0_RLC0_RB_CNTL,
 | 
						|
			     RB_ENABLE, 1);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL, data);
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static int kgd_hqd_sdma_dump(struct kgd_dev *kgd,
 | 
						|
			     uint32_t engine_id, uint32_t queue_id,
 | 
						|
			     uint32_t (**dump)[2], uint32_t *n_regs)
 | 
						|
{
 | 
						|
	struct amdgpu_device *adev = get_amdgpu_device(kgd);
 | 
						|
	uint32_t sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev,
 | 
						|
			engine_id, queue_id);
 | 
						|
	uint32_t i = 0, reg;
 | 
						|
#undef HQD_N_REGS
 | 
						|
#define HQD_N_REGS (19+6+7+10)
 | 
						|
 | 
						|
	*dump = kmalloc_array(HQD_N_REGS * 2, sizeof(uint32_t), GFP_KERNEL);
 | 
						|
	if (*dump == NULL)
 | 
						|
		return -ENOMEM;
 | 
						|
 | 
						|
	for (reg = mmSDMA0_RLC0_RB_CNTL; reg <= mmSDMA0_RLC0_DOORBELL; reg++)
 | 
						|
		DUMP_REG(sdma_rlc_reg_offset + reg);
 | 
						|
	for (reg = mmSDMA0_RLC0_STATUS; reg <= mmSDMA0_RLC0_CSA_ADDR_HI; reg++)
 | 
						|
		DUMP_REG(sdma_rlc_reg_offset + reg);
 | 
						|
	for (reg = mmSDMA0_RLC0_IB_SUB_REMAIN;
 | 
						|
	     reg <= mmSDMA0_RLC0_MINOR_PTR_UPDATE; reg++)
 | 
						|
		DUMP_REG(sdma_rlc_reg_offset + reg);
 | 
						|
	for (reg = mmSDMA0_RLC0_MIDCMD_DATA0;
 | 
						|
	     reg <= mmSDMA0_RLC0_MIDCMD_CNTL; reg++)
 | 
						|
		DUMP_REG(sdma_rlc_reg_offset + reg);
 | 
						|
 | 
						|
	WARN_ON_ONCE(i != HQD_N_REGS);
 | 
						|
	*n_regs = i;
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static bool kgd_hqd_sdma_is_occupied(struct kgd_dev *kgd, void *mqd)
 | 
						|
{
 | 
						|
	struct amdgpu_device *adev = get_amdgpu_device(kgd);
 | 
						|
	struct v9_sdma_mqd *m;
 | 
						|
	uint32_t sdma_rlc_reg_offset;
 | 
						|
	uint32_t sdma_rlc_rb_cntl;
 | 
						|
 | 
						|
	m = get_sdma_mqd(mqd);
 | 
						|
	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
 | 
						|
					    m->sdma_queue_id);
 | 
						|
 | 
						|
	sdma_rlc_rb_cntl = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL);
 | 
						|
 | 
						|
	if (sdma_rlc_rb_cntl & SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK)
 | 
						|
		return true;
 | 
						|
 | 
						|
	return false;
 | 
						|
}
 | 
						|
 | 
						|
static int kgd_hqd_sdma_destroy(struct kgd_dev *kgd, void *mqd,
 | 
						|
				unsigned int utimeout)
 | 
						|
{
 | 
						|
	struct amdgpu_device *adev = get_amdgpu_device(kgd);
 | 
						|
	struct v9_sdma_mqd *m;
 | 
						|
	uint32_t sdma_rlc_reg_offset;
 | 
						|
	uint32_t temp;
 | 
						|
	unsigned long end_jiffies = (utimeout * HZ / 1000) + jiffies;
 | 
						|
 | 
						|
	m = get_sdma_mqd(mqd);
 | 
						|
	sdma_rlc_reg_offset = get_sdma_rlc_reg_offset(adev, m->sdma_engine_id,
 | 
						|
					    m->sdma_queue_id);
 | 
						|
 | 
						|
	temp = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL);
 | 
						|
	temp = temp & ~SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK;
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL, temp);
 | 
						|
 | 
						|
	while (true) {
 | 
						|
		temp = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_CONTEXT_STATUS);
 | 
						|
		if (temp & SDMA0_RLC0_CONTEXT_STATUS__IDLE_MASK)
 | 
						|
			break;
 | 
						|
		if (time_after(jiffies, end_jiffies)) {
 | 
						|
			pr_err("SDMA RLC not idle in %s\n", __func__);
 | 
						|
			return -ETIME;
 | 
						|
		}
 | 
						|
		usleep_range(500, 1000);
 | 
						|
	}
 | 
						|
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_DOORBELL, 0);
 | 
						|
	WREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL,
 | 
						|
		RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_CNTL) |
 | 
						|
		SDMA0_RLC0_RB_CNTL__RB_ENABLE_MASK);
 | 
						|
 | 
						|
	m->sdmax_rlcx_rb_rptr = RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR);
 | 
						|
	m->sdmax_rlcx_rb_rptr_hi =
 | 
						|
		RREG32(sdma_rlc_reg_offset + mmSDMA0_RLC0_RB_RPTR_HI);
 | 
						|
 | 
						|
	return 0;
 | 
						|
}
 | 
						|
 | 
						|
static void kgd_set_vm_context_page_table_base(struct kgd_dev *kgd, uint32_t vmid,
 | 
						|
		uint64_t page_table_base)
 | 
						|
{
 | 
						|
	struct amdgpu_device *adev = get_amdgpu_device(kgd);
 | 
						|
 | 
						|
	if (!amdgpu_amdkfd_is_kfd_vmid(adev, vmid)) {
 | 
						|
		pr_err("trying to set page table base for wrong VMID %u\n",
 | 
						|
		       vmid);
 | 
						|
		return;
 | 
						|
	}
 | 
						|
 | 
						|
	mmhub_v9_4_setup_vm_pt_regs(adev, vmid, page_table_base);
 | 
						|
 | 
						|
	gfxhub_v1_0_setup_vm_pt_regs(adev, vmid, page_table_base);
 | 
						|
}
 | 
						|
 | 
						|
const struct kfd2kgd_calls arcturus_kfd2kgd = {
 | 
						|
	.program_sh_mem_settings = kgd_gfx_v9_program_sh_mem_settings,
 | 
						|
	.set_pasid_vmid_mapping = kgd_gfx_v9_set_pasid_vmid_mapping,
 | 
						|
	.init_interrupts = kgd_gfx_v9_init_interrupts,
 | 
						|
	.hqd_load = kgd_gfx_v9_hqd_load,
 | 
						|
	.hiq_mqd_load = kgd_gfx_v9_hiq_mqd_load,
 | 
						|
	.hqd_sdma_load = kgd_hqd_sdma_load,
 | 
						|
	.hqd_dump = kgd_gfx_v9_hqd_dump,
 | 
						|
	.hqd_sdma_dump = kgd_hqd_sdma_dump,
 | 
						|
	.hqd_is_occupied = kgd_gfx_v9_hqd_is_occupied,
 | 
						|
	.hqd_sdma_is_occupied = kgd_hqd_sdma_is_occupied,
 | 
						|
	.hqd_destroy = kgd_gfx_v9_hqd_destroy,
 | 
						|
	.hqd_sdma_destroy = kgd_hqd_sdma_destroy,
 | 
						|
	.address_watch_disable = kgd_gfx_v9_address_watch_disable,
 | 
						|
	.address_watch_execute = kgd_gfx_v9_address_watch_execute,
 | 
						|
	.wave_control_execute = kgd_gfx_v9_wave_control_execute,
 | 
						|
	.address_watch_get_offset = kgd_gfx_v9_address_watch_get_offset,
 | 
						|
	.get_atc_vmid_pasid_mapping_info =
 | 
						|
			kgd_gfx_v9_get_atc_vmid_pasid_mapping_info,
 | 
						|
	.set_vm_context_page_table_base = kgd_set_vm_context_page_table_base,
 | 
						|
	.get_hive_id = amdgpu_amdkfd_get_hive_id,
 | 
						|
};
 |