1 /*
2  * Copyright 2022 Advanced Micro Devices, Inc.
3  *
4  * Permission is hereby granted, free of charge, to any person obtaining a
5  * copy of this software and associated documentation files (the "Software"),
6  * to deal in the Software without restriction, including without limitation
7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8  * and/or sell copies of the Software, and to permit persons to whom the
9  * Software is furnished to do so, subject to the following conditions:
10  *
11  * The above copyright notice and this permission notice shall be included in
12  * all copies or substantial portions of the Software.
13  *
14  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
17  * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18  * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19  * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20  * OTHER DEALINGS IN THE SOFTWARE.
21  *
22  */
23 
24 #ifndef __MES_API_DEF_H__
25 #define __MES_API_DEF_H__
26 
27 #pragma pack(push, 4)
28 
29 #define MES_API_VERSION 1
30 
31 /* Driver submits one API(cmd) as a single Frame and this command size is same
32  * for all API to ease the debugging and parsing of ring buffer.
33  */
34 enum { API_FRAME_SIZE_IN_DWORDS = 64 };
35 
36 /* To avoid command in scheduler context to be overwritten whenenver mutilple
37  * interrupts come in, this creates another queue.
38  */
39 enum { API_NUMBER_OF_COMMAND_MAX = 32 };
40 
41 enum MES_API_TYPE {
42 	MES_API_TYPE_SCHEDULER = 1,
43 	MES_API_TYPE_MAX
44 };
45 
46 enum MES_SCH_API_OPCODE {
47 	MES_SCH_API_SET_HW_RSRC			= 0,
48 	MES_SCH_API_SET_SCHEDULING_CONFIG	= 1, /* agreegated db, quantums, etc */
49 	MES_SCH_API_ADD_QUEUE			= 2,
50 	MES_SCH_API_REMOVE_QUEUE		= 3,
51 	MES_SCH_API_PERFORM_YIELD		= 4,
52 	MES_SCH_API_SET_GANG_PRIORITY_LEVEL	= 5,
53 	MES_SCH_API_SUSPEND			= 6,
54 	MES_SCH_API_RESUME			= 7,
55 	MES_SCH_API_RESET			= 8,
56 	MES_SCH_API_SET_LOG_BUFFER		= 9,
57 	MES_SCH_API_CHANGE_GANG_PRORITY		= 10,
58 	MES_SCH_API_QUERY_SCHEDULER_STATUS	= 11,
59 	MES_SCH_API_PROGRAM_GDS			= 12,
60 	MES_SCH_API_SET_DEBUG_VMID		= 13,
61 	MES_SCH_API_MISC			= 14,
62 	MES_SCH_API_UPDATE_ROOT_PAGE_TABLE      = 15,
63 	MES_SCH_API_AMD_LOG                     = 16,
64 	MES_SCH_API_MAX				= 0xFF
65 };
66 
67 union MES_API_HEADER {
68 	struct {
69 		uint32_t type		: 4; /* 0 - Invalid; 1 - Scheduling; 2 - TBD */
70 		uint32_t opcode		: 8;
71 		uint32_t dwsize		: 8; /* including header */
72 		uint32_t reserved	: 12;
73 	};
74 
75 	uint32_t	u32All;
76 };
77 
78 enum MES_AMD_PRIORITY_LEVEL {
79 	AMD_PRIORITY_LEVEL_LOW		= 0,
80 	AMD_PRIORITY_LEVEL_NORMAL	= 1,
81 	AMD_PRIORITY_LEVEL_MEDIUM	= 2,
82 	AMD_PRIORITY_LEVEL_HIGH		= 3,
83 	AMD_PRIORITY_LEVEL_REALTIME	= 4,
84 	AMD_PRIORITY_NUM_LEVELS
85 };
86 
87 enum MES_QUEUE_TYPE {
88 	MES_QUEUE_TYPE_GFX,
89 	MES_QUEUE_TYPE_COMPUTE,
90 	MES_QUEUE_TYPE_SDMA,
91 	MES_QUEUE_TYPE_MAX,
92 };
93 
94 struct MES_API_STATUS {
95 	uint64_t	api_completion_fence_addr;
96 	uint64_t	api_completion_fence_value;
97 };
98 
99 enum { MAX_COMPUTE_PIPES = 8 };
100 enum { MAX_GFX_PIPES = 2 };
101 enum { MAX_SDMA_PIPES = 2 };
102 
103 enum { MAX_COMPUTE_HQD_PER_PIPE = 8 };
104 enum { MAX_GFX_HQD_PER_PIPE = 8 };
105 enum { MAX_SDMA_HQD_PER_PIPE = 10 };
106 enum { MAX_SDMA_HQD_PER_PIPE_11_0   = 8 };
107 
108 enum { MAX_QUEUES_IN_A_GANG = 8 };
109 
110 enum VM_HUB_TYPE {
111 	VM_HUB_TYPE_GC = 0,
112 	VM_HUB_TYPE_MM = 1,
113 	VM_HUB_TYPE_MAX,
114 };
115 
116 enum { VMID_INVALID = 0xffff };
117 
118 enum { MAX_VMID_GCHUB = 16 };
119 enum { MAX_VMID_MMHUB = 16 };
120 
121 enum SET_DEBUG_VMID_OPERATIONS {
122 	DEBUG_VMID_OP_PROGRAM = 0,
123 	DEBUG_VMID_OP_ALLOCATE = 1,
124 	DEBUG_VMID_OP_RELEASE = 2
125 };
126 
127 enum MES_LOG_OPERATION {
128 	MES_LOG_OPERATION_CONTEXT_STATE_CHANGE = 0,
129 	MES_LOG_OPERATION_QUEUE_NEW_WORK = 1,
130 	MES_LOG_OPERATION_QUEUE_UNWAIT_SYNC_OBJECT = 2,
131 	MES_LOG_OPERATION_QUEUE_NO_MORE_WORK = 3,
132 	MES_LOG_OPERATION_QUEUE_WAIT_SYNC_OBJECT = 4,
133 	MES_LOG_OPERATION_QUEUE_INVALID = 0xF,
134 };
135 
136 enum MES_LOG_CONTEXT_STATE {
137 	MES_LOG_CONTEXT_STATE_IDLE		= 0,
138 	MES_LOG_CONTEXT_STATE_RUNNING		= 1,
139 	MES_LOG_CONTEXT_STATE_READY		= 2,
140 	MES_LOG_CONTEXT_STATE_READY_STANDBY	= 3,
141 	MES_LOG_CONTEXT_STATE_INVALID           = 0xF,
142 };
143 
144 struct MES_LOG_CONTEXT_STATE_CHANGE {
145 	void				*h_context;
146 	enum MES_LOG_CONTEXT_STATE	new_context_state;
147 };
148 
149 struct MES_LOG_QUEUE_NEW_WORK {
150 	uint64_t                   h_queue;
151 	uint64_t                   reserved;
152 };
153 
154 struct MES_LOG_QUEUE_UNWAIT_SYNC_OBJECT {
155 	uint64_t                   h_queue;
156 	uint64_t                   h_sync_object;
157 };
158 
159 struct MES_LOG_QUEUE_NO_MORE_WORK {
160 	uint64_t                   h_queue;
161 	uint64_t                   reserved;
162 };
163 
164 struct MES_LOG_QUEUE_WAIT_SYNC_OBJECT {
165 	uint64_t                   h_queue;
166 	uint64_t                   h_sync_object;
167 };
168 
169 struct MES_LOG_ENTRY_HEADER {
170 	uint32_t	first_free_entry_index;
171 	uint32_t	wraparound_count;
172 	uint64_t	number_of_entries;
173 	uint64_t	reserved[2];
174 };
175 
176 struct MES_LOG_ENTRY_DATA {
177 	uint64_t	gpu_time_stamp;
178 	uint32_t	operation_type; /* operation_type is of MES_LOG_OPERATION type */
179 	uint32_t	reserved_operation_type_bits;
180 	union {
181 		struct MES_LOG_CONTEXT_STATE_CHANGE     context_state_change;
182 		struct MES_LOG_QUEUE_NEW_WORK           queue_new_work;
183 		struct MES_LOG_QUEUE_UNWAIT_SYNC_OBJECT queue_unwait_sync_object;
184 		struct MES_LOG_QUEUE_NO_MORE_WORK       queue_no_more_work;
185 		struct MES_LOG_QUEUE_WAIT_SYNC_OBJECT   queue_wait_sync_object;
186 		uint64_t                                all[2];
187 	};
188 };
189 
190 struct MES_LOG_BUFFER {
191 	struct MES_LOG_ENTRY_HEADER	header;
192 	struct MES_LOG_ENTRY_DATA	entries[1];
193 };
194 
195 enum MES_SWIP_TO_HWIP_DEF {
196 	MES_MAX_HWIP_SEGMENT = 8,
197 };
198 
199 union MESAPI_SET_HW_RESOURCES {
200 	struct {
201 		union MES_API_HEADER	header;
202 		uint32_t		vmid_mask_mmhub;
203 		uint32_t		vmid_mask_gfxhub;
204 		uint32_t		gds_size;
205 		uint32_t		paging_vmid;
206 		uint32_t		compute_hqd_mask[MAX_COMPUTE_PIPES];
207 		uint32_t		gfx_hqd_mask[MAX_GFX_PIPES];
208 		uint32_t		sdma_hqd_mask[MAX_SDMA_PIPES];
209 		uint32_t		aggregated_doorbells[AMD_PRIORITY_NUM_LEVELS];
210 		uint64_t		g_sch_ctx_gpu_mc_ptr;
211 		uint64_t		query_status_fence_gpu_mc_ptr;
212 		uint32_t		gc_base[MES_MAX_HWIP_SEGMENT];
213 		uint32_t		mmhub_base[MES_MAX_HWIP_SEGMENT];
214 		uint32_t		osssys_base[MES_MAX_HWIP_SEGMENT];
215 		struct MES_API_STATUS	api_status;
216 		union {
217 			struct {
218 				uint32_t disable_reset	: 1;
219 				uint32_t use_different_vmid_compute : 1;
220 				uint32_t disable_mes_log   : 1;
221 				uint32_t apply_mmhub_pgvm_invalidate_ack_loss_wa : 1;
222 				uint32_t apply_grbm_remote_register_dummy_read_wa : 1;
223 				uint32_t second_gfx_pipe_enabled : 1;
224 				uint32_t enable_level_process_quantum_check : 1;
225 				uint32_t reserved	: 25;
226 			};
227 			uint32_t	uint32_t_all;
228 		};
229 		uint32_t	oversubscription_timer;
230 	};
231 
232 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
233 };
234 
235 union MESAPI__ADD_QUEUE {
236 	struct {
237 		union MES_API_HEADER		header;
238 		uint32_t			process_id;
239 		uint64_t			page_table_base_addr;
240 		uint64_t			process_va_start;
241 		uint64_t			process_va_end;
242 		uint64_t			process_quantum;
243 		uint64_t			process_context_addr;
244 		uint64_t			gang_quantum;
245 		uint64_t			gang_context_addr;
246 		uint32_t			inprocess_gang_priority;
247 		enum MES_AMD_PRIORITY_LEVEL	gang_global_priority_level;
248 		uint32_t			doorbell_offset;
249 		uint64_t			mqd_addr;
250 		uint64_t			wptr_addr;
251 		uint64_t                        h_context;
252 		uint64_t                        h_queue;
253 		enum MES_QUEUE_TYPE		queue_type;
254 		uint32_t			gds_base;
255 		uint32_t			gds_size;
256 		uint32_t			gws_base;
257 		uint32_t			gws_size;
258 		uint32_t			oa_mask;
259 		uint64_t                        trap_handler_addr;
260 		uint32_t                        vm_context_cntl;
261 
262 		struct {
263 			uint32_t paging			: 1;
264 			uint32_t debug_vmid		: 4;
265 			uint32_t program_gds		: 1;
266 			uint32_t is_gang_suspended	: 1;
267 			uint32_t is_tmz_queue		: 1;
268 			uint32_t map_kiq_utility_queue  : 1;
269 			uint32_t is_kfd_process		: 1;
270 			uint32_t reserved		: 22;
271 		};
272 		struct MES_API_STATUS		api_status;
273 		uint64_t                        tma_addr;
274 	};
275 
276 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
277 };
278 
279 union MESAPI__REMOVE_QUEUE {
280 	struct {
281 		union MES_API_HEADER	header;
282 		uint32_t		doorbell_offset;
283 		uint64_t		gang_context_addr;
284 
285 		struct {
286 			uint32_t unmap_legacy_gfx_queue   : 1;
287 			uint32_t unmap_kiq_utility_queue  : 1;
288 			uint32_t preempt_legacy_gfx_queue : 1;
289 			uint32_t reserved                 : 29;
290 		};
291 		struct MES_API_STATUS	    api_status;
292 
293 		uint32_t                    pipe_id;
294 		uint32_t                    queue_id;
295 
296 		uint64_t                    tf_addr;
297 		uint32_t                    tf_data;
298 	};
299 
300 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
301 };
302 
303 union MESAPI__SET_SCHEDULING_CONFIG {
304 	struct {
305 		union MES_API_HEADER	header;
306 		/* Grace period when preempting another priority band for this
307 		 * priority band. The value for idle priority band is ignored,
308 		 * as it never preempts other bands.
309 		 */
310 		uint64_t		grace_period_other_levels[AMD_PRIORITY_NUM_LEVELS];
311 		/* Default quantum for scheduling across processes within
312 		 * a priority band.
313 		 */
314 		uint64_t		process_quantum_for_level[AMD_PRIORITY_NUM_LEVELS];
315 		/* Default grace period for processes that preempt each other
316 		 * within a priority band.
317 		 */
318 		uint64_t		process_grace_period_same_level[AMD_PRIORITY_NUM_LEVELS];
319 		/* For normal level this field specifies the target GPU
320 		 * percentage in situations when it's starved by the high level.
321 		 * Valid values are between 0 and 50, with the default being 10.
322 		 */
323 		uint32_t		normal_yield_percent;
324 		struct MES_API_STATUS	api_status;
325 	};
326 
327 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
328 };
329 
330 union MESAPI__PERFORM_YIELD {
331 	struct {
332 		union MES_API_HEADER	header;
333 		uint32_t		dummy;
334 		struct MES_API_STATUS	api_status;
335 	};
336 
337 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
338 };
339 
340 union MESAPI__CHANGE_GANG_PRIORITY_LEVEL {
341 	struct {
342 		union MES_API_HEADER		header;
343 		uint32_t			inprocess_gang_priority;
344 		enum MES_AMD_PRIORITY_LEVEL	gang_global_priority_level;
345 		uint64_t			gang_quantum;
346 		uint64_t			gang_context_addr;
347 		struct MES_API_STATUS		api_status;
348 	};
349 
350 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
351 };
352 
353 union MESAPI__SUSPEND {
354 	struct {
355 		union MES_API_HEADER	header;
356 		/* false - suspend all gangs; true - specific gang */
357 		struct {
358 			uint32_t suspend_all_gangs	: 1;
359 			uint32_t reserved		: 31;
360 		};
361 		/* gang_context_addr is valid only if suspend_all = false */
362 		uint64_t		gang_context_addr;
363 
364 		uint64_t		suspend_fence_addr;
365 		uint32_t		suspend_fence_value;
366 
367 		struct MES_API_STATUS	api_status;
368 	};
369 
370 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
371 };
372 
373 union MESAPI__RESUME {
374 	struct {
375 		union MES_API_HEADER	header;
376 		/* false - resume all gangs; true - specified gang */
377 		struct {
378 			uint32_t resume_all_gangs	: 1;
379 			uint32_t reserved		: 31;
380 		};
381 		/* valid only if resume_all_gangs = false */
382 		uint64_t		gang_context_addr;
383 
384 		struct MES_API_STATUS	api_status;
385 	};
386 
387 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
388 };
389 
390 union MESAPI__RESET {
391 	struct {
392 		union MES_API_HEADER		header;
393 
394 		struct {
395 			/* Only reset the queue given by doorbell_offset (not entire gang) */
396 			uint32_t                reset_queue_only : 1;
397 			/* Hang detection first then reset any queues that are hung */
398 			uint32_t                hang_detect_then_reset : 1;
399 			/* Only do hang detection (no reset) */
400 			uint32_t                hang_detect_only : 1;
401 			/* Rest HP and LP kernel queues not managed by MES */
402 			uint32_t                reset_legacy_gfx : 1;
403 			uint32_t                reserved : 28;
404 		};
405 
406 		uint64_t			gang_context_addr;
407 
408 		/* valid only if reset_queue_only = true */
409 		uint32_t			doorbell_offset;
410 
411 		/* valid only if hang_detect_then_reset = true */
412 		uint64_t			doorbell_offset_addr;
413 		enum MES_QUEUE_TYPE		queue_type;
414 
415 		/* valid only if reset_legacy_gfx = true */
416 		uint32_t			pipe_id_lp;
417 		uint32_t			queue_id_lp;
418 		uint32_t			vmid_id_lp;
419 		uint64_t			mqd_mc_addr_lp;
420 		uint32_t			doorbell_offset_lp;
421 		uint64_t			wptr_addr_lp;
422 
423 		uint32_t			pipe_id_hp;
424 		uint32_t			queue_id_hp;
425 		uint32_t			vmid_id_hp;
426 		uint64_t			mqd_mc_addr_hp;
427 		uint32_t			doorbell_offset_hp;
428 		uint64_t			wptr_addr_hp;
429 
430 		struct MES_API_STATUS		api_status;
431 	};
432 
433 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
434 };
435 
436 union MESAPI__SET_LOGGING_BUFFER {
437 	struct {
438 		union MES_API_HEADER	header;
439 		/* There are separate log buffers for each queue type */
440 		enum MES_QUEUE_TYPE	log_type;
441 		/* Log buffer GPU Address */
442 		uint64_t		logging_buffer_addr;
443 		/* number of entries in the log buffer */
444 		uint32_t		number_of_entries;
445 		/* Entry index at which CPU interrupt needs to be signalled */
446 		uint32_t		interrupt_entry;
447 
448 		struct MES_API_STATUS	api_status;
449 	};
450 
451 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
452 };
453 
454 union MESAPI__QUERY_MES_STATUS {
455 	struct {
456 		union MES_API_HEADER	header;
457 		bool			mes_healthy; /* 0 - not healthy, 1 - healthy */
458 		struct MES_API_STATUS	api_status;
459 	};
460 
461 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
462 };
463 
464 union MESAPI__PROGRAM_GDS {
465 	struct {
466 		union MES_API_HEADER	header;
467 		uint64_t		process_context_addr;
468 		uint32_t		gds_base;
469 		uint32_t		gds_size;
470 		uint32_t		gws_base;
471 		uint32_t		gws_size;
472 		uint32_t		oa_mask;
473 		struct MES_API_STATUS	api_status;
474 	};
475 
476 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
477 };
478 
479 union MESAPI__SET_DEBUG_VMID {
480 	struct {
481 		union MES_API_HEADER	header;
482 		struct MES_API_STATUS	api_status;
483 		union {
484 			struct {
485 				uint32_t use_gds	: 1;
486 				uint32_t operation      : 2;
487 				uint32_t reserved       : 29;
488 			} flags;
489 			uint32_t	u32All;
490 		};
491 		uint32_t		reserved;
492 		uint32_t		debug_vmid;
493 		uint64_t		process_context_addr;
494 		uint64_t		page_table_base_addr;
495 		uint64_t		process_va_start;
496 		uint64_t		process_va_end;
497 		uint32_t		gds_base;
498 		uint32_t		gds_size;
499 		uint32_t		gws_base;
500 		uint32_t		gws_size;
501 		uint32_t		oa_mask;
502 
503 		/* output addr of the acquired vmid value */
504 		uint64_t                output_addr;
505 	};
506 
507 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
508 };
509 
510 enum MESAPI_MISC_OPCODE {
511 	MESAPI_MISC__WRITE_REG,
512 	MESAPI_MISC__INV_GART,
513 	MESAPI_MISC__QUERY_STATUS,
514 	MESAPI_MISC__READ_REG,
515 	MESAPI_MISC__WAIT_REG_MEM,
516 	MESAPI_MISC__MAX,
517 };
518 
519 enum { MISC_DATA_MAX_SIZE_IN_DWORDS = 20 };
520 
521 struct WRITE_REG {
522 	uint32_t                  reg_offset;
523 	uint32_t                  reg_value;
524 };
525 
526 struct READ_REG {
527 	uint32_t                  reg_offset;
528 	uint64_t                  buffer_addr;
529 };
530 
531 enum WRM_OPERATION {
532 	WRM_OPERATION__WAIT_REG_MEM,
533 	WRM_OPERATION__WR_WAIT_WR_REG,
534 	WRM_OPERATION__MAX,
535 };
536 
537 struct WAIT_REG_MEM {
538 	enum WRM_OPERATION         op;
539 	uint32_t                   reference;
540 	uint32_t                   mask;
541 	uint32_t                   reg_offset1;
542 	uint32_t                   reg_offset2;
543 };
544 
545 struct INV_GART {
546 	uint64_t                  inv_range_va_start;
547 	uint64_t                  inv_range_size;
548 };
549 
550 struct QUERY_STATUS {
551 	uint32_t context_id;
552 };
553 
554 union MESAPI__MISC {
555 	struct {
556 		union MES_API_HEADER	header;
557 		enum MESAPI_MISC_OPCODE	opcode;
558 		struct MES_API_STATUS	api_status;
559 
560 		union {
561 			struct		WRITE_REG write_reg;
562 			struct		INV_GART inv_gart;
563 			struct		QUERY_STATUS query_status;
564 			struct		READ_REG read_reg;
565 			struct          WAIT_REG_MEM wait_reg_mem;
566 			uint32_t	data[MISC_DATA_MAX_SIZE_IN_DWORDS];
567 		};
568 	};
569 
570 	uint32_t	max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
571 };
572 
573 union MESAPI__UPDATE_ROOT_PAGE_TABLE {
574 	struct {
575 		union MES_API_HEADER        header;
576 		uint64_t                    page_table_base_addr;
577 		uint64_t                    process_context_addr;
578 		struct MES_API_STATUS       api_status;
579 	};
580 
581 	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
582 };
583 
584 union MESAPI_AMD_LOG {
585 	struct {
586 		union MES_API_HEADER        header;
587 		uint64_t                    p_buffer_memory;
588 		uint64_t                    p_buffer_size_used;
589 		struct MES_API_STATUS       api_status;
590 	};
591 
592 	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
593 };
594 
595 #pragma pack(pop)
596 #endif
597