1 /*
2 * Copyright 2014 Advanced Micro Devices, Inc.
3 *
4 * Permission is hereby granted, free of charge, to any person obtaining a
5 * copy of this software and associated documentation files (the "Software"),
6 * to deal in the Software without restriction, including without limitation
7 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
8 * and/or sell copies of the Software, and to permit persons to whom the
9 * Software is furnished to do so, subject to the following conditions:
10 *
11 * The above copyright notice and this permission notice shall be included in
12 * all copies or substantial portions of the Software.
13 *
14 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
15 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
16 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
17 * THE COPYRIGHT HOLDER(S) OR AUTHOR(S) BE LIABLE FOR ANY CLAIM, DAMAGES OR
18 * OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE,
19 * ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR
20 * OTHER DEALINGS IN THE SOFTWARE.
21 *
22 */
23
24 #include <stdlib.h>
25 #include <string.h>
26 #include <errno.h>
27 #include "amdgpu.h"
28 #include "amdgpu_drm.h"
29 #include "amdgpu_internal.h"
30 #include "util_math.h"
31
amdgpu_va_range_query(amdgpu_device_handle dev,enum amdgpu_gpu_va_range type,uint64_t * start,uint64_t * end)32 drm_public int amdgpu_va_range_query(amdgpu_device_handle dev,
33 enum amdgpu_gpu_va_range type,
34 uint64_t *start, uint64_t *end)
35 {
36 if (type != amdgpu_gpu_va_range_general)
37 return -EINVAL;
38
39 *start = dev->dev_info.virtual_address_offset;
40 *end = dev->dev_info.virtual_address_max;
41 return 0;
42 }
43
amdgpu_vamgr_init(struct amdgpu_bo_va_mgr * mgr,uint64_t start,uint64_t max,uint64_t alignment)44 drm_private void amdgpu_vamgr_init(struct amdgpu_bo_va_mgr *mgr, uint64_t start,
45 uint64_t max, uint64_t alignment)
46 {
47 struct amdgpu_bo_va_hole *n;
48
49 mgr->va_max = max;
50 mgr->va_alignment = alignment;
51
52 list_inithead(&mgr->va_holes);
53 pthread_mutex_init(&mgr->bo_va_mutex, NULL);
54 pthread_mutex_lock(&mgr->bo_va_mutex);
55 n = calloc(1, sizeof(struct amdgpu_bo_va_hole));
56 n->size = mgr->va_max - start;
57 n->offset = start;
58 list_add(&n->list, &mgr->va_holes);
59 pthread_mutex_unlock(&mgr->bo_va_mutex);
60 }
61
amdgpu_vamgr_deinit(struct amdgpu_bo_va_mgr * mgr)62 drm_private void amdgpu_vamgr_deinit(struct amdgpu_bo_va_mgr *mgr)
63 {
64 struct amdgpu_bo_va_hole *hole, *tmp;
65 LIST_FOR_EACH_ENTRY_SAFE(hole, tmp, &mgr->va_holes, list) {
66 list_del(&hole->list);
67 free(hole);
68 }
69 pthread_mutex_destroy(&mgr->bo_va_mutex);
70 }
71
72 static drm_private uint64_t
amdgpu_vamgr_find_va(struct amdgpu_bo_va_mgr * mgr,uint64_t size,uint64_t alignment,uint64_t base_required)73 amdgpu_vamgr_find_va(struct amdgpu_bo_va_mgr *mgr, uint64_t size,
74 uint64_t alignment, uint64_t base_required)
75 {
76 struct amdgpu_bo_va_hole *hole, *n;
77 uint64_t offset = 0, waste = 0;
78
79
80 alignment = MAX2(alignment, mgr->va_alignment);
81 size = ALIGN(size, mgr->va_alignment);
82
83 if (base_required % alignment)
84 return AMDGPU_INVALID_VA_ADDRESS;
85
86 pthread_mutex_lock(&mgr->bo_va_mutex);
87 LIST_FOR_EACH_ENTRY_SAFE_REV(hole, n, &mgr->va_holes, list) {
88 if (base_required) {
89 if (hole->offset > base_required ||
90 (hole->offset + hole->size) < (base_required + size))
91 continue;
92 waste = base_required - hole->offset;
93 offset = base_required;
94 } else {
95 offset = hole->offset;
96 waste = offset % alignment;
97 waste = waste ? alignment - waste : 0;
98 offset += waste;
99 if (offset >= (hole->offset + hole->size)) {
100 continue;
101 }
102 }
103 if (!waste && hole->size == size) {
104 offset = hole->offset;
105 list_del(&hole->list);
106 free(hole);
107 pthread_mutex_unlock(&mgr->bo_va_mutex);
108 return offset;
109 }
110 if ((hole->size - waste) > size) {
111 if (waste) {
112 n = calloc(1, sizeof(struct amdgpu_bo_va_hole));
113 n->size = waste;
114 n->offset = hole->offset;
115 list_add(&n->list, &hole->list);
116 }
117 hole->size -= (size + waste);
118 hole->offset += size + waste;
119 pthread_mutex_unlock(&mgr->bo_va_mutex);
120 return offset;
121 }
122 if ((hole->size - waste) == size) {
123 hole->size = waste;
124 pthread_mutex_unlock(&mgr->bo_va_mutex);
125 return offset;
126 }
127 }
128
129 pthread_mutex_unlock(&mgr->bo_va_mutex);
130 return AMDGPU_INVALID_VA_ADDRESS;
131 }
132
133 static drm_private void
amdgpu_vamgr_free_va(struct amdgpu_bo_va_mgr * mgr,uint64_t va,uint64_t size)134 amdgpu_vamgr_free_va(struct amdgpu_bo_va_mgr *mgr, uint64_t va, uint64_t size)
135 {
136 struct amdgpu_bo_va_hole *hole, *next;
137
138 if (va == AMDGPU_INVALID_VA_ADDRESS)
139 return;
140
141 size = ALIGN(size, mgr->va_alignment);
142
143 pthread_mutex_lock(&mgr->bo_va_mutex);
144 hole = container_of(&mgr->va_holes, hole, list);
145 LIST_FOR_EACH_ENTRY(next, &mgr->va_holes, list) {
146 if (next->offset < va)
147 break;
148 hole = next;
149 }
150
151 if (&hole->list != &mgr->va_holes) {
152 /* Grow upper hole if it's adjacent */
153 if (hole->offset == (va + size)) {
154 hole->offset = va;
155 hole->size += size;
156 /* Merge lower hole if it's adjacent */
157 if (next != hole &&
158 &next->list != &mgr->va_holes &&
159 (next->offset + next->size) == va) {
160 next->size += hole->size;
161 list_del(&hole->list);
162 free(hole);
163 }
164 goto out;
165 }
166 }
167
168 /* Grow lower hole if it's adjacent */
169 if (next != hole && &next->list != &mgr->va_holes &&
170 (next->offset + next->size) == va) {
171 next->size += size;
172 goto out;
173 }
174
175 /* FIXME on allocation failure we just lose virtual address space
176 * maybe print a warning
177 */
178 next = calloc(1, sizeof(struct amdgpu_bo_va_hole));
179 if (next) {
180 next->size = size;
181 next->offset = va;
182 list_add(&next->list, &hole->list);
183 }
184
185 out:
186 pthread_mutex_unlock(&mgr->bo_va_mutex);
187 }
188
amdgpu_va_range_alloc(amdgpu_device_handle dev,enum amdgpu_gpu_va_range va_range_type,uint64_t size,uint64_t va_base_alignment,uint64_t va_base_required,uint64_t * va_base_allocated,amdgpu_va_handle * va_range_handle,uint64_t flags)189 drm_public int amdgpu_va_range_alloc(amdgpu_device_handle dev,
190 enum amdgpu_gpu_va_range va_range_type,
191 uint64_t size,
192 uint64_t va_base_alignment,
193 uint64_t va_base_required,
194 uint64_t *va_base_allocated,
195 amdgpu_va_handle *va_range_handle,
196 uint64_t flags)
197 {
198 struct amdgpu_bo_va_mgr *vamgr;
199
200 /* Clear the flag when the high VA manager is not initialized */
201 if (flags & AMDGPU_VA_RANGE_HIGH && !dev->vamgr_high_32.va_max)
202 flags &= ~AMDGPU_VA_RANGE_HIGH;
203
204 if (flags & AMDGPU_VA_RANGE_HIGH) {
205 if (flags & AMDGPU_VA_RANGE_32_BIT)
206 vamgr = &dev->vamgr_high_32;
207 else
208 vamgr = &dev->vamgr_high;
209 } else {
210 if (flags & AMDGPU_VA_RANGE_32_BIT)
211 vamgr = &dev->vamgr_32;
212 else
213 vamgr = &dev->vamgr;
214 }
215
216 va_base_alignment = MAX2(va_base_alignment, vamgr->va_alignment);
217 size = ALIGN(size, vamgr->va_alignment);
218
219 *va_base_allocated = amdgpu_vamgr_find_va(vamgr, size,
220 va_base_alignment, va_base_required);
221
222 if (!(flags & AMDGPU_VA_RANGE_32_BIT) &&
223 (*va_base_allocated == AMDGPU_INVALID_VA_ADDRESS)) {
224 /* fallback to 32bit address */
225 if (flags & AMDGPU_VA_RANGE_HIGH)
226 vamgr = &dev->vamgr_high_32;
227 else
228 vamgr = &dev->vamgr_32;
229 *va_base_allocated = amdgpu_vamgr_find_va(vamgr, size,
230 va_base_alignment, va_base_required);
231 }
232
233 if (*va_base_allocated != AMDGPU_INVALID_VA_ADDRESS) {
234 struct amdgpu_va* va;
235 va = calloc(1, sizeof(struct amdgpu_va));
236 if(!va){
237 amdgpu_vamgr_free_va(vamgr, *va_base_allocated, size);
238 return -ENOMEM;
239 }
240 va->dev = dev;
241 va->address = *va_base_allocated;
242 va->size = size;
243 va->range = va_range_type;
244 va->vamgr = vamgr;
245 *va_range_handle = va;
246 } else {
247 return -EINVAL;
248 }
249
250 return 0;
251 }
252
amdgpu_va_range_free(amdgpu_va_handle va_range_handle)253 drm_public int amdgpu_va_range_free(amdgpu_va_handle va_range_handle)
254 {
255 if(!va_range_handle || !va_range_handle->address)
256 return 0;
257
258 amdgpu_vamgr_free_va(va_range_handle->vamgr,
259 va_range_handle->address,
260 va_range_handle->size);
261 free(va_range_handle);
262 return 0;
263 }
264