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 359
/* Character device interface */
int kfd_chardev_init(void);
void kfd_chardev_exit(void);
struct device *kfd_chardev(void);

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

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

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

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

B
Ben Goz 已提交
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 435
/**
 * 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.
 *
436 437 438
 * @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 已提交
439 440 441
 *
 * @doorbell_off: The doorbell offset in the doorbell pci-bar.
 *
442 443
 * @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 已提交
444
 *
445 446 447 448 449
 * @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 已提交
450
 *
451 452 453 454
 * @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 已提交
455 456 457 458 459 460 461
 * @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.
 *
 */
462

B
Ben Goz 已提交
463 464
struct queue_properties {
	enum kfd_queue_type type;
B
Ben Goz 已提交
465
	enum kfd_queue_format format;
B
Ben Goz 已提交
466 467 468 469 470 471 472
	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;
473
	void __iomem *doorbell_ptr;
B
Ben Goz 已提交
474 475
	uint32_t doorbell_off;
	bool is_interop;
476
	bool is_evicted;
B
Ben Goz 已提交
477
	bool is_active;
478
	bool is_gws;
B
Ben Goz 已提交
479 480
	/* Not relevant for user mode queues in cp scheduling */
	unsigned int vmid;
B
Ben Goz 已提交
481 482 483 484
	/* Relevant only for sdma queues*/
	uint32_t sdma_engine_id;
	uint32_t sdma_queue_id;
	uint32_t sdma_vm_addr;
485 486 487 488 489
	/* 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 已提交
490 491 492
	uint32_t ctl_stack_size;
	uint64_t tba_addr;
	uint64_t tma_addr;
B
Ben Goz 已提交
493 494
};

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

500 501 502 503 504 505 506 507 508 509 510 511 512
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;
};
513

B
Ben Goz 已提交
514 515 516 517 518
/**
 * struct queue
 *
 * @list: Queue linked list.
 *
519
 * @mqd: The queue MQD (memory queue descriptor).
B
Ben Goz 已提交
520 521 522 523 524 525 526 527
 *
 * @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
528
 *	 that the queue should be executed on.
B
Ben Goz 已提交
529
 *
530 531
 * @pipe: Used only in no cp scheduling mode and identifies the queue's pipe
 *	  id.
B
Ben Goz 已提交
532 533 534 535 536 537 538
 *
 * @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.
 *
539 540 541
 * @gws: Pointing to gws kgd_mem if this is a gws control queue; NULL
 * otherwise.
 *
B
Ben Goz 已提交
542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
 * 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 已提交
558
	unsigned int sdma_id;
559
	unsigned int doorbell_id;
B
Ben Goz 已提交
560

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

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

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

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

B
Ben Goz 已提交
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 610
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;
611
	unsigned int evicted; /* eviction counter, 0=active */
612 613 614 615 616 617

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

618 619 620 621 622 623 624 625
	/* 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;

626
	/* All the memory management data should be here too */
B
Ben Goz 已提交
627
	uint64_t gds_context_area;
628
	/* Contains page table flags such as AMDGPU_PTE_VALID since gfx9 */
629
	uint64_t page_table_base;
B
Ben Goz 已提交
630 631 632 633 634 635 636
	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;
637
	uint32_t sh_hidden_private_base;
F
Felix Kuehling 已提交
638 639

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

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

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

655 656 657 658 659 660 661 662 663
/* 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

664 665 666 667 668 669 670 671
/* 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)

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

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

/*
 * 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
686

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

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

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

698 699 700 701 702 703 704 705
	/*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;

706
	/* VM context for GPUVM allocations */
707
	struct file *drm_file;
708
	void *drm_priv;
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
struct kfd_topology_device *kfd_topology_device_by_id(uint32_t gpu_id);
1020 1021
struct kfd_dev *kfd_device_by_id(uint32_t gpu_id);
struct kfd_dev *kfd_device_by_pci_dev(const struct pci_dev *pdev);
1022
struct kfd_dev *kfd_device_by_adev(const struct amdgpu_device *adev);
1023
int kfd_topology_enum_kfd_devices(uint8_t idx, struct kfd_dev **kdev);
1024
int kfd_numa_node_to_apic_id(int numa_node_id);
1025
void kfd_double_confirm_iommu_support(struct kfd_dev *gpu);
1026

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

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

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

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

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

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

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

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

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

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

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

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);
1150 1151
/* CRIU - End */

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

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

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

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

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

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

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

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

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

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

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

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

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

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

1287 1288
uint64_t kfd_get_number_elems(struct kfd_dev *kfd);

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

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

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

1312 1313 1314 1315
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);
1316 1317

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

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

1323 1324
void kfd_signal_reset_event(struct kfd_dev *dev);

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

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

1329 1330
bool kfd_is_locked(void);

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

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

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

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

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

1366 1367 1368 1369 1370 1371 1372
#else

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

#endif

O
Oded Gabbay 已提交
1373
#endif