/* SPDX-License-Identifier: MIT
 *
 * Copyright (c) 2025, NVIDIA CORPORATION. All rights reserved.
 */
#include <rm/rm.h>

#include "nvrm/gsp.h"

static const struct nvkm_rm_wpr
r570_wpr_libos2 = {
	.os_carveout_size = GSP_FW_HEAP_PARAM_OS_SIZE_LIBOS2,
	.base_size = GSP_FW_HEAP_PARAM_BASE_RM_SIZE_TU10X,
	.heap_size_min = GSP_FW_HEAP_SIZE_OVERRIDE_LIBOS2_MIN_MB,
};

static const struct nvkm_rm_wpr
r570_wpr_libos3 = {
	.os_carveout_size = GSP_FW_HEAP_PARAM_OS_SIZE_LIBOS3_BAREMETAL,
	.base_size = GSP_FW_HEAP_PARAM_BASE_RM_SIZE_TU10X,
	.heap_size_min = GSP_FW_HEAP_SIZE_OVERRIDE_LIBOS3_BAREMETAL_MIN_MB,
};

static const struct nvkm_rm_wpr
r570_wpr_libos3_gh100 = {
	.os_carveout_size = GSP_FW_HEAP_PARAM_OS_SIZE_LIBOS3_BAREMETAL,
	.base_size = GSP_FW_HEAP_PARAM_BASE_RM_SIZE_GH100,
	.heap_size_min = GSP_FW_HEAP_SIZE_OVERRIDE_LIBOS3_BAREMETAL_MIN_MB,
	.heap_size_non_wpr = 0x200000,
	.offset_set_by_acr = true,
};

static const struct nvkm_rm_wpr
r570_wpr_libos3_gb10x = {
	.os_carveout_size = GSP_FW_HEAP_PARAM_OS_SIZE_LIBOS3_BAREMETAL,
	.base_size = GSP_FW_HEAP_PARAM_BASE_RM_SIZE_GH100,
	.heap_size_min = GSP_FW_HEAP_SIZE_OVERRIDE_LIBOS3_BAREMETAL_MIN_MB,
	.heap_size_non_wpr = 0x200000,
	.rsvd_size_pmu = ALIGN(0x0800000 + 0x1000000 + 0x0001000, 0x20000),
	.offset_set_by_acr = true,
};

static const struct nvkm_rm_wpr
r570_wpr_libos3_gb20x = {
	.os_carveout_size = GSP_FW_HEAP_PARAM_OS_SIZE_LIBOS3_BAREMETAL,
	.base_size = GSP_FW_HEAP_PARAM_BASE_RM_SIZE_GH100,
	.heap_size_min = GSP_FW_HEAP_SIZE_OVERRIDE_LIBOS3_BAREMETAL_MIN_MB,
	.heap_size_non_wpr = 0x220000,
	.rsvd_size_pmu = ALIGN(0x0800000 + 0x1000000 + 0x0001000, 0x20000),
	.offset_set_by_acr = true,
};

static const struct nvkm_rm_api
r570_api = {
	.gsp = &r570_gsp,
	.rpc = &r535_rpc,
	.ctrl = &r535_ctrl,
	.alloc = &r535_alloc,
	.client = &r570_client,
	.device = &r535_device,
	.fbsr = &r570_fbsr,
	.disp = &r570_disp,
	.fifo = &r570_fifo,
	.ce = &r535_ce,
	.gr = &r570_gr,
	.nvdec = &r535_nvdec,
	.nvenc = &r535_nvenc,
	.nvjpg = &r535_nvjpg,
	.ofa = &r570_ofa,
};

const struct nvkm_rm_impl
r570_rm_tu102 = {
	.wpr = &r570_wpr_libos2,
	.api = &r570_api,
};

const struct nvkm_rm_impl
r570_rm_ga102 = {
	.wpr = &r570_wpr_libos3,
	.api = &r570_api,
};

const struct nvkm_rm_impl
r570_rm_gh100 = {
	.wpr = &r570_wpr_libos3_gh100,
	.api = &r570_api,
};

const struct nvkm_rm_impl
r570_rm_gb10x = {
	.wpr = &r570_wpr_libos3_gb10x,
	.api = &r570_api,
};

const struct nvkm_rm_impl
r570_rm_gb20x = {
	.wpr = &r570_wpr_libos3_gb20x,
	.api = &r570_api,
};