142594600SBen Skeggs #ifndef __NVKM_MMU_H__
242594600SBen Skeggs #define __NVKM_MMU_H__
35ce3bf3cSBen Skeggs #include <core/subdev.h>
45ce3bf3cSBen Skeggs #include <core/mm.h>
542594600SBen Skeggs struct nvkm_device;
642594600SBen Skeggs struct nvkm_mem;
75ce3bf3cSBen Skeggs 
842594600SBen Skeggs struct nvkm_vm_pgt {
9d0659d32SBen Skeggs 	struct nvkm_memory *mem[2];
105ce3bf3cSBen Skeggs 	u32 refcount[2];
115ce3bf3cSBen Skeggs };
125ce3bf3cSBen Skeggs 
1342594600SBen Skeggs struct nvkm_vm_pgd {
145ce3bf3cSBen Skeggs 	struct list_head head;
1542594600SBen Skeggs 	struct nvkm_gpuobj *obj;
165ce3bf3cSBen Skeggs };
175ce3bf3cSBen Skeggs 
1842594600SBen Skeggs struct nvkm_vma {
195ce3bf3cSBen Skeggs 	struct list_head head;
205ce3bf3cSBen Skeggs 	int refcount;
2142594600SBen Skeggs 	struct nvkm_vm *vm;
2242594600SBen Skeggs 	struct nvkm_mm_node *node;
235ce3bf3cSBen Skeggs 	u64 offset;
245ce3bf3cSBen Skeggs 	u32 access;
255ce3bf3cSBen Skeggs };
265ce3bf3cSBen Skeggs 
2742594600SBen Skeggs struct nvkm_vm {
2842594600SBen Skeggs 	struct nvkm_mmu *mmu;
291de68568SBen Skeggs 
301de68568SBen Skeggs 	struct mutex mutex;
3142594600SBen Skeggs 	struct nvkm_mm mm;
325ce3bf3cSBen Skeggs 	struct kref refcount;
335ce3bf3cSBen Skeggs 
345ce3bf3cSBen Skeggs 	struct list_head pgd_list;
3568f3f702SBen Skeggs 	atomic_t engref[NVKM_SUBDEV_NR];
365ce3bf3cSBen Skeggs 
3742594600SBen Skeggs 	struct nvkm_vm_pgt *pgt;
385ce3bf3cSBen Skeggs 	u32 fpde;
395ce3bf3cSBen Skeggs 	u32 lpde;
405ce3bf3cSBen Skeggs };
415ce3bf3cSBen Skeggs 
4242594600SBen Skeggs int  nvkm_vm_new(struct nvkm_device *, u64 offset, u64 length, u64 mm_offset,
431de68568SBen Skeggs 		 struct lock_class_key *, struct nvkm_vm **);
4442594600SBen Skeggs int  nvkm_vm_ref(struct nvkm_vm *, struct nvkm_vm **, struct nvkm_gpuobj *pgd);
45d8e83994SBen Skeggs int  nvkm_vm_boot(struct nvkm_vm *, u64 size);
4642594600SBen Skeggs int  nvkm_vm_get(struct nvkm_vm *, u64 size, u32 page_shift, u32 access,
4742594600SBen Skeggs 		 struct nvkm_vma *);
4842594600SBen Skeggs void nvkm_vm_put(struct nvkm_vma *);
4942594600SBen Skeggs void nvkm_vm_map(struct nvkm_vma *, struct nvkm_mem *);
5042594600SBen Skeggs void nvkm_vm_map_at(struct nvkm_vma *, u64 offset, struct nvkm_mem *);
5142594600SBen Skeggs void nvkm_vm_unmap(struct nvkm_vma *);
5242594600SBen Skeggs void nvkm_vm_unmap_at(struct nvkm_vma *, u64 offset, u64 length);
53c9582455SBen Skeggs 
54c9582455SBen Skeggs struct nvkm_mmu {
55c9582455SBen Skeggs 	const struct nvkm_mmu_func *func;
56c9582455SBen Skeggs 	struct nvkm_subdev subdev;
57c9582455SBen Skeggs 
58c9582455SBen Skeggs 	u64 limit;
59c9582455SBen Skeggs 	u8  dma_bits;
60c9582455SBen Skeggs 	u8  lpg_shift;
61c9582455SBen Skeggs };
62c9582455SBen Skeggs 
63c9582455SBen Skeggs int nv04_mmu_new(struct nvkm_device *, int, struct nvkm_mmu **);
64c9582455SBen Skeggs int nv41_mmu_new(struct nvkm_device *, int, struct nvkm_mmu **);
65c9582455SBen Skeggs int nv44_mmu_new(struct nvkm_device *, int, struct nvkm_mmu **);
66c9582455SBen Skeggs int nv50_mmu_new(struct nvkm_device *, int, struct nvkm_mmu **);
67c9582455SBen Skeggs int gf100_mmu_new(struct nvkm_device *, int, struct nvkm_mmu **);
685ce3bf3cSBen Skeggs #endif
69