1*4882a593Smuzhiyun /* SPDX-License-Identifier: GPL-2.0 */ 2*4882a593Smuzhiyun /* 3*4882a593Smuzhiyun * Copyright (C) Rockchip Electronics Co.Ltd 4*4882a593Smuzhiyun * Author: Felix Zeng <felix.zeng@rock-chips.com> 5*4882a593Smuzhiyun */ 6*4882a593Smuzhiyun 7*4882a593Smuzhiyun #ifndef __LINUX_RKNPU_MM_H 8*4882a593Smuzhiyun #define __LINUX_RKNPU_MM_H 9*4882a593Smuzhiyun 10*4882a593Smuzhiyun #include <linux/mutex.h> 11*4882a593Smuzhiyun #include <linux/seq_file.h> 12*4882a593Smuzhiyun #include <linux/iommu.h> 13*4882a593Smuzhiyun #include <linux/iova.h> 14*4882a593Smuzhiyun 15*4882a593Smuzhiyun #include "rknpu_drv.h" 16*4882a593Smuzhiyun 17*4882a593Smuzhiyun struct rknpu_mm { 18*4882a593Smuzhiyun void *bitmap; 19*4882a593Smuzhiyun struct mutex lock; 20*4882a593Smuzhiyun unsigned int chunk_size; 21*4882a593Smuzhiyun unsigned int total_chunks; 22*4882a593Smuzhiyun unsigned int free_chunks; 23*4882a593Smuzhiyun }; 24*4882a593Smuzhiyun 25*4882a593Smuzhiyun struct rknpu_mm_obj { 26*4882a593Smuzhiyun uint32_t range_start; 27*4882a593Smuzhiyun uint32_t range_end; 28*4882a593Smuzhiyun }; 29*4882a593Smuzhiyun 30*4882a593Smuzhiyun int rknpu_mm_create(unsigned int mem_size, unsigned int chunk_size, 31*4882a593Smuzhiyun struct rknpu_mm **mm); 32*4882a593Smuzhiyun 33*4882a593Smuzhiyun void rknpu_mm_destroy(struct rknpu_mm *mm); 34*4882a593Smuzhiyun 35*4882a593Smuzhiyun int rknpu_mm_alloc(struct rknpu_mm *mm, unsigned int size, 36*4882a593Smuzhiyun struct rknpu_mm_obj **mm_obj); 37*4882a593Smuzhiyun 38*4882a593Smuzhiyun int rknpu_mm_free(struct rknpu_mm *mm, struct rknpu_mm_obj *mm_obj); 39*4882a593Smuzhiyun 40*4882a593Smuzhiyun int rknpu_mm_dump(struct seq_file *m, void *data); 41*4882a593Smuzhiyun 42*4882a593Smuzhiyun #endif 43