radv: fix descriptor pool allocation size
[mesa.git] / src / amd / vulkan / radv_descriptor_set.c
1 /*
2 * Copyright © 2016 Red Hat.
3 * Copyright © 2016 Bas Nieuwenhuizen
4 *
5 * Permission is hereby granted, free of charge, to any person obtaining a
6 * copy of this software and associated documentation files (the "Software"),
7 * to deal in the Software without restriction, including without limitation
8 * the rights to use, copy, modify, merge, publish, distribute, sublicense,
9 * and/or sell copies of the Software, and to permit persons to whom the
10 * Software is furnished to do so, subject to the following conditions:
11 *
12 * The above copyright notice and this permission notice (including the next
13 * paragraph) shall be included in all copies or substantial portions of the
14 * Software.
15 *
16 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
19 * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
21 * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
22 * IN THE SOFTWARE.
23 */
24 #include <assert.h>
25 #include <stdbool.h>
26 #include <string.h>
27 #include <unistd.h>
28 #include <fcntl.h>
29
30 #include "util/mesa-sha1.h"
31 #include "radv_private.h"
32 #include "sid.h"
33 #include "vk_util.h"
34
35
36 static bool has_equal_immutable_samplers(const VkSampler *samplers, uint32_t count)
37 {
38 if (!samplers)
39 return false;
40 for(uint32_t i = 1; i < count; ++i) {
41 if (memcmp(radv_sampler_from_handle(samplers[0])->state,
42 radv_sampler_from_handle(samplers[i])->state, 16)) {
43 return false;
44 }
45 }
46 return true;
47 }
48
49 static int binding_compare(const void* av, const void *bv)
50 {
51 const VkDescriptorSetLayoutBinding *a = (const VkDescriptorSetLayoutBinding*)av;
52 const VkDescriptorSetLayoutBinding *b = (const VkDescriptorSetLayoutBinding*)bv;
53
54 return (a->binding < b->binding) ? -1 : (a->binding > b->binding) ? 1 : 0;
55 }
56
57 static VkDescriptorSetLayoutBinding *
58 create_sorted_bindings(const VkDescriptorSetLayoutBinding *bindings, unsigned count) {
59 VkDescriptorSetLayoutBinding *sorted_bindings = malloc(count * sizeof(VkDescriptorSetLayoutBinding));
60 if (!sorted_bindings)
61 return NULL;
62
63 memcpy(sorted_bindings, bindings, count * sizeof(VkDescriptorSetLayoutBinding));
64
65 qsort(sorted_bindings, count, sizeof(VkDescriptorSetLayoutBinding), binding_compare);
66
67 return sorted_bindings;
68 }
69
70 VkResult radv_CreateDescriptorSetLayout(
71 VkDevice _device,
72 const VkDescriptorSetLayoutCreateInfo* pCreateInfo,
73 const VkAllocationCallbacks* pAllocator,
74 VkDescriptorSetLayout* pSetLayout)
75 {
76 RADV_FROM_HANDLE(radv_device, device, _device);
77 struct radv_descriptor_set_layout *set_layout;
78
79 assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_DESCRIPTOR_SET_LAYOUT_CREATE_INFO);
80 const VkDescriptorSetLayoutBindingFlagsCreateInfoEXT *variable_flags =
81 vk_find_struct_const(pCreateInfo->pNext, DESCRIPTOR_SET_LAYOUT_BINDING_FLAGS_CREATE_INFO_EXT);
82
83 uint32_t max_binding = 0;
84 uint32_t immutable_sampler_count = 0;
85 for (uint32_t j = 0; j < pCreateInfo->bindingCount; j++) {
86 max_binding = MAX2(max_binding, pCreateInfo->pBindings[j].binding);
87 if (pCreateInfo->pBindings[j].pImmutableSamplers)
88 immutable_sampler_count += pCreateInfo->pBindings[j].descriptorCount;
89 }
90
91 uint32_t samplers_offset = sizeof(struct radv_descriptor_set_layout) +
92 (max_binding + 1) * sizeof(set_layout->binding[0]);
93 size_t size = samplers_offset + immutable_sampler_count * 4 * sizeof(uint32_t);
94
95 set_layout = vk_alloc2(&device->alloc, pAllocator, size, 8,
96 VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
97 if (!set_layout)
98 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
99
100 set_layout->flags = pCreateInfo->flags;
101
102 /* We just allocate all the samplers at the end of the struct */
103 uint32_t *samplers = (uint32_t*)&set_layout->binding[max_binding + 1];
104
105 VkDescriptorSetLayoutBinding *bindings = create_sorted_bindings(pCreateInfo->pBindings,
106 pCreateInfo->bindingCount);
107 if (!bindings) {
108 vk_free2(&device->alloc, pAllocator, set_layout);
109 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
110 }
111
112 set_layout->binding_count = max_binding + 1;
113 set_layout->shader_stages = 0;
114 set_layout->dynamic_shader_stages = 0;
115 set_layout->has_immutable_samplers = false;
116 set_layout->size = 0;
117
118 memset(set_layout->binding, 0, size - sizeof(struct radv_descriptor_set_layout));
119
120 uint32_t buffer_count = 0;
121 uint32_t dynamic_offset_count = 0;
122
123 for (uint32_t j = 0; j < pCreateInfo->bindingCount; j++) {
124 const VkDescriptorSetLayoutBinding *binding = bindings + j;
125 uint32_t b = binding->binding;
126 uint32_t alignment;
127 unsigned binding_buffer_count = 0;
128
129 switch (binding->descriptorType) {
130 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
131 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
132 assert(!(pCreateInfo->flags & VK_DESCRIPTOR_SET_LAYOUT_CREATE_PUSH_DESCRIPTOR_BIT_KHR));
133 set_layout->binding[b].dynamic_offset_count = 1;
134 set_layout->dynamic_shader_stages |= binding->stageFlags;
135 set_layout->binding[b].size = 0;
136 binding_buffer_count = 1;
137 alignment = 1;
138 break;
139 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
140 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
141 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
142 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
143 set_layout->binding[b].size = 16;
144 binding_buffer_count = 1;
145 alignment = 16;
146 break;
147 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
148 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
149 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
150 /* main descriptor + fmask descriptor */
151 set_layout->binding[b].size = 64;
152 binding_buffer_count = 1;
153 alignment = 32;
154 break;
155 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
156 /* main descriptor + fmask descriptor + sampler */
157 set_layout->binding[b].size = 96;
158 binding_buffer_count = 1;
159 alignment = 32;
160 break;
161 case VK_DESCRIPTOR_TYPE_SAMPLER:
162 set_layout->binding[b].size = 16;
163 alignment = 16;
164 break;
165 default:
166 unreachable("unknown descriptor type\n");
167 break;
168 }
169
170 set_layout->size = align(set_layout->size, alignment);
171 set_layout->binding[b].type = binding->descriptorType;
172 set_layout->binding[b].array_size = binding->descriptorCount;
173 set_layout->binding[b].offset = set_layout->size;
174 set_layout->binding[b].buffer_offset = buffer_count;
175 set_layout->binding[b].dynamic_offset_offset = dynamic_offset_count;
176
177 if (variable_flags && binding->binding < variable_flags->bindingCount &&
178 (variable_flags->pBindingFlags[binding->binding] & VK_DESCRIPTOR_BINDING_VARIABLE_DESCRIPTOR_COUNT_BIT_EXT)) {
179 assert(!binding->pImmutableSamplers); /* Terribly ill defined how many samplers are valid */
180 assert(binding->binding == max_binding);
181
182 set_layout->has_variable_descriptors = true;
183 }
184
185 if (binding->pImmutableSamplers) {
186 set_layout->binding[b].immutable_samplers_offset = samplers_offset;
187 set_layout->binding[b].immutable_samplers_equal =
188 has_equal_immutable_samplers(binding->pImmutableSamplers, binding->descriptorCount);
189 set_layout->has_immutable_samplers = true;
190
191
192 for (uint32_t i = 0; i < binding->descriptorCount; i++)
193 memcpy(samplers + 4 * i, &radv_sampler_from_handle(binding->pImmutableSamplers[i])->state, 16);
194
195 /* Don't reserve space for the samplers if they're not accessed. */
196 if (set_layout->binding[b].immutable_samplers_equal) {
197 if (binding->descriptorType == VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER)
198 set_layout->binding[b].size -= 32;
199 else if (binding->descriptorType == VK_DESCRIPTOR_TYPE_SAMPLER)
200 set_layout->binding[b].size -= 16;
201 }
202 samplers += 4 * binding->descriptorCount;
203 samplers_offset += 4 * sizeof(uint32_t) * binding->descriptorCount;
204 }
205
206 set_layout->size += binding->descriptorCount * set_layout->binding[b].size;
207 buffer_count += binding->descriptorCount * binding_buffer_count;
208 dynamic_offset_count += binding->descriptorCount *
209 set_layout->binding[b].dynamic_offset_count;
210 set_layout->shader_stages |= binding->stageFlags;
211 }
212
213 free(bindings);
214
215 set_layout->buffer_count = buffer_count;
216 set_layout->dynamic_offset_count = dynamic_offset_count;
217
218 *pSetLayout = radv_descriptor_set_layout_to_handle(set_layout);
219
220 return VK_SUCCESS;
221 }
222
223 void radv_DestroyDescriptorSetLayout(
224 VkDevice _device,
225 VkDescriptorSetLayout _set_layout,
226 const VkAllocationCallbacks* pAllocator)
227 {
228 RADV_FROM_HANDLE(radv_device, device, _device);
229 RADV_FROM_HANDLE(radv_descriptor_set_layout, set_layout, _set_layout);
230
231 if (!set_layout)
232 return;
233
234 vk_free2(&device->alloc, pAllocator, set_layout);
235 }
236
237 void radv_GetDescriptorSetLayoutSupport(VkDevice device,
238 const VkDescriptorSetLayoutCreateInfo* pCreateInfo,
239 VkDescriptorSetLayoutSupport* pSupport)
240 {
241 VkDescriptorSetLayoutBinding *bindings = create_sorted_bindings(pCreateInfo->pBindings,
242 pCreateInfo->bindingCount);
243 if (!bindings) {
244 pSupport->supported = false;
245 return;
246 }
247
248 const VkDescriptorSetLayoutBindingFlagsCreateInfoEXT *variable_flags =
249 vk_find_struct_const(pCreateInfo->pNext, DESCRIPTOR_SET_LAYOUT_BINDING_FLAGS_CREATE_INFO_EXT);
250 VkDescriptorSetVariableDescriptorCountLayoutSupportEXT *variable_count =
251 vk_find_struct((void*)pCreateInfo->pNext, DESCRIPTOR_SET_VARIABLE_DESCRIPTOR_COUNT_LAYOUT_SUPPORT_EXT);
252 if (variable_count) {
253 variable_count->maxVariableDescriptorCount = 0;
254 }
255
256 bool supported = true;
257 uint64_t size = 0;
258 for (uint32_t i = 0; i < pCreateInfo->bindingCount; i++) {
259 const VkDescriptorSetLayoutBinding *binding = bindings + i;
260
261 uint64_t descriptor_size = 0;
262 uint64_t descriptor_alignment = 1;
263 switch (binding->descriptorType) {
264 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
265 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
266 break;
267 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
268 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
269 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
270 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
271 descriptor_size = 16;
272 descriptor_alignment = 16;
273 break;
274 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
275 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
276 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
277 descriptor_size = 64;
278 descriptor_alignment = 32;
279 break;
280 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
281 if (!has_equal_immutable_samplers(binding->pImmutableSamplers, binding->descriptorCount)) {
282 descriptor_size = 64;
283 } else {
284 descriptor_size = 96;
285 }
286 descriptor_alignment = 32;
287 break;
288 case VK_DESCRIPTOR_TYPE_SAMPLER:
289 if (!has_equal_immutable_samplers(binding->pImmutableSamplers, binding->descriptorCount)) {
290 descriptor_size = 16;
291 descriptor_alignment = 16;
292 }
293 break;
294 default:
295 unreachable("unknown descriptor type\n");
296 break;
297 }
298
299 if (size && !align_u64(size, descriptor_alignment)) {
300 supported = false;
301 }
302 size = align_u64(size, descriptor_alignment);
303
304 uint64_t max_count = UINT64_MAX;
305 if (descriptor_size)
306 max_count = (UINT64_MAX - size) / descriptor_size;
307
308 if (max_count < binding->descriptorCount) {
309 supported = false;
310 }
311 if (variable_flags && binding->binding <variable_flags->bindingCount && variable_count &&
312 (variable_flags->pBindingFlags[binding->binding] & VK_DESCRIPTOR_BINDING_VARIABLE_DESCRIPTOR_COUNT_BIT_EXT)) {
313 variable_count->maxVariableDescriptorCount = MIN2(UINT32_MAX, max_count);
314 }
315 size += binding->descriptorCount * descriptor_size;
316 }
317
318 free(bindings);
319
320 pSupport->supported = supported;
321 }
322
323 /*
324 * Pipeline layouts. These have nothing to do with the pipeline. They are
325 * just multiple descriptor set layouts pasted together.
326 */
327
328 VkResult radv_CreatePipelineLayout(
329 VkDevice _device,
330 const VkPipelineLayoutCreateInfo* pCreateInfo,
331 const VkAllocationCallbacks* pAllocator,
332 VkPipelineLayout* pPipelineLayout)
333 {
334 RADV_FROM_HANDLE(radv_device, device, _device);
335 struct radv_pipeline_layout *layout;
336 struct mesa_sha1 ctx;
337
338 assert(pCreateInfo->sType == VK_STRUCTURE_TYPE_PIPELINE_LAYOUT_CREATE_INFO);
339
340 layout = vk_alloc2(&device->alloc, pAllocator, sizeof(*layout), 8,
341 VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
342 if (layout == NULL)
343 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
344
345 layout->num_sets = pCreateInfo->setLayoutCount;
346
347 unsigned dynamic_offset_count = 0;
348
349
350 _mesa_sha1_init(&ctx);
351 for (uint32_t set = 0; set < pCreateInfo->setLayoutCount; set++) {
352 RADV_FROM_HANDLE(radv_descriptor_set_layout, set_layout,
353 pCreateInfo->pSetLayouts[set]);
354 layout->set[set].layout = set_layout;
355
356 layout->set[set].dynamic_offset_start = dynamic_offset_count;
357 for (uint32_t b = 0; b < set_layout->binding_count; b++) {
358 dynamic_offset_count += set_layout->binding[b].array_size * set_layout->binding[b].dynamic_offset_count;
359 if (set_layout->binding[b].immutable_samplers_offset)
360 _mesa_sha1_update(&ctx, radv_immutable_samplers(set_layout, set_layout->binding + b),
361 set_layout->binding[b].array_size * 4 * sizeof(uint32_t));
362 }
363 _mesa_sha1_update(&ctx, set_layout->binding,
364 sizeof(set_layout->binding[0]) * set_layout->binding_count);
365 }
366
367 layout->dynamic_offset_count = dynamic_offset_count;
368 layout->push_constant_size = 0;
369
370 for (unsigned i = 0; i < pCreateInfo->pushConstantRangeCount; ++i) {
371 const VkPushConstantRange *range = pCreateInfo->pPushConstantRanges + i;
372 layout->push_constant_size = MAX2(layout->push_constant_size,
373 range->offset + range->size);
374 }
375
376 layout->push_constant_size = align(layout->push_constant_size, 16);
377 _mesa_sha1_update(&ctx, &layout->push_constant_size,
378 sizeof(layout->push_constant_size));
379 _mesa_sha1_final(&ctx, layout->sha1);
380 *pPipelineLayout = radv_pipeline_layout_to_handle(layout);
381
382 return VK_SUCCESS;
383 }
384
385 void radv_DestroyPipelineLayout(
386 VkDevice _device,
387 VkPipelineLayout _pipelineLayout,
388 const VkAllocationCallbacks* pAllocator)
389 {
390 RADV_FROM_HANDLE(radv_device, device, _device);
391 RADV_FROM_HANDLE(radv_pipeline_layout, pipeline_layout, _pipelineLayout);
392
393 if (!pipeline_layout)
394 return;
395 vk_free2(&device->alloc, pAllocator, pipeline_layout);
396 }
397
398 #define EMPTY 1
399
400 static VkResult
401 radv_descriptor_set_create(struct radv_device *device,
402 struct radv_descriptor_pool *pool,
403 const struct radv_descriptor_set_layout *layout,
404 const uint32_t *variable_count,
405 struct radv_descriptor_set **out_set)
406 {
407 struct radv_descriptor_set *set;
408 unsigned range_offset = sizeof(struct radv_descriptor_set) +
409 sizeof(struct radeon_winsys_bo *) * layout->buffer_count;
410 unsigned mem_size = range_offset +
411 sizeof(struct radv_descriptor_range) * layout->dynamic_offset_count;
412
413 if (pool->host_memory_base) {
414 if (pool->host_memory_end - pool->host_memory_ptr < mem_size)
415 return vk_error(device->instance, VK_ERROR_OUT_OF_POOL_MEMORY_KHR);
416
417 set = (struct radv_descriptor_set*)pool->host_memory_ptr;
418 pool->host_memory_ptr += mem_size;
419 } else {
420 set = vk_alloc2(&device->alloc, NULL, mem_size, 8,
421 VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
422
423 if (!set)
424 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
425 }
426
427 memset(set, 0, mem_size);
428
429 if (layout->dynamic_offset_count) {
430 set->dynamic_descriptors = (struct radv_descriptor_range*)((uint8_t*)set + range_offset);
431 }
432
433 set->layout = layout;
434 uint32_t layout_size = align_u32(layout->size, 32);
435 if (layout_size) {
436 set->size = layout_size;
437
438 if (!pool->host_memory_base && pool->entry_count == pool->max_entry_count) {
439 vk_free2(&device->alloc, NULL, set);
440 return vk_error(device->instance, VK_ERROR_OUT_OF_POOL_MEMORY_KHR);
441 }
442
443 /* try to allocate linearly first, so that we don't spend
444 * time looking for gaps if the app only allocates &
445 * resets via the pool. */
446 if (pool->current_offset + layout_size <= pool->size) {
447 set->bo = pool->bo;
448 set->mapped_ptr = (uint32_t*)(pool->mapped_ptr + pool->current_offset);
449 set->va = radv_buffer_get_va(set->bo) + pool->current_offset;
450 if (!pool->host_memory_base) {
451 pool->entries[pool->entry_count].offset = pool->current_offset;
452 pool->entries[pool->entry_count].size = layout_size;
453 pool->entries[pool->entry_count].set = set;
454 pool->entry_count++;
455 }
456 pool->current_offset += layout_size;
457 } else if (!pool->host_memory_base) {
458 uint64_t offset = 0;
459 int index;
460
461 for (index = 0; index < pool->entry_count; ++index) {
462 if (pool->entries[index].offset - offset >= layout_size)
463 break;
464 offset = pool->entries[index].offset + pool->entries[index].size;
465 }
466
467 if (pool->size - offset < layout_size) {
468 vk_free2(&device->alloc, NULL, set);
469 return vk_error(device->instance, VK_ERROR_OUT_OF_POOL_MEMORY_KHR);
470 }
471 set->bo = pool->bo;
472 set->mapped_ptr = (uint32_t*)(pool->mapped_ptr + offset);
473 set->va = radv_buffer_get_va(set->bo) + offset;
474 memmove(&pool->entries[index + 1], &pool->entries[index],
475 sizeof(pool->entries[0]) * (pool->entry_count - index));
476 pool->entries[index].offset = offset;
477 pool->entries[index].size = layout_size;
478 pool->entries[index].set = set;
479 pool->entry_count++;
480 } else
481 return vk_error(device->instance, VK_ERROR_OUT_OF_POOL_MEMORY_KHR);
482 }
483
484 if (layout->has_immutable_samplers) {
485 for (unsigned i = 0; i < layout->binding_count; ++i) {
486 if (!layout->binding[i].immutable_samplers_offset ||
487 layout->binding[i].immutable_samplers_equal)
488 continue;
489
490 unsigned offset = layout->binding[i].offset / 4;
491 if (layout->binding[i].type == VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER)
492 offset += 16;
493
494 const uint32_t *samplers = (const uint32_t*)((const char*)layout + layout->binding[i].immutable_samplers_offset);
495 for (unsigned j = 0; j < layout->binding[i].array_size; ++j) {
496 memcpy(set->mapped_ptr + offset, samplers + 4 * j, 16);
497 offset += layout->binding[i].size / 4;
498 }
499
500 }
501 }
502 *out_set = set;
503 return VK_SUCCESS;
504 }
505
506 static void
507 radv_descriptor_set_destroy(struct radv_device *device,
508 struct radv_descriptor_pool *pool,
509 struct radv_descriptor_set *set,
510 bool free_bo)
511 {
512 assert(!pool->host_memory_base);
513
514 if (free_bo && set->size && !pool->host_memory_base) {
515 uint32_t offset = (uint8_t*)set->mapped_ptr - pool->mapped_ptr;
516 for (int i = 0; i < pool->entry_count; ++i) {
517 if (pool->entries[i].offset == offset) {
518 memmove(&pool->entries[i], &pool->entries[i+1],
519 sizeof(pool->entries[i]) * (pool->entry_count - i - 1));
520 --pool->entry_count;
521 break;
522 }
523 }
524 }
525 vk_free2(&device->alloc, NULL, set);
526 }
527
528 VkResult radv_CreateDescriptorPool(
529 VkDevice _device,
530 const VkDescriptorPoolCreateInfo* pCreateInfo,
531 const VkAllocationCallbacks* pAllocator,
532 VkDescriptorPool* pDescriptorPool)
533 {
534 RADV_FROM_HANDLE(radv_device, device, _device);
535 struct radv_descriptor_pool *pool;
536 int size = sizeof(struct radv_descriptor_pool);
537 uint64_t bo_size = 0, bo_count = 0, range_count = 0;
538
539
540 for (unsigned i = 0; i < pCreateInfo->poolSizeCount; ++i) {
541 if (pCreateInfo->pPoolSizes[i].type != VK_DESCRIPTOR_TYPE_SAMPLER)
542 bo_count += pCreateInfo->pPoolSizes[i].descriptorCount;
543
544 switch(pCreateInfo->pPoolSizes[i].type) {
545 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
546 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
547 range_count += pCreateInfo->pPoolSizes[i].descriptorCount;
548 break;
549 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
550 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
551 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
552 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
553 case VK_DESCRIPTOR_TYPE_SAMPLER:
554 /* 32 as we may need to align for images */
555 bo_size += 32 * pCreateInfo->pPoolSizes[i].descriptorCount;
556 break;
557 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
558 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
559 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
560 bo_size += 64 * pCreateInfo->pPoolSizes[i].descriptorCount;
561 break;
562 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
563 bo_size += 96 * pCreateInfo->pPoolSizes[i].descriptorCount;
564 break;
565 default:
566 unreachable("unknown descriptor type\n");
567 break;
568 }
569 }
570
571 if (!(pCreateInfo->flags & VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT)) {
572 uint64_t host_size = sizeof(struct radv_descriptor_set);
573 host_size += sizeof(struct radeon_winsys_bo*) * bo_count;
574 host_size += sizeof(struct radv_descriptor_range) * range_count;
575 host_size *= pCreateInfo->maxSets;
576 size += host_size;
577 } else {
578 size += sizeof(struct radv_descriptor_pool_entry) * pCreateInfo->maxSets;
579 }
580
581 pool = vk_alloc2(&device->alloc, pAllocator, size, 8,
582 VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
583 if (!pool)
584 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
585
586 memset(pool, 0, sizeof(*pool));
587
588 if (!(pCreateInfo->flags & VK_DESCRIPTOR_POOL_CREATE_FREE_DESCRIPTOR_SET_BIT)) {
589 pool->host_memory_base = (uint8_t*)pool + sizeof(struct radv_descriptor_pool);
590 pool->host_memory_ptr = pool->host_memory_base;
591 pool->host_memory_end = (uint8_t*)pool + size;
592 }
593
594 if (bo_size) {
595 pool->bo = device->ws->buffer_create(device->ws, bo_size, 32,
596 RADEON_DOMAIN_VRAM,
597 RADEON_FLAG_NO_INTERPROCESS_SHARING |
598 RADEON_FLAG_READ_ONLY |
599 RADEON_FLAG_32BIT);
600 pool->mapped_ptr = (uint8_t*)device->ws->buffer_map(pool->bo);
601 }
602 pool->size = bo_size;
603 pool->max_entry_count = pCreateInfo->maxSets;
604
605 *pDescriptorPool = radv_descriptor_pool_to_handle(pool);
606 return VK_SUCCESS;
607 }
608
609 void radv_DestroyDescriptorPool(
610 VkDevice _device,
611 VkDescriptorPool _pool,
612 const VkAllocationCallbacks* pAllocator)
613 {
614 RADV_FROM_HANDLE(radv_device, device, _device);
615 RADV_FROM_HANDLE(radv_descriptor_pool, pool, _pool);
616
617 if (!pool)
618 return;
619
620 if (!pool->host_memory_base) {
621 for(int i = 0; i < pool->entry_count; ++i) {
622 radv_descriptor_set_destroy(device, pool, pool->entries[i].set, false);
623 }
624 }
625
626 if (pool->bo)
627 device->ws->buffer_destroy(pool->bo);
628 vk_free2(&device->alloc, pAllocator, pool);
629 }
630
631 VkResult radv_ResetDescriptorPool(
632 VkDevice _device,
633 VkDescriptorPool descriptorPool,
634 VkDescriptorPoolResetFlags flags)
635 {
636 RADV_FROM_HANDLE(radv_device, device, _device);
637 RADV_FROM_HANDLE(radv_descriptor_pool, pool, descriptorPool);
638
639 if (!pool->host_memory_base) {
640 for(int i = 0; i < pool->entry_count; ++i) {
641 radv_descriptor_set_destroy(device, pool, pool->entries[i].set, false);
642 }
643 pool->entry_count = 0;
644 }
645
646 pool->current_offset = 0;
647 pool->host_memory_ptr = pool->host_memory_base;
648
649 return VK_SUCCESS;
650 }
651
652 VkResult radv_AllocateDescriptorSets(
653 VkDevice _device,
654 const VkDescriptorSetAllocateInfo* pAllocateInfo,
655 VkDescriptorSet* pDescriptorSets)
656 {
657 RADV_FROM_HANDLE(radv_device, device, _device);
658 RADV_FROM_HANDLE(radv_descriptor_pool, pool, pAllocateInfo->descriptorPool);
659
660 VkResult result = VK_SUCCESS;
661 uint32_t i;
662 struct radv_descriptor_set *set = NULL;
663
664 const VkDescriptorSetVariableDescriptorCountAllocateInfoEXT *variable_counts =
665 vk_find_struct_const(pAllocateInfo->pNext, DESCRIPTOR_SET_VARIABLE_DESCRIPTOR_COUNT_ALLOCATE_INFO_EXT);
666 const uint32_t zero = 0;
667
668 /* allocate a set of buffers for each shader to contain descriptors */
669 for (i = 0; i < pAllocateInfo->descriptorSetCount; i++) {
670 RADV_FROM_HANDLE(radv_descriptor_set_layout, layout,
671 pAllocateInfo->pSetLayouts[i]);
672
673 const uint32_t *variable_count = NULL;
674 if (variable_counts) {
675 if (i < variable_counts->descriptorSetCount)
676 variable_count = variable_counts->pDescriptorCounts + i;
677 else
678 variable_count = &zero;
679 }
680
681 assert(!(layout->flags & VK_DESCRIPTOR_SET_LAYOUT_CREATE_PUSH_DESCRIPTOR_BIT_KHR));
682
683 result = radv_descriptor_set_create(device, pool, layout, variable_count, &set);
684 if (result != VK_SUCCESS)
685 break;
686
687 pDescriptorSets[i] = radv_descriptor_set_to_handle(set);
688 }
689
690 if (result != VK_SUCCESS)
691 radv_FreeDescriptorSets(_device, pAllocateInfo->descriptorPool,
692 i, pDescriptorSets);
693 return result;
694 }
695
696 VkResult radv_FreeDescriptorSets(
697 VkDevice _device,
698 VkDescriptorPool descriptorPool,
699 uint32_t count,
700 const VkDescriptorSet* pDescriptorSets)
701 {
702 RADV_FROM_HANDLE(radv_device, device, _device);
703 RADV_FROM_HANDLE(radv_descriptor_pool, pool, descriptorPool);
704
705 for (uint32_t i = 0; i < count; i++) {
706 RADV_FROM_HANDLE(radv_descriptor_set, set, pDescriptorSets[i]);
707
708 if (set && !pool->host_memory_base)
709 radv_descriptor_set_destroy(device, pool, set, true);
710 }
711 return VK_SUCCESS;
712 }
713
714 static void write_texel_buffer_descriptor(struct radv_device *device,
715 struct radv_cmd_buffer *cmd_buffer,
716 unsigned *dst,
717 struct radeon_winsys_bo **buffer_list,
718 const VkBufferView _buffer_view)
719 {
720 RADV_FROM_HANDLE(radv_buffer_view, buffer_view, _buffer_view);
721
722 memcpy(dst, buffer_view->state, 4 * 4);
723
724 if (cmd_buffer)
725 radv_cs_add_buffer(device->ws, cmd_buffer->cs, buffer_view->bo);
726 else
727 *buffer_list = buffer_view->bo;
728 }
729
730 static void write_buffer_descriptor(struct radv_device *device,
731 struct radv_cmd_buffer *cmd_buffer,
732 unsigned *dst,
733 struct radeon_winsys_bo **buffer_list,
734 const VkDescriptorBufferInfo *buffer_info)
735 {
736 RADV_FROM_HANDLE(radv_buffer, buffer, buffer_info->buffer);
737 uint64_t va = radv_buffer_get_va(buffer->bo);
738 uint32_t range = buffer_info->range;
739
740 if (buffer_info->range == VK_WHOLE_SIZE)
741 range = buffer->size - buffer_info->offset;
742
743 va += buffer_info->offset + buffer->offset;
744 dst[0] = va;
745 dst[1] = S_008F04_BASE_ADDRESS_HI(va >> 32);
746 dst[2] = range;
747 dst[3] = S_008F0C_DST_SEL_X(V_008F0C_SQ_SEL_X) |
748 S_008F0C_DST_SEL_Y(V_008F0C_SQ_SEL_Y) |
749 S_008F0C_DST_SEL_Z(V_008F0C_SQ_SEL_Z) |
750 S_008F0C_DST_SEL_W(V_008F0C_SQ_SEL_W) |
751 S_008F0C_NUM_FORMAT(V_008F0C_BUF_NUM_FORMAT_FLOAT) |
752 S_008F0C_DATA_FORMAT(V_008F0C_BUF_DATA_FORMAT_32);
753
754 if (cmd_buffer)
755 radv_cs_add_buffer(device->ws, cmd_buffer->cs, buffer->bo);
756 else
757 *buffer_list = buffer->bo;
758 }
759
760 static void write_dynamic_buffer_descriptor(struct radv_device *device,
761 struct radv_descriptor_range *range,
762 struct radeon_winsys_bo **buffer_list,
763 const VkDescriptorBufferInfo *buffer_info)
764 {
765 RADV_FROM_HANDLE(radv_buffer, buffer, buffer_info->buffer);
766 uint64_t va = radv_buffer_get_va(buffer->bo);
767 unsigned size = buffer_info->range;
768
769 if (buffer_info->range == VK_WHOLE_SIZE)
770 size = buffer->size - buffer_info->offset;
771
772 va += buffer_info->offset + buffer->offset;
773 range->va = va;
774 range->size = size;
775
776 *buffer_list = buffer->bo;
777 }
778
779 static void
780 write_image_descriptor(struct radv_device *device,
781 struct radv_cmd_buffer *cmd_buffer,
782 unsigned *dst,
783 struct radeon_winsys_bo **buffer_list,
784 VkDescriptorType descriptor_type,
785 const VkDescriptorImageInfo *image_info)
786 {
787 RADV_FROM_HANDLE(radv_image_view, iview, image_info->imageView);
788 uint32_t *descriptor;
789
790 if (descriptor_type == VK_DESCRIPTOR_TYPE_STORAGE_IMAGE) {
791 descriptor = iview->storage_descriptor;
792 } else {
793 descriptor = iview->descriptor;
794 }
795
796 memcpy(dst, descriptor, 16 * 4);
797
798 if (cmd_buffer)
799 radv_cs_add_buffer(device->ws, cmd_buffer->cs, iview->bo);
800 else
801 *buffer_list = iview->bo;
802 }
803
804 static void
805 write_combined_image_sampler_descriptor(struct radv_device *device,
806 struct radv_cmd_buffer *cmd_buffer,
807 unsigned *dst,
808 struct radeon_winsys_bo **buffer_list,
809 VkDescriptorType descriptor_type,
810 const VkDescriptorImageInfo *image_info,
811 bool has_sampler)
812 {
813 RADV_FROM_HANDLE(radv_sampler, sampler, image_info->sampler);
814
815 write_image_descriptor(device, cmd_buffer, dst, buffer_list, descriptor_type, image_info);
816 /* copy over sampler state */
817 if (has_sampler)
818 memcpy(dst + 16, sampler->state, 16);
819 }
820
821 static void
822 write_sampler_descriptor(struct radv_device *device,
823 unsigned *dst,
824 const VkDescriptorImageInfo *image_info)
825 {
826 RADV_FROM_HANDLE(radv_sampler, sampler, image_info->sampler);
827
828 memcpy(dst, sampler->state, 16);
829 }
830
831 void radv_update_descriptor_sets(
832 struct radv_device* device,
833 struct radv_cmd_buffer* cmd_buffer,
834 VkDescriptorSet dstSetOverride,
835 uint32_t descriptorWriteCount,
836 const VkWriteDescriptorSet* pDescriptorWrites,
837 uint32_t descriptorCopyCount,
838 const VkCopyDescriptorSet* pDescriptorCopies)
839 {
840 uint32_t i, j;
841 for (i = 0; i < descriptorWriteCount; i++) {
842 const VkWriteDescriptorSet *writeset = &pDescriptorWrites[i];
843 RADV_FROM_HANDLE(radv_descriptor_set, set,
844 dstSetOverride ? dstSetOverride : writeset->dstSet);
845 const struct radv_descriptor_set_binding_layout *binding_layout =
846 set->layout->binding + writeset->dstBinding;
847 uint32_t *ptr = set->mapped_ptr;
848 struct radeon_winsys_bo **buffer_list = set->descriptors;
849 /* Immutable samplers are not copied into push descriptors when they are
850 * allocated, so if we are writing push descriptors we have to copy the
851 * immutable samplers into them now.
852 */
853 const bool copy_immutable_samplers = cmd_buffer &&
854 binding_layout->immutable_samplers_offset && !binding_layout->immutable_samplers_equal;
855 const uint32_t *samplers = radv_immutable_samplers(set->layout, binding_layout);
856
857 ptr += binding_layout->offset / 4;
858 ptr += binding_layout->size * writeset->dstArrayElement / 4;
859 buffer_list += binding_layout->buffer_offset;
860 buffer_list += writeset->dstArrayElement;
861 for (j = 0; j < writeset->descriptorCount; ++j) {
862 switch(writeset->descriptorType) {
863 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
864 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC: {
865 unsigned idx = writeset->dstArrayElement + j;
866 idx += binding_layout->dynamic_offset_offset;
867 assert(!(set->layout->flags & VK_DESCRIPTOR_SET_LAYOUT_CREATE_PUSH_DESCRIPTOR_BIT_KHR));
868 write_dynamic_buffer_descriptor(device, set->dynamic_descriptors + idx,
869 buffer_list, writeset->pBufferInfo + j);
870 break;
871 }
872 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
873 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
874 write_buffer_descriptor(device, cmd_buffer, ptr, buffer_list,
875 writeset->pBufferInfo + j);
876 break;
877 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
878 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
879 write_texel_buffer_descriptor(device, cmd_buffer, ptr, buffer_list,
880 writeset->pTexelBufferView[j]);
881 break;
882 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
883 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
884 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
885 write_image_descriptor(device, cmd_buffer, ptr, buffer_list,
886 writeset->descriptorType,
887 writeset->pImageInfo + j);
888 break;
889 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
890 write_combined_image_sampler_descriptor(device, cmd_buffer, ptr, buffer_list,
891 writeset->descriptorType,
892 writeset->pImageInfo + j,
893 !binding_layout->immutable_samplers_offset);
894 if (copy_immutable_samplers) {
895 const unsigned idx = writeset->dstArrayElement + j;
896 memcpy(ptr + 16, samplers + 4 * idx, 16);
897 }
898 break;
899 case VK_DESCRIPTOR_TYPE_SAMPLER:
900 if (!binding_layout->immutable_samplers_offset) {
901 write_sampler_descriptor(device, ptr,
902 writeset->pImageInfo + j);
903 } else if (copy_immutable_samplers) {
904 unsigned idx = writeset->dstArrayElement + j;
905 memcpy(ptr, samplers + 4 * idx, 16);
906 }
907 break;
908 default:
909 unreachable("unimplemented descriptor type");
910 break;
911 }
912 ptr += binding_layout->size / 4;
913 ++buffer_list;
914 }
915
916 }
917
918 for (i = 0; i < descriptorCopyCount; i++) {
919 const VkCopyDescriptorSet *copyset = &pDescriptorCopies[i];
920 RADV_FROM_HANDLE(radv_descriptor_set, src_set,
921 copyset->srcSet);
922 RADV_FROM_HANDLE(radv_descriptor_set, dst_set,
923 copyset->dstSet);
924 const struct radv_descriptor_set_binding_layout *src_binding_layout =
925 src_set->layout->binding + copyset->srcBinding;
926 const struct radv_descriptor_set_binding_layout *dst_binding_layout =
927 dst_set->layout->binding + copyset->dstBinding;
928 uint32_t *src_ptr = src_set->mapped_ptr;
929 uint32_t *dst_ptr = dst_set->mapped_ptr;
930 struct radeon_winsys_bo **src_buffer_list = src_set->descriptors;
931 struct radeon_winsys_bo **dst_buffer_list = dst_set->descriptors;
932
933 src_ptr += src_binding_layout->offset / 4;
934 dst_ptr += dst_binding_layout->offset / 4;
935
936 src_ptr += src_binding_layout->size * copyset->srcArrayElement / 4;
937 dst_ptr += dst_binding_layout->size * copyset->dstArrayElement / 4;
938
939 src_buffer_list += src_binding_layout->buffer_offset;
940 src_buffer_list += copyset->srcArrayElement;
941
942 dst_buffer_list += dst_binding_layout->buffer_offset;
943 dst_buffer_list += copyset->dstArrayElement;
944
945 for (j = 0; j < copyset->descriptorCount; ++j) {
946 switch (src_binding_layout->type) {
947 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
948 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC: {
949 unsigned src_idx = copyset->srcArrayElement + j;
950 unsigned dst_idx = copyset->dstArrayElement + j;
951 struct radv_descriptor_range *src_range, *dst_range;
952 src_idx += src_binding_layout->dynamic_offset_offset;
953 dst_idx += dst_binding_layout->dynamic_offset_offset;
954
955 src_range = src_set->dynamic_descriptors + src_idx;
956 dst_range = dst_set->dynamic_descriptors + dst_idx;
957 *dst_range = *src_range;
958 break;
959 }
960 default:
961 memcpy(dst_ptr, src_ptr, src_binding_layout->size);
962 }
963 src_ptr += src_binding_layout->size / 4;
964 dst_ptr += dst_binding_layout->size / 4;
965 dst_buffer_list[j] = src_buffer_list[j];
966 ++src_buffer_list;
967 ++dst_buffer_list;
968 }
969 }
970 }
971
972 void radv_UpdateDescriptorSets(
973 VkDevice _device,
974 uint32_t descriptorWriteCount,
975 const VkWriteDescriptorSet* pDescriptorWrites,
976 uint32_t descriptorCopyCount,
977 const VkCopyDescriptorSet* pDescriptorCopies)
978 {
979 RADV_FROM_HANDLE(radv_device, device, _device);
980
981 radv_update_descriptor_sets(device, NULL, VK_NULL_HANDLE, descriptorWriteCount, pDescriptorWrites,
982 descriptorCopyCount, pDescriptorCopies);
983 }
984
985 VkResult radv_CreateDescriptorUpdateTemplate(VkDevice _device,
986 const VkDescriptorUpdateTemplateCreateInfoKHR *pCreateInfo,
987 const VkAllocationCallbacks *pAllocator,
988 VkDescriptorUpdateTemplateKHR *pDescriptorUpdateTemplate)
989 {
990 RADV_FROM_HANDLE(radv_device, device, _device);
991 RADV_FROM_HANDLE(radv_descriptor_set_layout, set_layout, pCreateInfo->descriptorSetLayout);
992 const uint32_t entry_count = pCreateInfo->descriptorUpdateEntryCount;
993 const size_t size = sizeof(struct radv_descriptor_update_template) +
994 sizeof(struct radv_descriptor_update_template_entry) * entry_count;
995 struct radv_descriptor_update_template *templ;
996 uint32_t i;
997
998 templ = vk_alloc2(&device->alloc, pAllocator, size, 8, VK_SYSTEM_ALLOCATION_SCOPE_OBJECT);
999 if (!templ)
1000 return vk_error(device->instance, VK_ERROR_OUT_OF_HOST_MEMORY);
1001
1002 templ->entry_count = entry_count;
1003 templ->bind_point = pCreateInfo->pipelineBindPoint;
1004
1005 for (i = 0; i < entry_count; i++) {
1006 const VkDescriptorUpdateTemplateEntryKHR *entry = &pCreateInfo->pDescriptorUpdateEntries[i];
1007 const struct radv_descriptor_set_binding_layout *binding_layout =
1008 set_layout->binding + entry->dstBinding;
1009 const uint32_t buffer_offset = binding_layout->buffer_offset + entry->dstArrayElement;
1010 const uint32_t *immutable_samplers = NULL;
1011 uint32_t dst_offset;
1012 uint32_t dst_stride;
1013
1014 /* dst_offset is an offset into dynamic_descriptors when the descriptor
1015 is dynamic, and an offset into mapped_ptr otherwise */
1016 switch (entry->descriptorType) {
1017 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
1018 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC:
1019 assert(pCreateInfo->templateType == VK_DESCRIPTOR_UPDATE_TEMPLATE_TYPE_DESCRIPTOR_SET_KHR);
1020 dst_offset = binding_layout->dynamic_offset_offset + entry->dstArrayElement;
1021 dst_stride = 0; /* Not used */
1022 break;
1023 default:
1024 switch (entry->descriptorType) {
1025 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
1026 case VK_DESCRIPTOR_TYPE_SAMPLER:
1027 /* Immutable samplers are copied into push descriptors when they are pushed */
1028 if (pCreateInfo->templateType == VK_DESCRIPTOR_UPDATE_TEMPLATE_TYPE_PUSH_DESCRIPTORS_KHR &&
1029 binding_layout->immutable_samplers_offset && !binding_layout->immutable_samplers_equal) {
1030 immutable_samplers = radv_immutable_samplers(set_layout, binding_layout) + entry->dstArrayElement * 4;
1031 }
1032 break;
1033 default:
1034 break;
1035 }
1036 dst_offset = binding_layout->offset / 4 + binding_layout->size * entry->dstArrayElement / 4;
1037 dst_stride = binding_layout->size / 4;
1038 break;
1039 }
1040
1041 templ->entry[i] = (struct radv_descriptor_update_template_entry) {
1042 .descriptor_type = entry->descriptorType,
1043 .descriptor_count = entry->descriptorCount,
1044 .src_offset = entry->offset,
1045 .src_stride = entry->stride,
1046 .dst_offset = dst_offset,
1047 .dst_stride = dst_stride,
1048 .buffer_offset = buffer_offset,
1049 .has_sampler = !binding_layout->immutable_samplers_offset,
1050 .immutable_samplers = immutable_samplers
1051 };
1052 }
1053
1054 *pDescriptorUpdateTemplate = radv_descriptor_update_template_to_handle(templ);
1055 return VK_SUCCESS;
1056 }
1057
1058 void radv_DestroyDescriptorUpdateTemplate(VkDevice _device,
1059 VkDescriptorUpdateTemplateKHR descriptorUpdateTemplate,
1060 const VkAllocationCallbacks *pAllocator)
1061 {
1062 RADV_FROM_HANDLE(radv_device, device, _device);
1063 RADV_FROM_HANDLE(radv_descriptor_update_template, templ, descriptorUpdateTemplate);
1064
1065 if (!templ)
1066 return;
1067
1068 vk_free2(&device->alloc, pAllocator, templ);
1069 }
1070
1071 void radv_update_descriptor_set_with_template(struct radv_device *device,
1072 struct radv_cmd_buffer *cmd_buffer,
1073 struct radv_descriptor_set *set,
1074 VkDescriptorUpdateTemplateKHR descriptorUpdateTemplate,
1075 const void *pData)
1076 {
1077 RADV_FROM_HANDLE(radv_descriptor_update_template, templ, descriptorUpdateTemplate);
1078 uint32_t i;
1079
1080 for (i = 0; i < templ->entry_count; ++i) {
1081 struct radeon_winsys_bo **buffer_list = set->descriptors + templ->entry[i].buffer_offset;
1082 uint32_t *pDst = set->mapped_ptr + templ->entry[i].dst_offset;
1083 const uint8_t *pSrc = ((const uint8_t *) pData) + templ->entry[i].src_offset;
1084 uint32_t j;
1085
1086 for (j = 0; j < templ->entry[i].descriptor_count; ++j) {
1087 switch (templ->entry[i].descriptor_type) {
1088 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER_DYNAMIC:
1089 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER_DYNAMIC: {
1090 const unsigned idx = templ->entry[i].dst_offset + j;
1091 assert(!(set->layout->flags & VK_DESCRIPTOR_SET_LAYOUT_CREATE_PUSH_DESCRIPTOR_BIT_KHR));
1092 write_dynamic_buffer_descriptor(device, set->dynamic_descriptors + idx,
1093 buffer_list, (struct VkDescriptorBufferInfo *) pSrc);
1094 break;
1095 }
1096 case VK_DESCRIPTOR_TYPE_UNIFORM_BUFFER:
1097 case VK_DESCRIPTOR_TYPE_STORAGE_BUFFER:
1098 write_buffer_descriptor(device, cmd_buffer, pDst, buffer_list,
1099 (struct VkDescriptorBufferInfo *) pSrc);
1100 break;
1101 case VK_DESCRIPTOR_TYPE_UNIFORM_TEXEL_BUFFER:
1102 case VK_DESCRIPTOR_TYPE_STORAGE_TEXEL_BUFFER:
1103 write_texel_buffer_descriptor(device, cmd_buffer, pDst, buffer_list,
1104 *(VkBufferView *) pSrc);
1105 break;
1106 case VK_DESCRIPTOR_TYPE_SAMPLED_IMAGE:
1107 case VK_DESCRIPTOR_TYPE_STORAGE_IMAGE:
1108 case VK_DESCRIPTOR_TYPE_INPUT_ATTACHMENT:
1109 write_image_descriptor(device, cmd_buffer, pDst, buffer_list,
1110 templ->entry[i].descriptor_type,
1111 (struct VkDescriptorImageInfo *) pSrc);
1112 break;
1113 case VK_DESCRIPTOR_TYPE_COMBINED_IMAGE_SAMPLER:
1114 write_combined_image_sampler_descriptor(device, cmd_buffer, pDst, buffer_list,
1115 templ->entry[i].descriptor_type,
1116 (struct VkDescriptorImageInfo *) pSrc,
1117 templ->entry[i].has_sampler);
1118 if (templ->entry[i].immutable_samplers)
1119 memcpy(pDst + 16, templ->entry[i].immutable_samplers + 4 * j, 16);
1120 break;
1121 case VK_DESCRIPTOR_TYPE_SAMPLER:
1122 if (templ->entry[i].has_sampler)
1123 write_sampler_descriptor(device, pDst,
1124 (struct VkDescriptorImageInfo *) pSrc);
1125 else if (templ->entry[i].immutable_samplers)
1126 memcpy(pDst, templ->entry[i].immutable_samplers + 4 * j, 16);
1127 break;
1128 default:
1129 unreachable("unimplemented descriptor type");
1130 break;
1131 }
1132 pSrc += templ->entry[i].src_stride;
1133 pDst += templ->entry[i].dst_stride;
1134 ++buffer_list;
1135 }
1136 }
1137 }
1138
1139 void radv_UpdateDescriptorSetWithTemplate(VkDevice _device,
1140 VkDescriptorSet descriptorSet,
1141 VkDescriptorUpdateTemplateKHR descriptorUpdateTemplate,
1142 const void *pData)
1143 {
1144 RADV_FROM_HANDLE(radv_device, device, _device);
1145 RADV_FROM_HANDLE(radv_descriptor_set, set, descriptorSet);
1146
1147 radv_update_descriptor_set_with_template(device, NULL, set, descriptorUpdateTemplate, pData);
1148 }
1149
1150
1151 VkResult radv_CreateSamplerYcbcrConversion(VkDevice device,
1152 const VkSamplerYcbcrConversionCreateInfo* pCreateInfo,
1153 const VkAllocationCallbacks* pAllocator,
1154 VkSamplerYcbcrConversion* pYcbcrConversion)
1155 {
1156 *pYcbcrConversion = VK_NULL_HANDLE;
1157 return VK_SUCCESS;
1158 }
1159
1160
1161 void radv_DestroySamplerYcbcrConversion(VkDevice device,
1162 VkSamplerYcbcrConversion ycbcrConversion,
1163 const VkAllocationCallbacks* pAllocator)
1164 {
1165 /* Do nothing. */
1166 }