1 /* SPDX-License-Identifier: GPL-2.0 */
2 /*
3 * Copyright (C) 2024, Advanced Micro Devices, Inc.
4 */
5
6 #ifndef _AMDXDNA_GEM_H_
7 #define _AMDXDNA_GEM_H_
8
9 struct amdxdna_mem {
10 u64 userptr;
11 void *kva;
12 u64 dev_addr;
13 size_t size;
14 struct page **pages;
15 u32 nr_pages;
16 struct mmu_interval_notifier notifier;
17 unsigned long *pfns;
18 bool map_invalid;
19 };
20
21 struct amdxdna_gem_obj {
22 struct drm_gem_shmem_object base;
23 struct amdxdna_client *client;
24 u8 type;
25 bool pinned;
26 struct mutex lock; /* Protects: pinned */
27 struct amdxdna_mem mem;
28
29 /* Below members is uninitialized when needed */
30 struct drm_mm mm; /* For AMDXDNA_BO_DEV_HEAP */
31 struct amdxdna_gem_obj *dev_heap; /* For AMDXDNA_BO_DEV */
32 struct drm_mm_node mm_node; /* For AMDXDNA_BO_DEV */
33 u32 assigned_hwctx;
34 };
35
36 #define to_gobj(obj) (&(obj)->base.base)
37
to_xdna_obj(struct drm_gem_object * gobj)38 static inline struct amdxdna_gem_obj *to_xdna_obj(struct drm_gem_object *gobj)
39 {
40 return container_of(gobj, struct amdxdna_gem_obj, base.base);
41 }
42
43 struct amdxdna_gem_obj *amdxdna_gem_get_obj(struct amdxdna_client *client,
44 u32 bo_hdl, u8 bo_type);
amdxdna_gem_put_obj(struct amdxdna_gem_obj * abo)45 static inline void amdxdna_gem_put_obj(struct amdxdna_gem_obj *abo)
46 {
47 drm_gem_object_put(to_gobj(abo));
48 }
49
50 struct drm_gem_object *
51 amdxdna_gem_create_object_cb(struct drm_device *dev, size_t size);
52 struct amdxdna_gem_obj *
53 amdxdna_drm_alloc_dev_bo(struct drm_device *dev,
54 struct amdxdna_drm_create_bo *args,
55 struct drm_file *filp, bool use_vmap);
56
57 int amdxdna_gem_pin_nolock(struct amdxdna_gem_obj *abo);
58 int amdxdna_gem_pin(struct amdxdna_gem_obj *abo);
59 void amdxdna_gem_unpin(struct amdxdna_gem_obj *abo);
60
61 int amdxdna_drm_create_bo_ioctl(struct drm_device *dev, void *data, struct drm_file *filp);
62 int amdxdna_drm_get_bo_info_ioctl(struct drm_device *dev, void *data, struct drm_file *filp);
63 int amdxdna_drm_sync_bo_ioctl(struct drm_device *dev, void *data, struct drm_file *filp);
64
65 #endif /* _AMDXDNA_GEM_H_ */
66