/* SPDX-License-Identifier: MIT */
/*
 * Copyright 2023 Advanced Micro Devices, Inc.
 *
 * Permission is hereby granted, free of charge, to any person obtaining a
 * copy of this software and associated documentation files (the "Software"),
 * to deal in the Software without restriction, including without limitation
 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
 * and/or sell copies of the Software, and to permit persons to whom the
 * Software is furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
 * OTHER DEALINGS IN THE SOFTWARE.
 *
 */

#ifndef __UMSCH_MM_API_DEF_H__
#define __UMSCH_MM_API_DEF_H__

#pragma once

#pragma pack(push, 4)

#define UMSCH_API_VERSION 1

/*
 * Driver submits one API(cmd) as a single Frame and this command size is same for all API
 * to ease the debugging and parsing of ring buffer.
 */
enum { API_FRAME_SIZE_IN_DWORDS = 64 };

/*
 * To avoid command in scheduler context to be overwritten whenever multiple interrupts come in,
 * this creates another queue.
 */
enum { API_NUMBER_OF_COMMAND_MAX = 32 };

enum { UMSCH_INSTANCE_DB_OFFSET_MAX = 16 };

enum UMSCH_API_TYPE {
	UMSCH_API_TYPE_SCHEDULER = 1,
	UMSCH_API_TYPE_MAX
};

enum UMSCH_MS_LOG_CONTEXT_STATE {
	UMSCH_LOG_CONTEXT_STATE_IDLE = 0,
	UMSCH_LOG_CONTEXT_STATE_RUNNING = 1,
	UMSCH_LOG_CONTEXT_STATE_READY = 2,
	UMSCH_LOG_CONTEXT_STATE_READY_STANDBY = 3,
	UMSCH_LOG_CONTEXT_STATE_INVALID = 0xF,
};

enum UMSCH_MS_LOG_OPERATION {
	UMSCH_LOG_OPERATION_CONTEXT_STATE_CHANGE = 0,
	UMSCH_LOG_OPERATION_QUEUE_NEW_WORK = 1,
	UMSCH_LOG_OPERATION_QUEUE_UNWAIT_SYNC_OBJECT = 2,
	UMSCH_LOG_OPERATION_QUEUE_NO_MORE_WORK = 3,
	UMSCH_LOG_OPERATION_QUEUE_WAIT_SYNC_OBJECT = 4,
	UMSCH_LOG_OPERATION_QUEUE_INVALID = 0xF,
};

struct UMSCH_INSTANCE_DB_OFFSET {
	uint32_t instance_index;
	uint32_t doorbell_offset;
};

struct UMSCH_LOG_CONTEXT_STATE_CHANGE {
	uint64_t h_context;
	enum UMSCH_MS_LOG_CONTEXT_STATE new_context_state;
};

struct UMSCH_LOG_QUEUE_NEW_WORK {
	uint64_t h_queue;
	uint64_t reserved;
};

struct UMSCH_LOG_QUEUE_UNWAIT_SYNC_OBJECT {
	uint64_t h_queue;
	uint64_t h_sync_object;
};

struct UMSCH_LOG_QUEUE_NO_MORE_WORK {
	uint64_t h_queue;
	uint64_t reserved;
};

struct UMSCH_LOG_QUEUE_WAIT_SYNC_OBJECT {
	uint64_t h_queue;
	uint64_t h_sync_object;
};

struct UMSCH_LOG_ENTRY_HEADER {
	uint32_t first_free_entry_index;
	uint32_t wraparound_count;
	uint64_t number_of_entries;
	uint64_t reserved[2];
};

struct UMSCH_LOG_ENTRY_DATA {
	uint64_t gpu_time_stamp;
	uint32_t operation_type; /* operation_type is of UMSCH_LOG_OPERATION type */
	uint32_t reserved_operation_type_bits;
	union {
		struct UMSCH_LOG_CONTEXT_STATE_CHANGE context_state_change;
		struct UMSCH_LOG_QUEUE_NEW_WORK queue_new_work;
		struct UMSCH_LOG_QUEUE_UNWAIT_SYNC_OBJECT queue_unwait_sync_object;
		struct UMSCH_LOG_QUEUE_NO_MORE_WORK queue_no_more_work;
		struct UMSCH_LOG_QUEUE_WAIT_SYNC_OBJECT queue_wait_sync_object;
		uint64_t all[2];
	};
};

struct UMSCH_LOG_BUFFER {
	struct UMSCH_LOG_ENTRY_HEADER header;
	struct UMSCH_LOG_ENTRY_DATA entries[1];
};

enum UMSCH_API_OPCODE {
	UMSCH_API_SET_HW_RSRC = 0x00,
	UMSCH_API_SET_SCHEDULING_CONFIG = 0x1,
	UMSCH_API_ADD_QUEUE = 0x2,
	UMSCH_API_REMOVE_QUEUE = 0x3,
	UMSCH_API_PERFORM_YIELD = 0x4,
	UMSCH_API_SUSPEND = 0x5,
	UMSCH_API_RESUME = 0x6,
	UMSCH_API_RESET = 0x7,
	UMSCH_API_SET_LOG_BUFFER = 0x8,
	UMSCH_API_CHANGE_CONTEXT_PRIORITY = 0x9,
	UMSCH_API_QUERY_SCHEDULER_STATUS = 0xA,
	UMSCH_API_UPDATE_AFFINITY = 0xB,
	UMSCH_API_MAX = 0xFF
};

union UMSCH_API_HEADER {
	struct {
		uint32_t type : 4; /* 0 - Invalid; 1 - Scheduling; 2 - TBD */
		uint32_t opcode : 8;
		uint32_t dwsize : 8;
		uint32_t reserved : 12;
	};

	uint32_t u32All;
};

enum UMSCH_AMD_PRIORITY_LEVEL {
	AMD_PRIORITY_LEVEL_IDLE = 0,
	AMD_PRIORITY_LEVEL_NORMAL = 1,
	AMD_PRIORITY_LEVEL_FOCUS = 2,
	AMD_PRIORITY_LEVEL_REALTIME = 3,
	AMD_PRIORITY_NUM_LEVELS
};

enum UMSCH_ENGINE_TYPE {
	UMSCH_ENGINE_TYPE_VCN0 = 0,
	UMSCH_ENGINE_TYPE_VCN1 = 1,
	UMSCH_ENGINE_TYPE_VCN = 2,
	UMSCH_ENGINE_TYPE_VPE = 3,
	UMSCH_ENGINE_TYPE_MAX
};

#define AFFINITY_DISABLE 0
#define AFFINITY_ENABLE 1
#define AFFINITY_MAX 2

union UMSCH_AFFINITY {
	struct {
		unsigned int vcn0Affinity : 2; /* enable 1 disable 0 */
		unsigned int vcn1Affinity : 2;
		unsigned int reserved : 28;
	};
	unsigned int u32All;
};

struct UMSCH_API_STATUS {
	uint64_t api_completion_fence_addr;
	uint32_t api_completion_fence_value;
};

enum { MAX_VCN0_INSTANCES = 1 };
enum { MAX_VCN1_INSTANCES = 1 };
enum { MAX_VCN_INSTANCES = 2 };

enum { MAX_VPE_INSTANCES = 1 };

enum { MAX_VCN_QUEUES = 4 };
enum { MAX_VPE_QUEUES = 8 };

enum { MAX_QUEUES_IN_A_CONTEXT = 1 };

enum { UMSCH_MAX_HWIP_SEGMENT = 8 };

enum VM_HUB_TYPE {
	VM_HUB_TYPE_GC = 0,
	VM_HUB_TYPE_MM = 1,
	VM_HUB_TYPE_MAX,
};

enum { VMID_INVALID = 0xffff };

enum { MAX_VMID_MMHUB = 16 };

union UMSCHAPI__SET_HW_RESOURCES {
	struct {
		union UMSCH_API_HEADER header;
		uint32_t vmid_mask_mm_vcn;
		uint32_t vmid_mask_mm_vpe;
		uint32_t collaboration_mask_vpe;
		uint32_t engine_mask;
		uint32_t logging_vmid;
		uint32_t vcn0_hqd_mask[MAX_VCN0_INSTANCES];
		uint32_t vcn1_hqd_mask[MAX_VCN1_INSTANCES];
		uint32_t vcn_hqd_mask[MAX_VCN_INSTANCES];
		uint32_t vpe_hqd_mask[MAX_VPE_INSTANCES];
		uint64_t g_sch_ctx_gpu_mc_ptr;
		uint32_t mmhub_base[UMSCH_MAX_HWIP_SEGMENT];
		uint32_t mmhub_version;
		uint32_t osssys_base[UMSCH_MAX_HWIP_SEGMENT];
		uint32_t osssys_version;
		uint32_t vcn_version;
		uint32_t vpe_version;
		struct UMSCH_API_STATUS api_status;
		union {
			struct {
				uint32_t disable_reset : 1;
				uint32_t disable_umsch_log : 1;
				uint32_t enable_level_process_quantum_check : 1;
				uint32_t is_vcn0_enabled : 1;
				uint32_t is_vcn1_enabled : 1;
				uint32_t use_rs64mem_for_proc_ctx_csa : 1;
				uint32_t reserved : 26;
			};
			uint32_t uint32_all;
		};
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};
static_assert(sizeof(union UMSCHAPI__SET_HW_RESOURCES) <= API_FRAME_SIZE_IN_DWORDS * sizeof(uint32_t),
			  "size of UMSCHAPI__SET_HW_RESOURCES must be less than 256 bytes");

union UMSCHAPI__SET_SCHEDULING_CONFIG {
	struct {
		union UMSCH_API_HEADER header;
		/*
		 * Grace period when preempting another priority band for this priority band.
		 * The value for idle priority band is ignored, as it never preempts other bands.
		 */
		uint64_t grace_period_other_levels[AMD_PRIORITY_NUM_LEVELS];

		/* Default quantum for scheduling across processes within a priority band. */
		uint64_t process_quantum_for_level[AMD_PRIORITY_NUM_LEVELS];

		/* Default grace period for processes that preempt each other within a priority band. */
		uint64_t process_grace_period_same_level[AMD_PRIORITY_NUM_LEVELS];

		/*
		 * For normal level this field specifies the target GPU percentage in situations
		 * when it's starved by the high level. Valid values are between 0 and 50,
		 * with the default being 10.
		 */
		uint32_t normal_yield_percent;

		struct UMSCH_API_STATUS api_status;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__ADD_QUEUE {
	struct {
		union UMSCH_API_HEADER header;
		uint32_t process_id;
		uint64_t page_table_base_addr;
		uint64_t process_va_start;
		uint64_t process_va_end;
		uint64_t process_quantum;
		uint64_t process_csa_addr;
		uint64_t context_quantum;
		uint64_t context_csa_addr;
		uint32_t inprocess_context_priority;
		enum UMSCH_AMD_PRIORITY_LEVEL context_global_priority_level;
		uint32_t doorbell_offset_0;
		uint32_t doorbell_offset_1;
		union UMSCH_AFFINITY affinity;
		uint64_t mqd_addr;
		uint64_t h_context;
		uint64_t h_queue;
		enum UMSCH_ENGINE_TYPE engine_type;
		uint32_t vm_context_cntl;

		struct {
			uint32_t is_context_suspended : 1;
			uint32_t collaboration_mode : 1;
			uint32_t reserved : 30;
		};
		struct UMSCH_API_STATUS api_status;
		uint32_t process_csa_array_index;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};


union UMSCHAPI__REMOVE_QUEUE {
	struct {
		union UMSCH_API_HEADER header;
		uint32_t doorbell_offset_0;
		uint32_t doorbell_offset_1;
		uint64_t context_csa_addr;

		struct UMSCH_API_STATUS api_status;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__PERFORM_YIELD {
	struct {
		union UMSCH_API_HEADER header;
		uint32_t dummy;
		struct UMSCH_API_STATUS api_status;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__SUSPEND {
	struct {
		union UMSCH_API_HEADER header;
		uint64_t context_csa_addr;
		uint64_t suspend_fence_addr;
		uint32_t suspend_fence_value;

		struct UMSCH_API_STATUS api_status;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

enum UMSCH_RESUME_OPTION {
	CONTEXT_RESUME = 0,
	ENGINE_SCHEDULE_RESUME = 1,
};

union UMSCHAPI__RESUME {
	struct {
		union UMSCH_API_HEADER header;

		enum UMSCH_RESUME_OPTION resume_option;
		uint64_t context_csa_addr; /* valid only for UMSCH_SWIP_CONTEXT_RESUME */
		enum UMSCH_ENGINE_TYPE engine_type;

		struct UMSCH_API_STATUS api_status;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

enum UMSCH_RESET_OPTION {
	HANG_DETECT_AND_RESET = 0,
	HANG_DETECT_ONLY = 1,
};

union UMSCHAPI__RESET {
	struct {
		union UMSCH_API_HEADER header;

		enum UMSCH_RESET_OPTION reset_option;
		uint64_t doorbell_offset_addr;
		enum UMSCH_ENGINE_TYPE engine_type;

		struct UMSCH_API_STATUS api_status;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__SET_LOGGING_BUFFER {
	struct {
		union UMSCH_API_HEADER header;
		/* There are separate log buffers for each queue type */
		enum UMSCH_ENGINE_TYPE log_type;
		/* Log buffer GPU Address */
		uint64_t logging_buffer_addr;
		/* Number of entries in the log buffer */
		uint32_t number_of_entries;
		/* Entry index at which CPU interrupt needs to be signalled */
		uint32_t interrupt_entry;

		struct UMSCH_API_STATUS api_status;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__UPDATE_AFFINITY {
	struct {
		union UMSCH_API_HEADER header;
		union UMSCH_AFFINITY affinity;
		uint64_t context_csa_addr;
		struct UMSCH_API_STATUS api_status;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__CHANGE_CONTEXT_PRIORITY_LEVEL {
	struct {
		union UMSCH_API_HEADER header;
		uint32_t inprocess_context_priority;
		enum UMSCH_AMD_PRIORITY_LEVEL context_global_priority_level;
		uint64_t context_quantum;
		uint64_t context_csa_addr;
		struct UMSCH_API_STATUS api_status;
		uint32_t context_csa_array_index;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

union UMSCHAPI__QUERY_UMSCH_STATUS {
	struct {
		union UMSCH_API_HEADER header;
		bool umsch_mm_healthy; /* 0 - not healthy, 1 - healthy */
		struct UMSCH_API_STATUS api_status;
	};

	uint32_t max_dwords_in_api[API_FRAME_SIZE_IN_DWORDS];
};

#pragma pack(pop)

#endif