kfd_priv.h 40.6 KB
Newer Older
1
/* SPDX-License-Identifier: GPL-2.0 OR MIT */
O
Oded Gabbay 已提交
2
/*
3
 * Copyright 2014-2022 Advanced Micro Devices, Inc.
O
Oded Gabbay 已提交
4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
 *
 * 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.
 */

#ifndef KFD_PRIV_H_INCLUDED
#define KFD_PRIV_H_INCLUDED

#include <linux/hashtable.h>
#include <linux/mmu_notifier.h>
#include <linux/mutex.h>
#include <linux/types.h>
#include <linux/atomic.h>
#include <linux/workqueue.h>
#include <linux/spinlock.h>
34
#include <linux/kfd_ioctl.h>
35
#include <linux/idr.h>
36
#include <linux/kfifo.h>
37
#include <linux/seq_file.h>
38
#include <linux/kref.h>
39
#include <linux/sysfs.h>
40
#include <linux/device_cgroup.h>
41 42 43
#include <drm/drm_file.h>
#include <drm/drm_drv.h>
#include <drm/drm_device.h>
44
#include <drm/drm_ioctl.h>
O
Oded Gabbay 已提交
45
#include <kgd_kfd_interface.h>
46
#include <linux/swap.h>
O
Oded Gabbay 已提交
47

48
#include "amd_shared.h"
49
#include "amdgpu.h"
50

L
Laura Abbott 已提交
51 52
#define KFD_MAX_RING_ENTRY_SIZE	8

53 54
#define KFD_SYSFS_FILE_MODE 0444

55 56 57 58 59 60 61 62 63 64 65
/* GPU ID hash width in bits */
#define KFD_GPU_ID_HASH_WIDTH 16

/* Use upper bits of mmap offset to store KFD driver specific information.
 * BITS[63:62] - Encode MMAP type
 * BITS[61:46] - Encode gpu_id. To identify to which GPU the offset belongs to
 * BITS[45:0]  - MMAP offset value
 *
 * NOTE: struct vm_area_struct.vm_pgoff uses offset in pages. Hence, these
 *  defines are w.r.t to PAGE_SIZE
 */
66
#define KFD_MMAP_TYPE_SHIFT	62
67 68 69 70
#define KFD_MMAP_TYPE_MASK	(0x3ULL << KFD_MMAP_TYPE_SHIFT)
#define KFD_MMAP_TYPE_DOORBELL	(0x3ULL << KFD_MMAP_TYPE_SHIFT)
#define KFD_MMAP_TYPE_EVENTS	(0x2ULL << KFD_MMAP_TYPE_SHIFT)
#define KFD_MMAP_TYPE_RESERVED_MEM	(0x1ULL << KFD_MMAP_TYPE_SHIFT)
O
Oak Zeng 已提交
71
#define KFD_MMAP_TYPE_MMIO	(0x0ULL << KFD_MMAP_TYPE_SHIFT)
72

73
#define KFD_MMAP_GPU_ID_SHIFT 46
74 75 76 77
#define KFD_MMAP_GPU_ID_MASK (((1ULL << KFD_GPU_ID_HASH_WIDTH) - 1) \
				<< KFD_MMAP_GPU_ID_SHIFT)
#define KFD_MMAP_GPU_ID(gpu_id) ((((uint64_t)gpu_id) << KFD_MMAP_GPU_ID_SHIFT)\
				& KFD_MMAP_GPU_ID_MASK)
78
#define KFD_MMAP_GET_GPU_ID(offset)    ((offset & KFD_MMAP_GPU_ID_MASK) \
79 80
				>> KFD_MMAP_GPU_ID_SHIFT)

B
Ben Goz 已提交
81 82
/*
 * When working with cp scheduler we should assign the HIQ manually or via
83
 * the amdgpu driver to a fixed hqd slot, here are the fixed HIQ hqd slot
B
Ben Goz 已提交
84 85 86 87 88 89 90
 * definitions for Kaveri. In Kaveri only the first ME queues participates
 * in the cp scheduling taking that in mind we set the HIQ slot in the
 * second ME.
 */
#define KFD_CIK_HIQ_PIPE 4
#define KFD_CIK_HIQ_QUEUE 0

91 92 93 94
/* Macro for allocating structures */
#define kfd_alloc_struct(ptr_to_struct)	\
	((typeof(ptr_to_struct)) kzalloc(sizeof(*ptr_to_struct), GFP_KERNEL))

95
#define KFD_MAX_NUM_OF_PROCESSES 512
96
#define KFD_MAX_NUM_OF_QUEUES_PER_PROCESS 1024
97

F
Felix Kuehling 已提交
98 99 100 101
/*
 * Size of the per-process TBA+TMA buffer: 2 pages
 *
 * The first page is the TBA used for the CWSR ISA code. The second
102
 * page is used as TMA for user-mode trap handler setup in daisy-chain mode.
F
Felix Kuehling 已提交
103 104 105 106
 */
#define KFD_CWSR_TBA_TMA_SIZE (PAGE_SIZE * 2)
#define KFD_CWSR_TMA_OFFSET PAGE_SIZE

107 108 109 110 111 112
#define KFD_MAX_NUM_OF_QUEUES_PER_DEVICE		\
	(KFD_MAX_NUM_OF_PROCESSES *			\
			KFD_MAX_NUM_OF_QUEUES_PER_PROCESS)

#define KFD_KERNEL_QUEUE_SIZE 2048

113 114
#define KFD_UNMAP_LATENCY_MS	(4000)

115 116 117 118 119 120 121 122 123 124
/*
 * 512 = 0x200
 * The doorbell index distance between SDMA RLC (2*i) and (2*i+1) in the
 * same SDMA engine on SOC15, which has 8-byte doorbells for SDMA.
 * 512 8-byte doorbell distance (i.e. one page away) ensures that SDMA RLC
 * (2*i+1) doorbells (in terms of the lower 12 bit address) lie exactly in
 * the OFFSET and SIZE set in registers like BIF_SDMA0_DOORBELL_RANGE.
 */
#define KFD_QUEUE_DOORBELL_MIRROR_OFFSET 512

125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144
/**
 * enum kfd_ioctl_flags - KFD ioctl flags
 * Various flags that can be set in &amdkfd_ioctl_desc.flags to control how
 * userspace can use a given ioctl.
 */
enum kfd_ioctl_flags {
	/*
	 * @KFD_IOC_FLAG_CHECKPOINT_RESTORE:
	 * Certain KFD ioctls such as AMDKFD_IOC_CRIU_OP can potentially
	 * perform privileged operations and load arbitrary data into MQDs and
	 * eventually HQD registers when the queue is mapped by HWS. In order to
	 * prevent this we should perform additional security checks.
	 *
	 * This is equivalent to callers with the CHECKPOINT_RESTORE capability.
	 *
	 * Note: Since earlier versions of docker do not support CHECKPOINT_RESTORE,
	 * we also allow ioctls with SYS_ADMIN capability.
	 */
	KFD_IOC_FLAG_CHECKPOINT_RESTORE = BIT(0),
};
145
/*
146 147
 * Kernel module parameter to specify maximum number of supported queues per
 * device
148
 */
149
extern int max_num_of_queues_per_device;
150

B
Ben Goz 已提交
151

152 153 154
/* Kernel module parameter to specify the scheduling policy */
extern int sched_policy;

155 156 157 158 159 160
/*
 * Kernel module parameter to specify the maximum process
 * number per HW scheduler
 */
extern int hws_max_conc_proc;

F
Felix Kuehling 已提交
161 162
extern int cwsr_enable;

163 164 165 166 167 168
/*
 * Kernel module parameter to specify whether to send sigterm to HSA process on
 * unhandled exception
 */
extern int send_sigterm;

169 170 171 172 173 174
/*
 * This kernel module is used to simulate large bar machine on non-large bar
 * enabled machines.
 */
extern int debug_largebar;

175 176 177 178 179 180
/*
 * Ignore CRAT table during KFD initialization, can be used to work around
 * broken CRAT tables on some AMD systems
 */
extern int ignore_crat;

181
/* Set sh_mem_config.retry_disable on GFX v9 */
182
extern int amdgpu_noretry;
183

184
/* Halt if HWS hang is detected */
185 186
extern int halt_if_hws_hang;

187
/* Whether MEC FW support GWS barriers */
188 189
extern bool hws_gws_support;

190
/* Queue preemption timeout in ms */
191 192
extern int queue_preemption_timeout_ms;

193 194 195 196 197
/*
 * Don't evict process queues on vm fault
 */
extern int amdgpu_no_queue_eviction_on_vm_fault;

198
/* Enable eviction debug messages */
199 200
extern bool debug_evictions;

B
Ben Goz 已提交
201 202 203 204 205
enum cache_policy {
	cache_policy_coherent,
	cache_policy_noncoherent
};

206 207
#define KFD_GC_VERSION(dev) ((dev)->adev->ip_versions[GC_HWIP][0])
#define KFD_IS_SOC15(dev)   ((KFD_GC_VERSION(dev)) >= (IP_VERSION(9, 0, 1)))
208

209 210
struct kfd_event_interrupt_class {
	bool (*interrupt_isr)(struct kfd_dev *dev,
211 212
			const uint32_t *ih_ring_entry, uint32_t *patched_ihre,
			bool *patched_flag);
213
	void (*interrupt_wq)(struct kfd_dev *dev,
214
			const uint32_t *ih_ring_entry);
215 216
};

O
Oded Gabbay 已提交
217
struct kfd_device_info {
218
	uint32_t gfx_target_version;
219
	const struct kfd_event_interrupt_class *event_interrupt_class;
O
Oded Gabbay 已提交
220
	unsigned int max_pasid_bits;
221
	unsigned int max_no_of_hqd;
222
	unsigned int doorbell_size;
O
Oded Gabbay 已提交
223
	size_t ih_ring_entry_size;
224
	uint8_t num_of_watch_points;
225
	uint16_t mqd_size_aligned;
F
Felix Kuehling 已提交
226
	bool supports_cwsr;
227
	bool needs_iommu_device;
228
	bool needs_pci_atomics;
229
	uint32_t no_atomic_fw_version;
230
	unsigned int num_sdma_queues_per_engine;
O
Oded Gabbay 已提交
231 232
};

233 234 235
unsigned int kfd_get_num_sdma_engines(struct kfd_dev *kdev);
unsigned int kfd_get_num_xgmi_sdma_engines(struct kfd_dev *kdev);

236 237 238 239 240
struct kfd_mem_obj {
	uint32_t range_start;
	uint32_t range_end;
	uint64_t gpu_addr;
	uint32_t *cpu_ptr;
241
	void *gtt_mem;
242 243
};

244 245 246 247 248 249
struct kfd_vmid_info {
	uint32_t first_vmid_kfd;
	uint32_t last_vmid_kfd;
	uint32_t vmid_num_kfd;
};

O
Oded Gabbay 已提交
250
struct kfd_dev {
251
	struct amdgpu_device *adev;
O
Oded Gabbay 已提交
252

253
	struct kfd_device_info device_info;
O
Oded Gabbay 已提交
254
	struct pci_dev *pdev;
255
	struct drm_device *ddev;
O
Oded Gabbay 已提交
256 257 258

	unsigned int id;		/* topology stub index */

259 260 261 262
	phys_addr_t doorbell_base;	/* Start of actual doorbells used by
					 * KFD. It is aligned for mapping
					 * into user mode
					 */
263 264 265 266
	size_t doorbell_base_dw_offset;	/* Offset from the start of the PCI
					 * doorbell BAR to the first KFD
					 * doorbell in dwords. GFX reserves
					 * the segment before this offset.
267 268 269 270 271
					 */
	u32 __iomem *doorbell_kernel_ptr; /* This is a pointer for a doorbells
					   * page used by kernel queue
					   */

O
Oded Gabbay 已提交
272
	struct kgd2kfd_shared_resources shared_resources;
273
	struct kfd_vmid_info vm_info;
O
Oded Gabbay 已提交
274

275 276
	const struct kfd2kgd_calls *kfd2kgd;
	struct mutex doorbell_mutex;
J
Joe Perches 已提交
277 278
	DECLARE_BITMAP(doorbell_available_index,
			KFD_MAX_NUM_OF_QUEUES_PER_PROCESS);
279

280 281 282 283 284 285 286 287
	void *gtt_mem;
	uint64_t gtt_start_gpu_addr;
	void *gtt_start_cpu_ptr;
	void *gtt_sa_bitmap;
	struct mutex gtt_sa_lock;
	unsigned int gtt_sa_chunk_size;
	unsigned int gtt_sa_num_of_chunks;

288
	/* Interrupts */
289
	struct kfifo ih_fifo;
290
	struct workqueue_struct *ih_wq;
291 292 293
	struct work_struct interrupt_work;
	spinlock_t interrupt_lock;

B
Ben Goz 已提交
294 295
	/* QCM Device instance */
	struct device_queue_manager *dqm;
O
Oded Gabbay 已提交
296

B
Ben Goz 已提交
297
	bool init_complete;
298 299 300 301 302
	/*
	 * Interrupts of interest to KFD are copied
	 * from the HW ring into a SW ring.
	 */
	bool interrupts_active;
303

304 305
	/* Firmware versions */
	uint16_t mec_fw_version;
306
	uint16_t mec2_fw_version;
307 308
	uint16_t sdma_fw_version;

309 310 311
	/* Maximum process number mapped to HW scheduler */
	unsigned int max_proc_per_quantum;

F
Felix Kuehling 已提交
312 313 314 315
	/* CWSR */
	bool cwsr_enabled;
	const void *cwsr_isa;
	unsigned int cwsr_isa_size;
316 317 318

	/* xGMI */
	uint64_t hive_id;
319

320
	bool pci_atomic_requested;
321

322 323 324
	/* Use IOMMU v2 flag */
	bool use_iommu_v2;

325 326
	/* SRAM ECC flag */
	atomic_t sram_ecc_flag;
327 328 329

	/* Compute Profile ref. count */
	atomic_t compute_profile;
330

331
	/* Global GWS resource shared between processes */
332
	void *gws;
A
Amber Lin 已提交
333 334 335 336

	/* Clients watching SMI events */
	struct list_head smi_clients;
	spinlock_t smi_lock;
337 338

	uint32_t reset_seq_num;
339 340 341

	struct ida doorbell_ida;
	unsigned int max_doorbell_slices;
342 343

	int noretry;
344 345 346

	/* HMM page migration MEMORY_DEVICE_PRIVATE mapping */
	struct dev_pagemap pgmap;
O
Oded Gabbay 已提交
347 348
};

349 350 351 352 353 354
enum kfd_mempool {
	KFD_MEMPOOL_SYSTEM_CACHEABLE = 1,
	KFD_MEMPOOL_SYSTEM_WRITECOMBINE = 2,
	KFD_MEMPOOL_FRAMEBUFFER = 3,
};

O
Oded Gabbay 已提交
355 356 357 358
/* Character device interface */
int kfd_chardev_init(void);
void kfd_chardev_exit(void);

B
Ben Goz 已提交
359
/**
360
 * enum kfd_unmap_queues_filter - Enum for queue filters.
B
Ben Goz 已提交
361
 *
362
 * @KFD_UNMAP_QUEUES_FILTER_ALL_QUEUES: Preempts all queues in the
B
Ben Goz 已提交
363 364
 *						running queues list.
 *
365 366 367
 * @KFD_UNMAP_QUEUES_FILTER_DYNAMIC_QUEUES: Preempts all non-static queues
 *						in the run list.
 *
368
 * @KFD_UNMAP_QUEUES_FILTER_BY_PASID: Preempts queues that belongs to
B
Ben Goz 已提交
369 370 371
 *						specific process.
 *
 */
372
enum kfd_unmap_queues_filter {
373 374 375
	KFD_UNMAP_QUEUES_FILTER_ALL_QUEUES = 1,
	KFD_UNMAP_QUEUES_FILTER_DYNAMIC_QUEUES = 2,
	KFD_UNMAP_QUEUES_FILTER_BY_PASID = 3
B
Ben Goz 已提交
376
};
377

B
Ben Goz 已提交
378
/**
379
 * enum kfd_queue_type - Enum for various queue types.
B
Ben Goz 已提交
380 381 382
 *
 * @KFD_QUEUE_TYPE_COMPUTE: Regular user mode queue type.
 *
383
 * @KFD_QUEUE_TYPE_SDMA: SDMA user mode queue type.
B
Ben Goz 已提交
384 385 386 387
 *
 * @KFD_QUEUE_TYPE_HIQ: HIQ queue type.
 *
 * @KFD_QUEUE_TYPE_DIQ: DIQ queue type.
388 389
 *
 * @KFD_QUEUE_TYPE_SDMA_XGMI: Special SDMA queue for XGMI interface.
B
Ben Goz 已提交
390 391 392 393 394
 */
enum kfd_queue_type  {
	KFD_QUEUE_TYPE_COMPUTE,
	KFD_QUEUE_TYPE_SDMA,
	KFD_QUEUE_TYPE_HIQ,
395 396
	KFD_QUEUE_TYPE_DIQ,
	KFD_QUEUE_TYPE_SDMA_XGMI
B
Ben Goz 已提交
397 398
};

B
Ben Goz 已提交
399 400 401 402 403
enum kfd_queue_format {
	KFD_QUEUE_FORMAT_PM4,
	KFD_QUEUE_FORMAT_AQL
};

404 405 406 407 408
enum KFD_QUEUE_PRIORITY {
	KFD_QUEUE_PRIORITY_MINIMUM = 0,
	KFD_QUEUE_PRIORITY_MAXIMUM = 15
};

B
Ben Goz 已提交
409 410 411 412 413 414 415 416 417 418 419 420 421 422 423 424 425 426 427 428 429 430 431 432 433 434
/**
 * struct queue_properties
 *
 * @type: The queue type.
 *
 * @queue_id: Queue identifier.
 *
 * @queue_address: Queue ring buffer address.
 *
 * @queue_size: Queue ring buffer size.
 *
 * @priority: Defines the queue priority relative to other queues in the
 * process.
 * This is just an indication and HW scheduling may override the priority as
 * necessary while keeping the relative prioritization.
 * the priority granularity is from 0 to f which f is the highest priority.
 * currently all queues are initialized with the highest priority.
 *
 * @queue_percent: This field is partially implemented and currently a zero in
 * this field defines that the queue is non active.
 *
 * @read_ptr: User space address which points to the number of dwords the
 * cp read from the ring buffer. This field updates automatically by the H/W.
 *
 * @write_ptr: Defines the number of dwords written to the ring buffer.
 *
435 436 437
 * @doorbell_ptr: Notifies the H/W of new packet written to the queue ring
 * buffer. This field should be similar to write_ptr and the user should
 * update this field after updating the write_ptr.
B
Ben Goz 已提交
438 439 440
 *
 * @doorbell_off: The doorbell offset in the doorbell pci-bar.
 *
441 442
 * @is_interop: Defines if this is a interop queue. Interop queue means that
 * the queue can access both graphics and compute resources.
B
Ben Goz 已提交
443
 *
444 445 446 447 448
 * @is_evicted: Defines if the queue is evicted. Only active queues
 * are evicted, rendering them inactive.
 *
 * @is_active: Defines if the queue is active or not. @is_active and
 * @is_evicted are protected by the DQM lock.
B
Ben Goz 已提交
449
 *
450 451 452 453
 * @is_gws: Defines if the queue has been updated to be GWS-capable or not.
 * @is_gws should be protected by the DQM lock, since changing it can yield the
 * possibility of updating DQM state on number of GWS queues.
 *
B
Ben Goz 已提交
454 455 456 457 458 459 460
 * @vmid: If the scheduling mode is no cp scheduling the field defines the vmid
 * of the queue.
 *
 * This structure represents the queue properties for each queue no matter if
 * it's user mode or kernel mode queue.
 *
 */
461

B
Ben Goz 已提交
462 463
struct queue_properties {
	enum kfd_queue_type type;
B
Ben Goz 已提交
464
	enum kfd_queue_format format;
B
Ben Goz 已提交
465 466 467 468 469 470 471
	unsigned int queue_id;
	uint64_t queue_address;
	uint64_t  queue_size;
	uint32_t priority;
	uint32_t queue_percent;
	uint32_t *read_ptr;
	uint32_t *write_ptr;
472
	void __iomem *doorbell_ptr;
B
Ben Goz 已提交
473 474
	uint32_t doorbell_off;
	bool is_interop;
475
	bool is_evicted;
B
Ben Goz 已提交
476
	bool is_active;
477
	bool is_gws;
B
Ben Goz 已提交
478 479
	/* Not relevant for user mode queues in cp scheduling */
	unsigned int vmid;
B
Ben Goz 已提交
480 481 482 483
	/* Relevant only for sdma queues*/
	uint32_t sdma_engine_id;
	uint32_t sdma_queue_id;
	uint32_t sdma_vm_addr;
484 485 486 487 488
	/* Relevant only for VI */
	uint64_t eop_ring_buffer_address;
	uint32_t eop_ring_buffer_size;
	uint64_t ctx_save_restore_area_address;
	uint32_t ctx_save_restore_area_size;
F
Felix Kuehling 已提交
489 490 491
	uint32_t ctl_stack_size;
	uint64_t tba_addr;
	uint64_t tma_addr;
B
Ben Goz 已提交
492 493
};

494 495 496 497 498
#define QUEUE_IS_ACTIVE(q) ((q).queue_size > 0 &&	\
			    (q).queue_address != 0 &&	\
			    (q).queue_percent > 0 &&	\
			    !(q).is_evicted)

499 500 501 502 503 504 505 506 507 508 509 510 511
enum mqd_update_flag {
	UPDATE_FLAG_CU_MASK = 0,
};

struct mqd_update_info {
	union {
		struct {
			uint32_t count; /* Must be a multiple of 32 */
			uint32_t *ptr;
		} cu_mask;
	};
	enum mqd_update_flag update_flag;
};
512

B
Ben Goz 已提交
513 514 515 516 517
/**
 * struct queue
 *
 * @list: Queue linked list.
 *
518
 * @mqd: The queue MQD (memory queue descriptor).
B
Ben Goz 已提交
519 520 521 522 523 524 525 526
 *
 * @mqd_mem_obj: The MQD local gpu memory object.
 *
 * @gart_mqd_addr: The MQD gart mc address.
 *
 * @properties: The queue properties.
 *
 * @mec: Used only in no cp scheduling mode and identifies to micro engine id
527
 *	 that the queue should be executed on.
B
Ben Goz 已提交
528
 *
529 530
 * @pipe: Used only in no cp scheduling mode and identifies the queue's pipe
 *	  id.
B
Ben Goz 已提交
531 532 533 534 535 536 537
 *
 * @queue: Used only in no cp scheduliong mode and identifies the queue's slot.
 *
 * @process: The kfd process that created this queue.
 *
 * @device: The kfd device that created this queue.
 *
538 539 540
 * @gws: Pointing to gws kgd_mem if this is a gws control queue; NULL
 * otherwise.
 *
B
Ben Goz 已提交
541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556
 * This structure represents user mode compute queues.
 * It contains all the necessary data to handle such queues.
 *
 */

struct queue {
	struct list_head list;
	void *mqd;
	struct kfd_mem_obj *mqd_mem_obj;
	uint64_t gart_mqd_addr;
	struct queue_properties properties;

	uint32_t mec;
	uint32_t pipe;
	uint32_t queue;

B
Ben Goz 已提交
557
	unsigned int sdma_id;
558
	unsigned int doorbell_id;
B
Ben Goz 已提交
559

B
Ben Goz 已提交
560 561
	struct kfd_process	*process;
	struct kfd_dev		*device;
562
	void *gws;
563 564 565

	/* procfs */
	struct kobject kobj;
B
Ben Goz 已提交
566 567
};

B
Ben Goz 已提交
568
enum KFD_MQD_TYPE {
569
	KFD_MQD_TYPE_HIQ = 0,		/* for hiq */
570 571
	KFD_MQD_TYPE_CP,		/* for cp queues and diq */
	KFD_MQD_TYPE_SDMA,		/* for sdma queues */
572
	KFD_MQD_TYPE_DIQ,		/* for diq */
B
Ben Goz 已提交
573 574 575
	KFD_MQD_TYPE_MAX
};

576 577 578 579 580 581
enum KFD_PIPE_PRIORITY {
	KFD_PIPE_PRIORITY_CS_LOW = 0,
	KFD_PIPE_PRIORITY_CS_MEDIUM,
	KFD_PIPE_PRIORITY_CS_HIGH
};

B
Ben Goz 已提交
582 583 584 585 586 587 588 589 590 591 592 593 594 595 596 597 598 599 600 601 602 603 604 605 606 607 608 609
struct scheduling_resources {
	unsigned int vmid_mask;
	enum kfd_queue_type type;
	uint64_t queue_mask;
	uint64_t gws_mask;
	uint32_t oac_mask;
	uint32_t gds_heap_base;
	uint32_t gds_heap_size;
};

struct process_queue_manager {
	/* data */
	struct kfd_process	*process;
	struct list_head	queues;
	unsigned long		*queue_slot_bitmap;
};

struct qcm_process_device {
	/* The Device Queue Manager that owns this data */
	struct device_queue_manager *dqm;
	struct process_queue_manager *pqm;
	/* Queues list */
	struct list_head queues_list;
	struct list_head priv_queue_list;

	unsigned int queue_count;
	unsigned int vmid;
	bool is_debug;
610
	unsigned int evicted; /* eviction counter, 0=active */
611 612 613 614 615 616

	/* This flag tells if we should reset all wavefronts on
	 * process termination
	 */
	bool reset_wavefronts;

617 618 619 620 621 622 623 624
	/* This flag tells us if this process has a GWS-capable
	 * queue that will be mapped into the runlist. It's
	 * possible to request a GWS BO, but not have the queue
	 * currently mapped, and this changes how the MAP_PROCESS
	 * PM4 packet is configured.
	 */
	bool mapped_gws_queue;

625
	/* All the memory management data should be here too */
B
Ben Goz 已提交
626
	uint64_t gds_context_area;
627
	/* Contains page table flags such as AMDGPU_PTE_VALID since gfx9 */
628
	uint64_t page_table_base;
B
Ben Goz 已提交
629 630 631 632 633 634 635
	uint32_t sh_mem_config;
	uint32_t sh_mem_bases;
	uint32_t sh_mem_ape1_base;
	uint32_t sh_mem_ape1_limit;
	uint32_t gds_size;
	uint32_t num_gws;
	uint32_t num_oac;
636
	uint32_t sh_hidden_private_base;
F
Felix Kuehling 已提交
637 638

	/* CWSR memory */
639
	struct kgd_mem *cwsr_mem;
F
Felix Kuehling 已提交
640
	void *cwsr_kaddr;
641
	uint64_t cwsr_base;
F
Felix Kuehling 已提交
642 643
	uint64_t tba_addr;
	uint64_t tma_addr;
644 645

	/* IB memory */
646
	struct kgd_mem *ib_mem;
647
	uint64_t ib_base;
648
	void *ib_kaddr;
649 650 651

	/* doorbell resources per process per device */
	unsigned long *doorbell_bitmap;
B
Ben Goz 已提交
652 653
};

654 655 656 657 658 659 660 661 662
/* KFD Memory Eviction */

/* Approx. wait time before attempting to restore evicted BOs */
#define PROCESS_RESTORE_TIME_MS 100
/* Approx. back off time if restore fails due to lack of memory */
#define PROCESS_BACK_OFF_TIME_MS 100
/* Approx. time before evicting the process again */
#define PROCESS_ACTIVE_TIME_MS 10

663 664 665 666 667 668 669 670
/* 8 byte handle containing GPU ID in the most significant 4 bytes and
 * idr_handle in the least significant 4 bytes
 */
#define MAKE_HANDLE(gpu_id, idr_handle) \
	(((uint64_t)(gpu_id) << 32) + idr_handle)
#define GET_GPU_ID(handle) (handle >> 32)
#define GET_IDR_HANDLE(handle) (handle & 0xFFFFFFFF)

671 672 673 674 675 676
enum kfd_pdd_bound {
	PDD_UNBOUND = 0,
	PDD_BOUND,
	PDD_BOUND_SUSPENDED,
};

677
#define MAX_SYSFS_FILENAME_LEN 15
678 679 680 681 682 683 684

/*
 * SDMA counter runs at 100MHz frequency.
 * We display SDMA activity in microsecond granularity in sysfs.
 * As a result, the divisor is 100.
 */
#define SDMA_ACTIVITY_DIVISOR  100
685

686 687 688 689 690
/* Data that is per-process-per device. */
struct kfd_process_device {
	/* The device that owns this data. */
	struct kfd_dev *dev;

691 692
	/* The process that owns this kfd_process_device. */
	struct kfd_process *process;
693

694 695 696
	/* per-process-per device QCM data structure */
	struct qcm_process_device qpd;

697 698 699 700 701 702 703 704
	/*Apertures*/
	uint64_t lds_base;
	uint64_t lds_limit;
	uint64_t gpuvm_base;
	uint64_t gpuvm_limit;
	uint64_t scratch_base;
	uint64_t scratch_limit;

705
	/* VM context for GPUVM allocations */
706
	struct file *drm_file;
707
	void *drm_priv;
708

709 710 711
	/* GPUVM allocations storage */
	struct idr alloc_idr;

712 713 714 715
	/* Flag used to tell the pdd has dequeued from the dqm.
	 * This is used to prevent dev->dqm->ops.process_termination() from
	 * being called twice when it is already called in IOMMU callback
	 * function.
716
	 */
717
	bool already_dequeued;
718
	bool runtime_inuse;
719 720 721

	/* Is this process/pasid bound to this device? (amd_iommu_bind_pasid) */
	enum kfd_pdd_bound bound;
722 723 724 725

	/* VRAM usage */
	uint64_t vram_usage;
	struct attribute attr_vram;
726 727 728 729 730 731
	char vram_filename[MAX_SYSFS_FILENAME_LEN];

	/* SDMA activity tracking */
	uint64_t sdma_past_activity_counter;
	struct attribute attr_sdma;
	char sdma_filename[MAX_SYSFS_FILENAME_LEN];
732 733 734 735 736 737 738

	/* Eviction activity tracking */
	uint64_t last_evict_timestamp;
	atomic64_t evict_duration_counter;
	struct attribute attr_evict;

	struct kobject *kobj_stats;
739
	unsigned int doorbell_index;
740 741 742 743 744 745 746 747 748 749 750 751 752 753 754 755 756 757 758 759 760 761 762 763 764

	/*
	 * @cu_occupancy: Reports occupancy of Compute Units (CU) of a process
	 * that is associated with device encoded by "this" struct instance. The
	 * value reflects CU usage by all of the waves launched by this process
	 * on this device. A very important property of occupancy parameter is
	 * that its value is a snapshot of current use.
	 *
	 * Following is to be noted regarding how this parameter is reported:
	 *
	 *  The number of waves that a CU can launch is limited by couple of
	 *  parameters. These are encoded by struct amdgpu_cu_info instance
	 *  that is part of every device definition. For GFX9 devices this
	 *  translates to 40 waves (simd_per_cu * max_waves_per_simd) when waves
	 *  do not use scratch memory and 32 waves (max_scratch_slots_per_cu)
	 *  when they do use scratch memory. This could change for future
	 *  devices and therefore this example should be considered as a guide.
	 *
	 *  All CU's of a device are available for the process. This may not be true
	 *  under certain conditions - e.g. CU masking.
	 *
	 *  Finally number of CU's that are occupied by a process is affected by both
	 *  number of CU's a device has along with number of other competing processes
	 */
	struct attribute attr_cu_occupancy;
765 766 767 768 769 770 771 772 773

	/* sysfs counters for GPU retry fault and page migration tracking */
	struct kobject *kobj_counters;
	struct attribute attr_faults;
	struct attribute attr_page_in;
	struct attribute attr_page_out;
	uint64_t faults;
	uint64_t page_in;
	uint64_t page_out;
774 775 776 777 778 779
	/*
	 * If this process has been checkpointed before, then the user
	 * application will use the original gpu_id on the
	 * checkpointed node to refer to this device.
	 */
	uint32_t user_gpu_id;
780 781
};

782 783
#define qpd_to_pdd(x) container_of(x, struct kfd_process_device, qpd)

P
Philip Yang 已提交
784 785 786 787
struct svm_range_list {
	struct mutex			lock;
	struct rb_root_cached		objects;
	struct list_head		list;
P
Philip Yang 已提交
788 789
	struct work_struct		deferred_list_work;
	struct list_head		deferred_range_list;
790
	struct list_head                criu_svm_metadata_list;
P
Philip Yang 已提交
791
	spinlock_t			deferred_list_lock;
792
	atomic_t			evicted_ranges;
793
	atomic_t			drain_pagefaults;
794
	struct delayed_work		restore_work;
795
	DECLARE_BITMAP(bitmap_supported, MAX_GPU_INSTANCE);
796
	struct task_struct		*faulting_task;
P
Philip Yang 已提交
797 798
};

O
Oded Gabbay 已提交
799 800
/* Process data */
struct kfd_process {
801 802 803 804 805 806
	/*
	 * kfd_process are stored in an mm_struct*->kfd_process*
	 * hash table (kfd_processes in kfd_process.c)
	 */
	struct hlist_node kfd_processes;

807 808 809 810 811 812
	/*
	 * Opaque pointer to mm_struct. We don't hold a reference to
	 * it so it should never be dereferenced from here. This is
	 * only used for looking up processes by their mm.
	 */
	void *mm;
813

814 815 816
	struct kref ref;
	struct work_struct release_work;

817 818 819 820 821 822
	struct mutex mutex;

	/*
	 * In any process, the thread that started main() is the lead
	 * thread and outlives the rest.
	 * It is here because amd_iommu_bind_pasid wants a task_struct.
F
Felix Kuehling 已提交
823 824
	 * It can also be used for safely getting a reference to the
	 * mm_struct of the process.
825 826 827 828 829 830
	 */
	struct task_struct *lead_thread;

	/* We want to receive a notification when the mm_struct is destroyed */
	struct mmu_notifier mmu_notifier;

831
	u32 pasid;
832 833

	/*
834
	 * Array of kfd_process_device pointers,
835 836
	 * one for each device the process is using.
	 */
837 838
	struct kfd_process_device *pdds[MAX_GPU_INSTANCE];
	uint32_t n_pdds;
839

840 841
	struct process_queue_manager pqm;

842 843
	/*Is the user space process 32 bit?*/
	bool is_32bit_user_mode;
844 845 846

	/* Event-related data */
	struct mutex event_mutex;
847 848
	/* Event ID allocator and lookup */
	struct idr event_idr;
849
	/* Event page */
850
	u64 signal_handle;
851
	struct kfd_signal_page *signal_page;
852
	size_t signal_mapped_size;
853
	size_t signal_event_count;
854
	bool signal_event_limit_reached;
855 856 857 858 859 860 861 862

	/* Information used for memory eviction */
	void *kgd_process_info;
	/* Eviction fence that is attached to all the BOs of this process. The
	 * fence will be triggered during eviction and new one will be created
	 * during restore
	 */
	struct dma_fence *ef;
863 864 865 866 867 868 869 870 871 872

	/* Work items for evicting and restoring BOs */
	struct delayed_work eviction_work;
	struct delayed_work restore_work;
	/* seqno of the last scheduled eviction */
	unsigned int last_eviction_seqno;
	/* Approx. the last timestamp (in jiffies) when the process was
	 * restored after an eviction
	 */
	unsigned long last_restore_timestamp;
873 874 875

	/* Kobj for our procfs */
	struct kobject *kobj;
876
	struct kobject *kobj_queues;
877
	struct attribute attr_pasid;
P
Philip Yang 已提交
878

P
Philip Yang 已提交
879 880
	/* shared virtual memory registered by this process */
	struct svm_range_list svms;
881 882

	bool xnack_enabled;
883 884

	atomic_t poison;
885 886
	/* Queues are in paused stated because we are in the process of doing a CRIU checkpoint */
	bool queues_paused;
O
Oded Gabbay 已提交
887 888
};

889 890 891 892
#define KFD_PROCESS_TABLE_SIZE 5 /* bits: 32 entries */
extern DECLARE_HASHTABLE(kfd_processes_table, KFD_PROCESS_TABLE_SIZE);
extern struct srcu_struct kfd_processes_srcu;

893
/**
894 895 896 897 898
 * typedef amdkfd_ioctl_t - typedef for ioctl function pointer.
 *
 * @filep: pointer to file structure.
 * @p: amdkfd process pointer.
 * @data: pointer to arg that was copied from user.
899
 *
900
 * Return: returns ioctl completion code.
901 902 903 904 905 906 907 908 909 910 911
 */
typedef int amdkfd_ioctl_t(struct file *filep, struct kfd_process *p,
				void *data);

struct amdkfd_ioctl_desc {
	unsigned int cmd;
	int flags;
	amdkfd_ioctl_t *func;
	unsigned int cmd_drv;
	const char *name;
};
912
bool kfd_dev_is_large_bar(struct kfd_dev *dev);
913

914
int kfd_process_create_wq(void);
915
void kfd_process_destroy_wq(void);
F
Felix Kuehling 已提交
916
struct kfd_process *kfd_create_process(struct file *filep);
917
struct kfd_process *kfd_get_process(const struct task_struct *task);
918
struct kfd_process *kfd_lookup_process_by_pasid(u32 pasid);
919
struct kfd_process *kfd_lookup_process_by_mm(const struct mm_struct *mm);
920 921

int kfd_process_gpuidx_from_gpuid(struct kfd_process *p, uint32_t gpu_id);
922
int kfd_process_gpuid_from_adev(struct kfd_process *p,
923 924
			       struct amdgpu_device *adev, uint32_t *gpuid,
			       uint32_t *gpuidx);
925 926 927 928 929 930 931 932 933
static inline int kfd_process_gpuid_from_gpuidx(struct kfd_process *p,
				uint32_t gpuidx, uint32_t *gpuid) {
	return gpuidx < p->n_pdds ? p->pdds[gpuidx]->dev->id : -EINVAL;
}
static inline struct kfd_process_device *kfd_process_device_from_gpuidx(
				struct kfd_process *p, uint32_t gpuidx) {
	return gpuidx < p->n_pdds ? p->pdds[gpuidx] : NULL;
}

934
void kfd_unref_process(struct kfd_process *p);
935 936
int kfd_process_evict_queues(struct kfd_process *p);
int kfd_process_restore_queues(struct kfd_process *p);
937 938
void kfd_suspend_all_processes(void);
int kfd_resume_all_processes(void);
939

940 941 942 943 944
struct kfd_process_device *kfd_process_device_data_by_id(struct kfd_process *process,
							 uint32_t gpu_id);

int kfd_process_get_user_gpu_id(struct kfd_process *p, uint32_t actual_gpu_id);

945 946
int kfd_process_device_init_vm(struct kfd_process_device *pdd,
			       struct file *drm_file);
947
struct kfd_process_device *kfd_bind_process_to_device(struct kfd_dev *dev,
948
						struct kfd_process *p);
949
struct kfd_process_device *kfd_get_process_device_data(struct kfd_dev *dev,
950 951 952
							struct kfd_process *p);
struct kfd_process_device *kfd_create_process_device_data(struct kfd_dev *dev,
							struct kfd_process *p);
953

954 955
bool kfd_process_xnack_mode(struct kfd_process *p, bool supported);

956
int kfd_reserved_mem_mmap(struct kfd_dev *dev, struct kfd_process *process,
F
Felix Kuehling 已提交
957 958
			  struct vm_area_struct *vma);

959 960 961 962 963 964 965
/* KFD process API for creating and translating handles */
int kfd_process_device_create_obj_handle(struct kfd_process_device *pdd,
					void *mem);
void *kfd_process_device_translate_handle(struct kfd_process_device *p,
					int handle);
void kfd_process_device_remove_obj_handle(struct kfd_process_device *pdd,
					int handle);
966
struct kfd_process *kfd_lookup_process_by_pid(struct pid *pid);
967

968 969 970 971 972
/* PASIDs */
int kfd_pasid_init(void);
void kfd_pasid_exit(void);
bool kfd_set_pasid_limit(unsigned int new_limit);
unsigned int kfd_get_pasid_limit(void);
973 974
u32 kfd_pasid_alloc(void);
void kfd_pasid_free(u32 pasid);
975 976

/* Doorbells */
977
size_t kfd_doorbell_process_slice(struct kfd_dev *kfd);
978 979
int kfd_doorbell_init(struct kfd_dev *kfd);
void kfd_doorbell_fini(struct kfd_dev *kfd);
980 981
int kfd_doorbell_mmap(struct kfd_dev *dev, struct kfd_process *process,
		      struct vm_area_struct *vma);
982
void __iomem *kfd_get_kernel_doorbell(struct kfd_dev *kfd,
983 984 985
					unsigned int *doorbell_off);
void kfd_release_kernel_doorbell(struct kfd_dev *kfd, u32 __iomem *db_addr);
u32 read_kernel_doorbell(u32 __iomem *db);
986
void write_kernel_doorbell(void __iomem *db, u32 value);
987
void write_kernel_doorbell64(void __iomem *db, u64 value);
988
unsigned int kfd_get_doorbell_dw_offset_in_bar(struct kfd_dev *kfd,
989
					struct kfd_process_device *pdd,
990
					unsigned int doorbell_id);
991 992 993 994 995
phys_addr_t kfd_get_process_doorbells(struct kfd_process_device *pdd);
int kfd_alloc_process_doorbells(struct kfd_dev *kfd,
				unsigned int *doorbell_index);
void kfd_free_process_doorbells(struct kfd_dev *kfd,
				unsigned int doorbell_index);
996 997 998 999 1000 1001 1002
/* GTT Sub-Allocator */

int kfd_gtt_sa_allocate(struct kfd_dev *kfd, unsigned int size,
			struct kfd_mem_obj **mem_obj);

int kfd_gtt_sa_free(struct kfd_dev *kfd, struct kfd_mem_obj *mem_obj);

O
Oded Gabbay 已提交
1003 1004
extern struct device *kfd_device;

1005 1006 1007
/* KFD's procfs */
void kfd_procfs_init(void);
void kfd_procfs_shutdown(void);
1008 1009
int kfd_procfs_add_queue(struct queue *q);
void kfd_procfs_del_queue(struct queue *q);
1010

1011 1012 1013 1014 1015
/* Topology */
int kfd_topology_init(void);
void kfd_topology_shutdown(void);
int kfd_topology_add_device(struct kfd_dev *gpu);
int kfd_topology_remove_device(struct kfd_dev *gpu);
1016 1017
struct kfd_topology_device *kfd_topology_device_by_proximity_domain(
						uint32_t proximity_domain);
1018
struct kfd_topology_device *kfd_topology_device_by_id(uint32_t gpu_id);
1019 1020
struct kfd_dev *kfd_device_by_id(uint32_t gpu_id);
struct kfd_dev *kfd_device_by_pci_dev(const struct pci_dev *pdev);
1021
struct kfd_dev *kfd_device_by_adev(const struct amdgpu_device *adev);
1022
int kfd_topology_enum_kfd_devices(uint8_t idx, struct kfd_dev **kdev);
1023
int kfd_numa_node_to_apic_id(int numa_node_id);
1024
void kfd_double_confirm_iommu_support(struct kfd_dev *gpu);
1025

O
Oded Gabbay 已提交
1026
/* Interrupts */
1027 1028 1029
int kfd_interrupt_init(struct kfd_dev *dev);
void kfd_interrupt_exit(struct kfd_dev *dev);
bool enqueue_ih_ring_entry(struct kfd_dev *kfd,	const void *ih_ring_entry);
1030 1031 1032
bool interrupt_is_wanted(struct kfd_dev *dev,
				const uint32_t *ih_ring_entry,
				uint32_t *patched_ihre, bool *flag);
O
Oded Gabbay 已提交
1033

1034 1035 1036
/* amdkfd Apertures */
int kfd_init_apertures(struct kfd_process *process);

1037 1038 1039 1040
void kfd_process_set_trap_handler(struct qcm_process_device *qpd,
				  uint64_t tba_addr,
				  uint64_t tma_addr);

1041 1042 1043 1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
/* CRIU */
/*
 * Need to increment KFD_CRIU_PRIV_VERSION each time a change is made to any of the CRIU private
 * structures:
 * kfd_criu_process_priv_data
 * kfd_criu_device_priv_data
 * kfd_criu_bo_priv_data
 * kfd_criu_queue_priv_data
 * kfd_criu_event_priv_data
 * kfd_criu_svm_range_priv_data
 */

#define KFD_CRIU_PRIV_VERSION 1

struct kfd_criu_process_priv_data {
	uint32_t version;
1057
	uint32_t xnack_mode;
1058 1059 1060 1061 1062 1063 1064 1065
};

struct kfd_criu_device_priv_data {
	/* For future use */
	uint64_t reserved;
};

struct kfd_criu_bo_priv_data {
1066 1067 1068
	uint64_t user_addr;
	uint32_t idr_handle;
	uint32_t mapped_gpuids[MAX_GPU_INSTANCE];
1069 1070
};

1071 1072 1073 1074 1075 1076 1077 1078 1079 1080
/*
 * The first 4 bytes of kfd_criu_queue_priv_data, kfd_criu_event_priv_data,
 * kfd_criu_svm_range_priv_data is the object type
 */
enum kfd_criu_object_type {
	KFD_CRIU_OBJECT_TYPE_QUEUE,
	KFD_CRIU_OBJECT_TYPE_EVENT,
	KFD_CRIU_OBJECT_TYPE_SVM_RANGE,
};

1081 1082
struct kfd_criu_svm_range_priv_data {
	uint32_t object_type;
1083 1084 1085
	uint64_t start_addr;
	uint64_t size;
	/* Variable length array of attributes */
1086
	struct kfd_ioctl_svm_attribute attrs[];
1087 1088 1089 1090
};

struct kfd_criu_queue_priv_data {
	uint32_t object_type;
1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110
	uint64_t q_address;
	uint64_t q_size;
	uint64_t read_ptr_addr;
	uint64_t write_ptr_addr;
	uint64_t doorbell_off;
	uint64_t eop_ring_buffer_address;
	uint64_t ctx_save_restore_area_address;
	uint32_t gpu_id;
	uint32_t type;
	uint32_t format;
	uint32_t q_id;
	uint32_t priority;
	uint32_t q_percent;
	uint32_t doorbell_id;
	uint32_t is_gws;
	uint32_t sdma_id;
	uint32_t eop_ring_buffer_size;
	uint32_t ctx_save_restore_area_size;
	uint32_t ctl_stack_size;
	uint32_t mqd_size;
1111 1112 1113 1114
};

struct kfd_criu_event_priv_data {
	uint32_t object_type;
1115 1116 1117 1118 1119 1120 1121 1122 1123 1124
	uint64_t user_handle;
	uint32_t event_id;
	uint32_t auto_reset;
	uint32_t type;
	uint32_t signaled;

	union {
		struct kfd_hsa_memory_exception_data memory_exception_data;
		struct kfd_hsa_hw_exception_data hw_exception_data;
	};
1125 1126
};

1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137 1138
int kfd_process_get_queue_info(struct kfd_process *p,
			       uint32_t *num_queues,
			       uint64_t *priv_data_sizes);

int kfd_criu_checkpoint_queues(struct kfd_process *p,
			 uint8_t __user *user_priv_data,
			 uint64_t *priv_data_offset);

int kfd_criu_restore_queue(struct kfd_process *p,
			   uint8_t __user *user_priv_data,
			   uint64_t *priv_data_offset,
			   uint64_t max_priv_data_size);
1139 1140 1141 1142 1143 1144 1145 1146 1147 1148

int kfd_criu_checkpoint_events(struct kfd_process *p,
			 uint8_t __user *user_priv_data,
			 uint64_t *priv_data_offset);

int kfd_criu_restore_event(struct file *devkfd,
			   struct kfd_process *p,
			   uint8_t __user *user_priv_data,
			   uint64_t *priv_data_offset,
			   uint64_t max_priv_data_size);
1149 1150
/* CRIU - End */

B
Ben Goz 已提交
1151
/* Queue Context Management */
1152
int init_queue(struct queue **q, const struct queue_properties *properties);
B
Ben Goz 已提交
1153
void uninit_queue(struct queue *q);
1154
void print_queue_properties(struct queue_properties *q);
B
Ben Goz 已提交
1155 1156
void print_queue(struct queue *q);

1157 1158
struct mqd_manager *mqd_manager_init_cik(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1159 1160
struct mqd_manager *mqd_manager_init_cik_hawaii(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1161 1162
struct mqd_manager *mqd_manager_init_vi(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1163 1164
struct mqd_manager *mqd_manager_init_vi_tonga(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1165 1166
struct mqd_manager *mqd_manager_init_v9(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1167 1168
struct mqd_manager *mqd_manager_init_v10(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1169 1170
struct device_queue_manager *device_queue_manager_init(struct kfd_dev *dev);
void device_queue_manager_uninit(struct device_queue_manager *dqm);
B
Ben Goz 已提交
1171 1172
struct kernel_queue *kernel_queue_init(struct kfd_dev *dev,
					enum kfd_queue_type type);
1173
void kernel_queue_uninit(struct kernel_queue *kq, bool hanging);
1174
int kfd_dqm_evict_pasid(struct device_queue_manager *dqm, u32 pasid);
B
Ben Goz 已提交
1175

1176 1177 1178 1179 1180 1181 1182
/* Process Queue Manager */
struct process_queue_node {
	struct queue *q;
	struct kernel_queue *kq;
	struct list_head process_queue_list;
};

1183 1184
void kfd_process_dequeue_from_device(struct kfd_process_device *pdd);
void kfd_process_dequeue_from_all_devices(struct kfd_process *p);
1185 1186 1187 1188 1189 1190
int pqm_init(struct process_queue_manager *pqm, struct kfd_process *p);
void pqm_uninit(struct process_queue_manager *pqm);
int pqm_create_queue(struct process_queue_manager *pqm,
			    struct kfd_dev *dev,
			    struct file *f,
			    struct queue_properties *properties,
1191
			    unsigned int *qid,
1192
			    const struct kfd_criu_queue_priv_data *q_data,
1193
			    const void *restore_mqd,
1194
			    const void *restore_ctl_stack,
1195
			    uint32_t *p_doorbell_offset_in_process);
1196
int pqm_destroy_queue(struct process_queue_manager *pqm, unsigned int qid);
1197
int pqm_update_queue_properties(struct process_queue_manager *pqm, unsigned int qid,
1198
			struct queue_properties *p);
1199 1200
int pqm_update_mqd(struct process_queue_manager *pqm, unsigned int qid,
			struct mqd_update_info *minfo);
1201 1202
int pqm_set_gws(struct process_queue_manager *pqm, unsigned int qid,
			void *gws);
1203 1204
struct kernel_queue *pqm_get_kernel_queue(struct process_queue_manager *pqm,
						unsigned int qid);
1205 1206
struct queue *pqm_get_user_queue(struct process_queue_manager *pqm,
						unsigned int qid);
1207 1208 1209 1210 1211
int pqm_get_wave_state(struct process_queue_manager *pqm,
		       unsigned int qid,
		       void __user *ctl_stack,
		       u32 *ctl_stack_used_size,
		       u32 *save_area_used_size);
1212

1213 1214
int amdkfd_fence_wait_timeout(uint64_t *fence_addr,
			      uint64_t fence_value,
1215
			      unsigned int timeout_ms);
1216

1217 1218
int pqm_get_queue_checkpoint_info(struct process_queue_manager *pqm,
				  unsigned int qid,
1219 1220
				  u32 *mqd_size,
				  u32 *ctl_stack_size);
B
Ben Goz 已提交
1221 1222
/* Packet Manager */

1223 1224
#define KFD_FENCE_COMPLETED (100)
#define KFD_FENCE_INIT   (10)
B
Ben Goz 已提交
1225

B
Ben Goz 已提交
1226 1227 1228 1229 1230 1231
struct packet_manager {
	struct device_queue_manager *dqm;
	struct kernel_queue *priv_queue;
	struct mutex lock;
	bool allocated;
	struct kfd_mem_obj *ib_buffer_obj;
1232
	unsigned int ib_size_bytes;
1233
	bool is_over_subscription;
1234 1235 1236 1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249

	const struct packet_manager_funcs *pmf;
};

struct packet_manager_funcs {
	/* Support ASIC-specific packet formats for PM4 packets */
	int (*map_process)(struct packet_manager *pm, uint32_t *buffer,
			struct qcm_process_device *qpd);
	int (*runlist)(struct packet_manager *pm, uint32_t *buffer,
			uint64_t ib, size_t ib_size_in_dwords, bool chain);
	int (*set_resources)(struct packet_manager *pm, uint32_t *buffer,
			struct scheduling_resources *res);
	int (*map_queues)(struct packet_manager *pm, uint32_t *buffer,
			struct queue *q, bool is_static);
	int (*unmap_queues)(struct packet_manager *pm, uint32_t *buffer,
			enum kfd_unmap_queues_filter mode,
1250
			uint32_t filter_param, bool reset);
1251
	int (*query_status)(struct packet_manager *pm, uint32_t *buffer,
1252
			uint64_t fence_address,	uint64_t fence_value);
1253 1254 1255 1256 1257 1258 1259 1260 1261 1262
	int (*release_mem)(uint64_t gpu_addr, uint32_t *buffer);

	/* Packet sizes */
	int map_process_size;
	int runlist_size;
	int set_resources_size;
	int map_queues_size;
	int unmap_queues_size;
	int query_status_size;
	int release_mem_size;
B
Ben Goz 已提交
1263 1264
};

1265
extern const struct packet_manager_funcs kfd_vi_pm_funcs;
1266
extern const struct packet_manager_funcs kfd_v9_pm_funcs;
1267
extern const struct packet_manager_funcs kfd_aldebaran_pm_funcs;
1268

1269
int pm_init(struct packet_manager *pm, struct device_queue_manager *dqm);
1270
void pm_uninit(struct packet_manager *pm, bool hanging);
1271 1272 1273 1274
int pm_send_set_resources(struct packet_manager *pm,
				struct scheduling_resources *res);
int pm_send_runlist(struct packet_manager *pm, struct list_head *dqm_queues);
int pm_send_query_status(struct packet_manager *pm, uint64_t fence_address,
1275
				uint64_t fence_value);
1276

1277
int pm_send_unmap_queue(struct packet_manager *pm,
1278
			enum kfd_unmap_queues_filter mode,
1279
			uint32_t filter_param, bool reset);
1280

B
Ben Goz 已提交
1281 1282
void pm_release_ib(struct packet_manager *pm);

1283 1284 1285
/* Following PM funcs can be shared among VI and AI */
unsigned int pm_build_pm4_header(unsigned int opcode, size_t packet_size);

1286 1287
uint64_t kfd_get_number_elems(struct kfd_dev *kfd);

1288 1289
/* Events */
extern const struct kfd_event_interrupt_class event_interrupt_class_cik;
1290 1291
extern const struct kfd_event_interrupt_class event_interrupt_class_v9;

1292
extern const struct kfd_device_global_init_class device_global_init_class_cik;
1293 1294 1295 1296 1297

void kfd_event_init_process(struct kfd_process *p);
void kfd_event_free_process(struct kfd_process *p);
int kfd_event_mmap(struct kfd_process *process, struct vm_area_struct *vma);
int kfd_wait_on_events(struct kfd_process *p,
1298
		       uint32_t num_events, void __user *data,
1299
		       bool all, uint32_t user_timeout_ms,
1300
		       uint32_t *wait_result);
1301
void kfd_signal_event_interrupt(u32 pasid, uint32_t partial_id,
1302
				uint32_t valid_id_bits);
1303
void kfd_signal_iommu_event(struct kfd_dev *dev,
1304 1305 1306
			    u32 pasid, unsigned long address,
			    bool is_write_requested, bool is_execute_requested);
void kfd_signal_hw_exception_event(u32 pasid);
1307 1308
int kfd_set_event(struct kfd_process *p, uint32_t event_id);
int kfd_reset_event(struct kfd_process *p, uint32_t event_id);
1309 1310
int kfd_kmap_event_page(struct kfd_process *p, uint64_t event_page_offset);

1311 1312 1313 1314
int kfd_event_create(struct file *devkfd, struct kfd_process *p,
		     uint32_t event_type, bool auto_reset, uint32_t node_id,
		     uint32_t *event_id, uint32_t *event_trigger_data,
		     uint64_t *event_page_offset, uint32_t *event_slot_index);
1315 1316

int kfd_get_num_events(struct kfd_process *p);
1317 1318
int kfd_event_destroy(struct kfd_process *p, uint32_t event_id);

1319
void kfd_signal_vm_fault_event(struct kfd_dev *dev, u32 pasid,
S
shaoyunl 已提交
1320 1321
				struct kfd_vm_fault_info *info);

1322 1323
void kfd_signal_reset_event(struct kfd_dev *dev);

1324 1325
void kfd_signal_poison_consumed_event(struct kfd_dev *dev, u32 pasid);

1326
void kfd_flush_tlb(struct kfd_process_device *pdd, enum TLB_FLUSH_TYPE type);
1327

1328 1329
bool kfd_is_locked(void);

1330 1331 1332 1333
/* Compute profile */
void kfd_inc_compute_active(struct kfd_dev *dev);
void kfd_dec_compute_active(struct kfd_dev *dev);

1334 1335 1336 1337
/* Cgroup Support */
/* Check with device cgroup if @kfd device is accessible */
static inline int kfd_devcgroup_check_permission(struct kfd_dev *kfd)
{
1338
#if defined(CONFIG_CGROUP_DEVICE) || defined(CONFIG_CGROUP_BPF)
1339 1340
	struct drm_device *ddev = kfd->ddev;

1341
	return devcgroup_check_permission(DEVCG_DEV_CHAR, DRM_MAJOR,
1342 1343 1344 1345 1346 1347 1348
					  ddev->render->index,
					  DEVCG_ACC_WRITE | DEVCG_ACC_READ);
#else
	return 0;
#endif
}

1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359 1360
/* Debugfs */
#if defined(CONFIG_DEBUG_FS)

void kfd_debugfs_init(void);
void kfd_debugfs_fini(void);
int kfd_debugfs_mqds_by_process(struct seq_file *m, void *data);
int pqm_debugfs_mqds(struct seq_file *m, void *data);
int kfd_debugfs_hqds_by_device(struct seq_file *m, void *data);
int dqm_debugfs_hqds(struct seq_file *m, void *data);
int kfd_debugfs_rls_by_device(struct seq_file *m, void *data);
int pm_debugfs_runlist(struct seq_file *m, void *data);

1361 1362
int kfd_debugfs_hang_hws(struct kfd_dev *dev);
int pm_debugfs_hang_hws(struct packet_manager *pm);
1363
int dqm_debugfs_hang_hws(struct device_queue_manager *dqm);
1364

1365 1366 1367 1368 1369 1370 1371
#else

static inline void kfd_debugfs_init(void) {}
static inline void kfd_debugfs_fini(void) {}

#endif

O
Oded Gabbay 已提交
1372
#endif