summaryrefslogtreecommitdiffstats
path: root/kernel/include/memory.h
blob: 14a35e95dad08147a535c0b86bfcb305f73e2788 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
#ifndef _NV_VGPU_MEMORY_H_
#define _NV_VGPU_MEMORY_H_

#include "nvidia-vgpu-vfio.h"

#define NV_VGPU_MAX_MAPPINGS 1024

struct nv_vgpu_mapping {
	u64 gpa;
	u64 iova;
	u64 size;
	struct page **pages;
	int page_count;
	bool pinned;
	bool mapped;
	struct list_head list;
};

struct nv_vgpu_mm_ctx {
	struct nv_vgpu_device *vgpu;
	struct mutex lock;
	struct list_head mappings;
	u32 mapping_count;
	struct iommu_domain *iommu_domain;
	struct dma_iommu_mapping *dma_mapping;
	bool use_simple;
};

int nv_vgpu_memory_init(struct nv_vgpu_device *vgpu);
void nv_vgpu_memory_cleanup(struct nv_vgpu_device *vgpu);

int nv_vgpu_memory_map(struct nv_vgpu_device *vgpu, u64 gpa, u64 size, u64 *iova);
int nv_vgpu_memory_unmap(struct nv_vgpu_device *vgpu, u64 iova, u64 size);

int nv_vgpu_memory_pin(struct nv_vgpu_device *vgpu, u64 gpa, u64 size);
int nv_vgpu_memory_unpin(struct nv_vgpu_device *vgpu, u64 gpa, u64 size);

int nv_vgpu_memory_mmap(struct nv_vgpu_device *vgpu, struct vm_area_struct *vma);

static inline struct nv_vgpu_mm_ctx *vgpu_to_mm_ctx(struct nv_vgpu_device *vgpu)
{
	return vgpu->mm_ctx;
}

#endif