kfd_priv.h 40.7 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
	atomic64_t tlb_seq;
709

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

713 714 715 716
	/* 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.
717
	 */
718
	bool already_dequeued;
719
	bool runtime_inuse;
720 721 722

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

	/* VRAM usage */
	uint64_t vram_usage;
	struct attribute attr_vram;
727 728 729 730 731 732
	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];
733 734 735 736 737 738 739

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

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

	/*
	 * @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;
766 767 768 769 770 771 772 773 774

	/* 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;
775 776 777 778 779 780
	/*
	 * 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;
781 782
};

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

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

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

808 809 810 811 812 813
	/*
	 * 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;
814

815 816 817
	struct kref ref;
	struct work_struct release_work;

818 819 820 821 822 823
	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 已提交
824 825
	 * It can also be used for safely getting a reference to the
	 * mm_struct of the process.
826 827 828 829 830 831
	 */
	struct task_struct *lead_thread;

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

832
	u32 pasid;
833 834

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

841 842
	struct process_queue_manager pqm;

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

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

	/* 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;
864 865 866 867 868 869 870 871 872 873

	/* 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;
874 875 876

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

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

	bool xnack_enabled;
884 885

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

890 891 892 893
#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;

894
/**
895 896 897 898 899
 * 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.
900
 *
901
 * Return: returns ioctl completion code.
902 903 904 905 906 907 908 909 910 911 912
 */
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;
};
913
bool kfd_dev_is_large_bar(struct kfd_dev *dev);
914

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

int kfd_process_gpuidx_from_gpuid(struct kfd_process *p, uint32_t gpu_id);
923
int kfd_process_gpuid_from_adev(struct kfd_process *p,
924 925
			       struct amdgpu_device *adev, uint32_t *gpuid,
			       uint32_t *gpuidx);
926 927 928 929 930 931 932 933 934
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;
}

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

941 942 943 944 945
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);

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

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

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

960 961 962 963 964 965 966
/* 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);
967
struct kfd_process *kfd_lookup_process_by_pid(struct pid *pid);
968

969 970 971 972 973
/* 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);
974 975
u32 kfd_pasid_alloc(void);
void kfd_pasid_free(u32 pasid);
976 977

/* Doorbells */
978
size_t kfd_doorbell_process_slice(struct kfd_dev *kfd);
979 980
int kfd_doorbell_init(struct kfd_dev *kfd);
void kfd_doorbell_fini(struct kfd_dev *kfd);
981 982
int kfd_doorbell_mmap(struct kfd_dev *dev, struct kfd_process *process,
		      struct vm_area_struct *vma);
983
void __iomem *kfd_get_kernel_doorbell(struct kfd_dev *kfd,
984 985 986
					unsigned int *doorbell_off);
void kfd_release_kernel_doorbell(struct kfd_dev *kfd, u32 __iomem *db_addr);
u32 read_kernel_doorbell(u32 __iomem *db);
987
void write_kernel_doorbell(void __iomem *db, u32 value);
988
void write_kernel_doorbell64(void __iomem *db, u64 value);
989
unsigned int kfd_get_doorbell_dw_offset_in_bar(struct kfd_dev *kfd,
990
					struct kfd_process_device *pdd,
991
					unsigned int doorbell_id);
992 993 994 995 996
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);
997 998 999 1000 1001 1002 1003
/* 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 已提交
1004 1005
extern struct device *kfd_device;

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

1012 1013 1014 1015 1016
/* 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);
1017 1018
struct kfd_topology_device *kfd_topology_device_by_proximity_domain(
						uint32_t proximity_domain);
1019 1020
struct kfd_topology_device *kfd_topology_device_by_proximity_domain_no_lock(
						uint32_t proximity_domain);
1021
struct kfd_topology_device *kfd_topology_device_by_id(uint32_t gpu_id);
1022 1023
struct kfd_dev *kfd_device_by_id(uint32_t gpu_id);
struct kfd_dev *kfd_device_by_pci_dev(const struct pci_dev *pdev);
1024
struct kfd_dev *kfd_device_by_adev(const struct amdgpu_device *adev);
1025
int kfd_topology_enum_kfd_devices(uint8_t idx, struct kfd_dev **kdev);
1026
int kfd_numa_node_to_apic_id(int numa_node_id);
1027
void kfd_double_confirm_iommu_support(struct kfd_dev *gpu);
1028

O
Oded Gabbay 已提交
1029
/* Interrupts */
1030 1031 1032
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);
1033 1034 1035
bool interrupt_is_wanted(struct kfd_dev *dev,
				const uint32_t *ih_ring_entry,
				uint32_t *patched_ihre, bool *flag);
O
Oded Gabbay 已提交
1036

1037 1038 1039
/* amdkfd Apertures */
int kfd_init_apertures(struct kfd_process *process);

1040 1041 1042 1043
void kfd_process_set_trap_handler(struct qcm_process_device *qpd,
				  uint64_t tba_addr,
				  uint64_t tma_addr);

1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056 1057 1058 1059
/* 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;
1060
	uint32_t xnack_mode;
1061 1062 1063 1064 1065 1066 1067 1068
};

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

struct kfd_criu_bo_priv_data {
1069 1070 1071
	uint64_t user_addr;
	uint32_t idr_handle;
	uint32_t mapped_gpuids[MAX_GPU_INSTANCE];
1072 1073
};

1074 1075 1076 1077 1078 1079 1080 1081 1082 1083
/*
 * 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,
};

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

struct kfd_criu_queue_priv_data {
	uint32_t object_type;
1094 1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113
	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;
1114 1115 1116 1117
};

struct kfd_criu_event_priv_data {
	uint32_t object_type;
1118 1119 1120 1121 1122 1123 1124 1125 1126 1127
	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;
	};
1128 1129
};

1130 1131 1132 1133 1134 1135 1136 1137 1138 1139 1140 1141
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);
1142 1143 1144 1145 1146 1147 1148 1149 1150 1151

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);
1152 1153
/* CRIU - End */

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

1160 1161
struct mqd_manager *mqd_manager_init_cik(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1162 1163
struct mqd_manager *mqd_manager_init_cik_hawaii(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1164 1165
struct mqd_manager *mqd_manager_init_vi(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1166 1167
struct mqd_manager *mqd_manager_init_vi_tonga(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1168 1169
struct mqd_manager *mqd_manager_init_v9(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1170 1171
struct mqd_manager *mqd_manager_init_v10(enum KFD_MQD_TYPE type,
		struct kfd_dev *dev);
1172 1173
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 已提交
1174 1175
struct kernel_queue *kernel_queue_init(struct kfd_dev *dev,
					enum kfd_queue_type type);
1176
void kernel_queue_uninit(struct kernel_queue *kq, bool hanging);
1177
int kfd_dqm_evict_pasid(struct device_queue_manager *dqm, u32 pasid);
B
Ben Goz 已提交
1178

1179 1180 1181 1182 1183 1184 1185
/* Process Queue Manager */
struct process_queue_node {
	struct queue *q;
	struct kernel_queue *kq;
	struct list_head process_queue_list;
};

1186 1187
void kfd_process_dequeue_from_device(struct kfd_process_device *pdd);
void kfd_process_dequeue_from_all_devices(struct kfd_process *p);
1188 1189 1190 1191 1192 1193
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,
1194
			    unsigned int *qid,
1195
			    const struct kfd_criu_queue_priv_data *q_data,
1196
			    const void *restore_mqd,
1197
			    const void *restore_ctl_stack,
1198
			    uint32_t *p_doorbell_offset_in_process);
1199
int pqm_destroy_queue(struct process_queue_manager *pqm, unsigned int qid);
1200
int pqm_update_queue_properties(struct process_queue_manager *pqm, unsigned int qid,
1201
			struct queue_properties *p);
1202 1203
int pqm_update_mqd(struct process_queue_manager *pqm, unsigned int qid,
			struct mqd_update_info *minfo);
1204 1205
int pqm_set_gws(struct process_queue_manager *pqm, unsigned int qid,
			void *gws);
1206 1207
struct kernel_queue *pqm_get_kernel_queue(struct process_queue_manager *pqm,
						unsigned int qid);
1208 1209
struct queue *pqm_get_user_queue(struct process_queue_manager *pqm,
						unsigned int qid);
1210 1211 1212 1213 1214
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);
1215

1216 1217
int amdkfd_fence_wait_timeout(uint64_t *fence_addr,
			      uint64_t fence_value,
1218
			      unsigned int timeout_ms);
1219

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

1226 1227
#define KFD_FENCE_COMPLETED (100)
#define KFD_FENCE_INIT   (10)
B
Ben Goz 已提交
1228

B
Ben Goz 已提交
1229 1230 1231 1232 1233 1234
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;
1235
	unsigned int ib_size_bytes;
1236
	bool is_over_subscription;
1237 1238 1239 1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252

	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,
1253
			uint32_t filter_param, bool reset);
1254
	int (*query_status)(struct packet_manager *pm, uint32_t *buffer,
1255
			uint64_t fence_address,	uint64_t fence_value);
1256 1257 1258 1259 1260 1261 1262 1263 1264 1265
	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 已提交
1266 1267
};

1268
extern const struct packet_manager_funcs kfd_vi_pm_funcs;
1269
extern const struct packet_manager_funcs kfd_v9_pm_funcs;
1270
extern const struct packet_manager_funcs kfd_aldebaran_pm_funcs;
1271

1272
int pm_init(struct packet_manager *pm, struct device_queue_manager *dqm);
1273
void pm_uninit(struct packet_manager *pm, bool hanging);
1274 1275 1276 1277
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,
1278
				uint64_t fence_value);
1279

1280
int pm_send_unmap_queue(struct packet_manager *pm,
1281
			enum kfd_unmap_queues_filter mode,
1282
			uint32_t filter_param, bool reset);
1283

B
Ben Goz 已提交
1284 1285
void pm_release_ib(struct packet_manager *pm);

1286 1287 1288
/* Following PM funcs can be shared among VI and AI */
unsigned int pm_build_pm4_header(unsigned int opcode, size_t packet_size);

1289 1290
uint64_t kfd_get_number_elems(struct kfd_dev *kfd);

1291 1292
/* Events */
extern const struct kfd_event_interrupt_class event_interrupt_class_cik;
1293 1294
extern const struct kfd_event_interrupt_class event_interrupt_class_v9;

1295
extern const struct kfd_device_global_init_class device_global_init_class_cik;
1296 1297 1298 1299 1300

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,
1301
		       uint32_t num_events, void __user *data,
1302
		       bool all, uint32_t user_timeout_ms,
1303
		       uint32_t *wait_result);
1304
void kfd_signal_event_interrupt(u32 pasid, uint32_t partial_id,
1305
				uint32_t valid_id_bits);
1306
void kfd_signal_iommu_event(struct kfd_dev *dev,
1307 1308 1309
			    u32 pasid, unsigned long address,
			    bool is_write_requested, bool is_execute_requested);
void kfd_signal_hw_exception_event(u32 pasid);
1310 1311
int kfd_set_event(struct kfd_process *p, uint32_t event_id);
int kfd_reset_event(struct kfd_process *p, uint32_t event_id);
1312 1313
int kfd_kmap_event_page(struct kfd_process *p, uint64_t event_page_offset);

1314 1315 1316 1317
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);
1318 1319

int kfd_get_num_events(struct kfd_process *p);
1320 1321
int kfd_event_destroy(struct kfd_process *p, uint32_t event_id);

1322
void kfd_signal_vm_fault_event(struct kfd_dev *dev, u32 pasid,
S
shaoyunl 已提交
1323 1324
				struct kfd_vm_fault_info *info);

1325 1326
void kfd_signal_reset_event(struct kfd_dev *dev);

1327 1328
void kfd_signal_poison_consumed_event(struct kfd_dev *dev, u32 pasid);

1329
void kfd_flush_tlb(struct kfd_process_device *pdd, enum TLB_FLUSH_TYPE type);
1330

1331 1332
bool kfd_is_locked(void);

1333 1334 1335 1336
/* Compute profile */
void kfd_inc_compute_active(struct kfd_dev *dev);
void kfd_dec_compute_active(struct kfd_dev *dev);

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

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

1352 1353 1354 1355 1356 1357 1358 1359 1360 1361 1362 1363
/* 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);

1364 1365
int kfd_debugfs_hang_hws(struct kfd_dev *dev);
int pm_debugfs_hang_hws(struct packet_manager *pm);
1366
int dqm_debugfs_hang_hws(struct device_queue_manager *dqm);
1367

1368 1369 1370 1371 1372 1373 1374
#else

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

#endif

O
Oded Gabbay 已提交
1375
#endif