src/amd/vulkan/bvh/header.comp

/*
 * Copyright © 2023 Valve Corporation
 *
 * SPDX-License-Identifier: MIT
 */

#version 460

layout(local_size_x = 64, local_size_y = 1, local_size_z = 1) in;

#include "bvh_defines.h"
#define VK_USED_BUILD_FLAGS RADV_HEADER_BUILD_FLAGS
#include "bvh_helpers.h"

layout(push_constant) uniform CONSTS
{
   header_args args;
};

void
main(void)
{
   DEREF(args.dst).update_dispatch_size[0] =
      ((DEREF(args.src).dst_node_offset - args.internal_nodes_offset) / SIZEOF(radv_gfx12_box_node) + 1) * 8;
   DEREF(args.dst).update_dispatch_size[1] = 1;
   DEREF(args.dst).update_dispatch_size[2] = 1;

   uint32_t compacted_size = args.bvh_offset + DEREF(args.src).dst_node_offset;

   uint32_t blas_addrs_size = 0;
   if (args.type == VK_ACCELERATION_STRUCTURE_TYPE_TOP_LEVEL_KHR)
      blas_addrs_size = args.primitive_count * SIZEOF(uint64_t);

   uint32_t serialization_size =
      compacted_size + align(SIZEOF(radv_accel_struct_serialization_header) + blas_addrs_size, 128);

   uint32_t size = serialization_size - SIZEOF(radv_accel_struct_serialization_header) - blas_addrs_size;

   DEREF(args.dst).compacted_size = compacted_size;

   DEREF(args.dst).copy_dispatch_size[0] = DIV_ROUND_UP(compacted_size, 16 * 64);
   DEREF(args.dst).copy_dispatch_size[1] = 1;
   DEREF(args.dst).copy_dispatch_size[2] = 1;

   DEREF(args.dst).serialization_size = serialization_size;

   DEREF(args.dst).size = size;
}