root/sys/dev/pci/drm/amd/amdgpu/amdgpu_doorbell_mgr.c
// SPDX-License-Identifier: MIT
/*
 * Copyright 2022 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.
 *
 */

#include "amdgpu.h"

/**
 * amdgpu_mm_rdoorbell - read a doorbell dword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 *
 * Returns the value in the doorbell aperture at the
 * requested doorbell index (CIK).
 */
u32 amdgpu_mm_rdoorbell(struct amdgpu_device *adev, u32 index)
{
        if (amdgpu_device_skip_hw_access(adev))
                return 0;

        if (index < adev->doorbell.num_kernel_doorbells)
                return readl(adev->doorbell.cpu_addr + index);

        dev_err(adev->dev, "reading beyond doorbell aperture: 0x%08x!\n",
                index);
        return 0;
}

/**
 * amdgpu_mm_wdoorbell - write a doorbell dword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 * @v: value to write
 *
 * Writes @v to the doorbell aperture at the
 * requested doorbell index (CIK).
 */
void amdgpu_mm_wdoorbell(struct amdgpu_device *adev, u32 index, u32 v)
{
        if (amdgpu_device_skip_hw_access(adev))
                return;

        if (index < adev->doorbell.num_kernel_doorbells)
                writel(v, adev->doorbell.cpu_addr + index);
        else
                dev_err(adev->dev,
                        "writing beyond doorbell aperture: 0x%08x!\n", index);
}

/**
 * amdgpu_mm_rdoorbell64 - read a doorbell Qword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 *
 * Returns the value in the doorbell aperture at the
 * requested doorbell index (VEGA10+).
 */
u64 amdgpu_mm_rdoorbell64(struct amdgpu_device *adev, u32 index)
{
        if (amdgpu_device_skip_hw_access(adev))
                return 0;

        if (index < adev->doorbell.num_kernel_doorbells)
                return atomic64_read((atomic64_t *)(adev->doorbell.cpu_addr + index));

        dev_err(adev->dev, "reading beyond doorbell aperture: 0x%08x!\n",
                index);
        return 0;
}

/**
 * amdgpu_mm_wdoorbell64 - write a doorbell Qword
 *
 * @adev: amdgpu_device pointer
 * @index: doorbell index
 * @v: value to write
 *
 * Writes @v to the doorbell aperture at the
 * requested doorbell index (VEGA10+).
 */
void amdgpu_mm_wdoorbell64(struct amdgpu_device *adev, u32 index, u64 v)
{
        if (amdgpu_device_skip_hw_access(adev))
                return;

        if (index < adev->doorbell.num_kernel_doorbells)
                atomic64_set((atomic64_t *)(adev->doorbell.cpu_addr + index), v);
        else
                dev_err(adev->dev,
                        "writing beyond doorbell aperture: 0x%08x!\n", index);
}

/**
 * amdgpu_doorbell_index_on_bar - Find doorbell's absolute offset in BAR
 *
 * @adev: amdgpu_device pointer
 * @db_bo: doorbell object's bo
 * @doorbell_index: doorbell relative index in this doorbell object
 * @db_size: doorbell size is in byte
 *
 * returns doorbell's absolute index in BAR
 */
uint32_t amdgpu_doorbell_index_on_bar(struct amdgpu_device *adev,
                                      struct amdgpu_bo *db_bo,
                                      uint32_t doorbell_index,
                                      uint32_t db_size)
{
        int db_bo_offset;

        db_bo_offset = amdgpu_bo_gpu_offset_no_check(db_bo);

        /* doorbell index is 32 bit but doorbell's size can be 32 bit
         * or 64 bit, so *db_size(in byte)/4 for alignment.
         */
        return db_bo_offset / sizeof(u32) + doorbell_index *
               DIV_ROUND_UP(db_size, 4);
}

/**
 * amdgpu_doorbell_create_kernel_doorbells - Create kernel doorbells for graphics
 *
 * @adev: amdgpu_device pointer
 *
 * Creates doorbells for graphics driver usages.
 * returns 0 on success, error otherwise.
 */
int amdgpu_doorbell_create_kernel_doorbells(struct amdgpu_device *adev)
{
        int r;
        int size;

        /* SI HW does not have doorbells, skip allocation */
        if (adev->doorbell.num_kernel_doorbells == 0)
                return 0;

        /* Reserve first num_kernel_doorbells (page-aligned) for kernel ops */
        size = ALIGN(adev->doorbell.num_kernel_doorbells * sizeof(u32), PAGE_SIZE);

        /* Allocate an extra page for MES kernel usages (ring test) */
        adev->mes.db_start_dw_offset = size / sizeof(u32);
        size += PAGE_SIZE;

        r = amdgpu_bo_create_kernel(adev,
                                    size,
                                    PAGE_SIZE,
                                    AMDGPU_GEM_DOMAIN_DOORBELL,
                                    &adev->doorbell.kernel_doorbells,
                                    NULL,
                                    (void **)&adev->doorbell.cpu_addr);
        if (r) {
                dev_err(adev->dev,
                        "Failed to allocate kernel doorbells, err=%d\n", r);
                return r;
        }

        adev->doorbell.num_kernel_doorbells = size / sizeof(u32);
        return 0;
}

/*
 * GPU doorbell aperture helpers function.
 */
/**
 * amdgpu_doorbell_init - Init doorbell driver information.
 *
 * @adev: amdgpu_device pointer
 *
 * Init doorbell driver information (CIK)
 * Returns 0 on success, error on failure.
 */
int amdgpu_doorbell_init(struct amdgpu_device *adev)
{

        /* No doorbell on SI hardware generation */
        if (adev->asic_type < CHIP_BONAIRE) {
                adev->doorbell.base = 0;
                adev->doorbell.size = 0;
                adev->doorbell.num_kernel_doorbells = 0;
                return 0;
        }

#ifdef __linux__
        if (pci_resource_flags(adev->pdev, 2) & IORESOURCE_UNSET)
                return -EINVAL;
#endif

        amdgpu_asic_init_doorbell_index(adev);

        /* doorbell bar mapping */
#ifdef __linux__
        adev->doorbell.base = pci_resource_start(adev->pdev, 2);
        adev->doorbell.size = pci_resource_len(adev->pdev, 2);
#else
        {
                pcireg_t mtype;
                mtype = pci_mapreg_type(adev->pdev->pc, adev->pdev->tag, 0x18);
                if (pci_mapreg_info(adev->pdev->pc, adev->pdev->tag, 0x18,
                    mtype, &adev->doorbell.base, &adev->doorbell.size, NULL))
                        return -EINVAL;
        }
#endif

        adev->doorbell.num_kernel_doorbells =
                min_t(u32, adev->doorbell.size / sizeof(u32),
                      adev->doorbell_index.max_assignment + 1);
        if (adev->doorbell.num_kernel_doorbells == 0)
                return -EINVAL;

        /*
         * For Vega, reserve and map two pages on doorbell BAR since SDMA
         * paging queue doorbell use the second page. The
         * AMDGPU_DOORBELL64_MAX_ASSIGNMENT definition assumes all the
         * doorbells are in the first page. So with paging queue enabled,
         * the max num_kernel_doorbells should + 1 page (0x400 in dword)
         */
        if (adev->asic_type >= CHIP_VEGA10)
                adev->doorbell.num_kernel_doorbells += 0x400;

        return 0;
}

/**
 * amdgpu_doorbell_fini - Tear down doorbell driver information.
 *
 * @adev: amdgpu_device pointer
 *
 * Tear down doorbell driver information (CIK)
 */
void amdgpu_doorbell_fini(struct amdgpu_device *adev)
{
        amdgpu_bo_free_kernel(&adev->doorbell.kernel_doorbells,
                              NULL,
                              (void **)&adev->doorbell.cpu_addr);
}