anv: Change as_build to show num tlas/blas

As_build now shows the number of tlas/blas that was built.

Signed-off-by: Michael Cheng <michael.cheng@intel.com>
Reviewed-by: Lionel Landwerlin <lionel.g.landwerlin@intel.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/33720>
This commit is contained in:
Michael Cheng 2025-03-04 15:01:23 -08:00 committed by Marge Bot
parent 9ba7005d25
commit 014f376755
3 changed files with 15 additions and 2 deletions

View file

@ -229,7 +229,9 @@ def define_tracepoints(args):
tp_args=[Arg(type='uint32_t', var='count', c_format='%u'),],
need_cs_param=True)
begin_end_tp('as_build')
begin_end_tp('as_build',
tp_args=[Arg(type='uint32_t', var='n_tlas', c_format='%u'),
Arg(type='uint32_t', var='n_blas', c_format='%u'),])
begin_end_tp('as_build_leaves', repeat_last=True)
begin_end_tp('as_morton_generate', repeat_last=True)
begin_end_tp('as_morton_sort', repeat_last=True)

View file

@ -4086,6 +4086,8 @@ struct anv_cmd_ray_tracing_state {
} scratch;
uint32_t debug_marker_count;
uint32_t num_tlas;
uint32_t num_blas;
enum vk_acceleration_structure_build_step debug_markers[5];
struct anv_address build_priv_mem_addr;

View file

@ -43,8 +43,15 @@ begin_debug_marker(VkCommandBuffer commandBuffer,
step;
switch (step) {
case VK_ACCELERATION_STRUCTURE_BUILD_STEP_TOP:
{
va_list args;
va_start(args, format);
cmd_buffer->state.rt.num_tlas = va_arg(args, uint32_t);
cmd_buffer->state.rt.num_blas = va_arg(args, uint32_t);
va_end(args);
trace_intel_begin_as_build(&cmd_buffer->trace);
break;
}
case VK_ACCELERATION_STRUCTURE_BUILD_STEP_BUILD_LEAVES:
trace_intel_begin_as_build_leaves(&cmd_buffer->trace);
break;
@ -76,7 +83,9 @@ end_debug_marker(VkCommandBuffer commandBuffer)
cmd_buffer->state.rt.debug_marker_count--;
switch (cmd_buffer->state.rt.debug_markers[cmd_buffer->state.rt.debug_marker_count]) {
case VK_ACCELERATION_STRUCTURE_BUILD_STEP_TOP:
trace_intel_end_as_build(&cmd_buffer->trace);
trace_intel_end_as_build(&cmd_buffer->trace,
cmd_buffer->state.rt.num_tlas,
cmd_buffer->state.rt.num_blas);
break;
case VK_ACCELERATION_STRUCTURE_BUILD_STEP_BUILD_LEAVES:
trace_intel_end_as_build_leaves(&cmd_buffer->trace);