fdo-mirrors/mesa
1
0
Fork 0
mirror of https://gitlab.freedesktop.org/mesa/mesa.git synced 2025-12-24 15:20:10 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

radv: Add metadata to acceleration structures

Browse source 
This metadata is used for dumping RRA traces.

Reviewed-by: Konstantin Seurer <konstantin.seurer@gmail.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/18166>
This commit is contained in:
Friedrich Vock 2022-08-29 13:33:54 +02:00 • committed by Marge Bot
parent 0fefdadbd3
commit 07eceb4fe4
2 changed files with 38 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
src/amd/vulkan/bvh/bvh.h
View file

@ -46,6 +46,12 @@ struct radv_accel_struct_serialization_header {
#endif
};
struct radv_accel_struct_geometry_info {
uint32_t primitive_count;
uint32_t flags;
uint32_t type;
};
struct radv_accel_struct_header {
uint32_t root_node_offset;
uint32_t reserved;
@ -55,10 +61,12 @@ struct radv_accel_struct_header {
uint64_t compacted_size;
uint64_t serialization_size;
uint32_t copy_dispatch_size[3];
uint32_t reserved2;
uint32_t geometry_count;
uint64_t instance_offset;
uint64_t instance_count;
uint64_t size;
uint32_t build_flags;
uint32_t internal_node_count;
};
struct radv_bvh_triangle_node {

30
src/amd/vulkan/radv_acceleration_structure.c
View file

@ -94,6 +94,8 @@ radv_GetAccelerationStructureBuildSizesKHR(
uint64_t size = boxes * 128 + instances * 128 + triangles * 64 + internal_nodes * 128 +
ALIGN(sizeof(struct radv_accel_struct_header), 64);
size +=
pBuildInfo->geometryCount * sizeof(struct radv_accel_struct_geometry_info);
pSizeInfo->accelerationStructureSize = size;
@ -657,6 +659,7 @@ struct bvh_state {
uint32_t leaf_node_offset;
uint32_t leaf_node_count;
uint32_t internal_node_count;
};
VKAPI_ATTR void VKAPI_CALL
@ -890,6 +893,7 @@ radv_CmdBuildAccelerationStructuresKHR(
if (!final_iter)
bvh_states[i].node_offset += dst_node_count * 128;
bvh_states[i].node_count = dst_node_count;
bvh_states[i].internal_node_count += dst_node_count;
bvh_states[i].scratch_offset = dst_scratch_offset;
}
}
@ -901,9 +905,12 @@ radv_CmdBuildAccelerationStructuresKHR(
bool is_tlas = pInfos[i].type == VK_ACCELERATION_STRUCTURE_TYPE_TOP_LEVEL_KHR;
uint64_t geometry_infos_size =
pInfos[i].geometryCount * sizeof(struct radv_accel_struct_geometry_info);
header.instance_offset = bvh_states[i].leaf_node_offset;
header.instance_count = is_tlas ? bvh_states[i].leaf_node_count : 0;
header.compacted_size = bvh_states[i].node_offset;
header.compacted_size = bvh_states[i].node_offset + geometry_infos_size;
header.copy_dispatch_size[0] = DIV_ROUND_UP(header.compacted_size, 16 * 64);
header.copy_dispatch_size[1] = 1;
@ -918,10 +925,31 @@ radv_CmdBuildAccelerationStructuresKHR(
sizeof(struct radv_accel_struct_serialization_header) -
sizeof(uint64_t) * header.instance_count;
header.build_flags = pInfos[i].flags;
header.geometry_count = pInfos[i].geometryCount;
header.internal_node_count = bvh_states[i].internal_node_count;
struct radv_accel_struct_geometry_info *geometry_infos = malloc(geometry_infos_size);
if (!geometry_infos)
goto fail;
for (uint32_t j = 0; j < pInfos[i].geometryCount; ++j) {
const VkAccelerationStructureGeometryKHR *geometry =
pInfos[i].pGeometries ? pInfos[i].pGeometries + j : pInfos[i].ppGeometries[j];
geometry_infos[j].type = geometry->geometryType;
geometry_infos[j].flags = geometry->flags;
geometry_infos[j].primitive_count = ppBuildRangeInfos[i][j].primitiveCount;
}
radv_update_buffer_cp(cmd_buffer,
radv_buffer_get_va(accel_struct->bo) + accel_struct->mem_offset + base,
(const char *)&header + base, sizeof(header) - base);
radv_update_buffer_cp(cmd_buffer,
radv_accel_struct_get_va(accel_struct) + bvh_states[i].node_offset,
geometry_infos, geometry_infos_size);
}
fail:
free(bvh_states);
radv_meta_restore(&saved_state, cmd_buffer);
}