/* SPDX-License-Identifier: GPL-2.0 */
|
/*
|
* Copyright (C) Rockchip Electronics Co.Ltd
|
* Author: Felix Zeng <felix.zeng@rock-chips.com>
|
*/
|
|
#ifndef __LINUX_RKNPU_MM_H
|
#define __LINUX_RKNPU_MM_H
|
|
#include <linux/mutex.h>
|
#include <linux/seq_file.h>
|
#include <linux/iommu.h>
|
#include <linux/dma-iommu.h>
|
#include <linux/iova.h>
|
|
#include "rknpu_drv.h"
|
|
struct rknpu_mm {
|
void *bitmap;
|
struct mutex lock;
|
unsigned int chunk_size;
|
unsigned int total_chunks;
|
unsigned int free_chunks;
|
};
|
|
struct rknpu_mm_obj {
|
uint32_t range_start;
|
uint32_t range_end;
|
};
|
|
int rknpu_mm_create(unsigned int mem_size, unsigned int chunk_size,
|
struct rknpu_mm **mm);
|
|
void rknpu_mm_destroy(struct rknpu_mm *mm);
|
|
int rknpu_mm_alloc(struct rknpu_mm *mm, unsigned int size,
|
struct rknpu_mm_obj **mm_obj);
|
|
int rknpu_mm_free(struct rknpu_mm *mm, struct rknpu_mm_obj *mm_obj);
|
|
int rknpu_mm_dump(struct seq_file *m, void *data);
|
|
enum iommu_dma_cookie_type {
|
IOMMU_DMA_IOVA_COOKIE,
|
IOMMU_DMA_MSI_COOKIE,
|
};
|
|
struct rknpu_iommu_dma_cookie {
|
enum iommu_dma_cookie_type type;
|
|
/* Full allocator for IOMMU_DMA_IOVA_COOKIE */
|
struct iova_domain iovad;
|
};
|
|
dma_addr_t rknpu_iommu_dma_alloc_iova(struct iommu_domain *domain, size_t size,
|
u64 dma_limit, struct device *dev);
|
|
void rknpu_iommu_dma_free_iova(struct rknpu_iommu_dma_cookie *cookie,
|
dma_addr_t iova, size_t size);
|
|
#endif
|
