mirror of
https://gitlab.freedesktop.org/mesa/mesa.git
synced 2026-05-22 00:08:09 +02:00
ae9ea27e0d
Acked-by: Alyssa Rosenzweig <alyssa.rosenzweig@intel.com> Acked-by: Pierre-Eric Pelloux-Prayer <pierre-eric.pelloux-prayer@amd.com> Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/40383>
141 lines
3.8 KiB
C
141 lines
3.8 KiB
C
/*
|
|
* Copyright © 2016 Red Hat.
|
|
* Copyright © 2016 Bas Nieuwenhuizen
|
|
*
|
|
* based in part on anv driver which is:
|
|
* Copyright © 2015 Intel Corporation
|
|
*
|
|
* SPDX-License-Identifier: MIT
|
|
*/
|
|
|
|
#ifndef RADV_PIPELINE_RT_H
|
|
#define RADV_PIPELINE_RT_H
|
|
|
|
#include "util/bitset.h"
|
|
#include "aco_nir_call_attribs.h"
|
|
#include "radv_pipeline_compute.h"
|
|
#include "radv_shader.h"
|
|
|
|
struct radv_ray_tracing_pipeline {
|
|
struct radv_compute_pipeline base;
|
|
|
|
struct radv_shader *prolog;
|
|
|
|
struct radv_ray_tracing_stage *stages;
|
|
struct radv_ray_tracing_group *groups;
|
|
unsigned stage_count;
|
|
unsigned non_imported_stage_count;
|
|
unsigned group_count;
|
|
|
|
uint32_t stack_size;
|
|
uint32_t traversal_stack_size;
|
|
|
|
/* set if any shaders from this pipeline require robustness2 in the merged traversal shader */
|
|
bool traversal_storage_robustness2 : 1;
|
|
bool traversal_uniform_robustness2 : 1;
|
|
};
|
|
|
|
RADV_DECL_PIPELINE_DOWNCAST(ray_tracing, RADV_PIPELINE_RAY_TRACING)
|
|
|
|
struct radv_pipeline_group_handle {
|
|
uint64_t recursive_shader_ptr;
|
|
uint64_t ahit_isec_ptr;
|
|
|
|
union {
|
|
uint32_t general_index;
|
|
uint32_t closest_hit_index;
|
|
};
|
|
union {
|
|
uint32_t intersection_index;
|
|
uint32_t any_hit_index;
|
|
};
|
|
};
|
|
|
|
struct radv_rt_capture_replay_handle {
|
|
struct radv_serialized_shader_arena_block recursive_shader_alloc;
|
|
struct radv_serialized_shader_arena_block ahit_isec_alloc;
|
|
uint32_t non_recursive_idx;
|
|
};
|
|
|
|
struct radv_ray_tracing_group {
|
|
VkRayTracingShaderGroupTypeKHR type;
|
|
uint32_t recursive_shader; /* generalShader or closestHitShader */
|
|
uint32_t any_hit_shader;
|
|
uint32_t intersection_shader;
|
|
struct radv_shader *ahit_isec_shader;
|
|
struct radv_pipeline_group_handle handle;
|
|
};
|
|
|
|
enum radv_rt_const_arg_state {
|
|
RADV_RT_CONST_ARG_STATE_UNINITIALIZED,
|
|
RADV_RT_CONST_ARG_STATE_VALID,
|
|
RADV_RT_CONST_ARG_STATE_INVALID,
|
|
};
|
|
|
|
struct radv_rt_const_arg_info {
|
|
enum radv_rt_const_arg_state state;
|
|
uint32_t value;
|
|
};
|
|
|
|
struct radv_ray_tracing_stage_info {
|
|
bool can_inline;
|
|
bool has_position_fetch;
|
|
|
|
BITSET_DECLARE(unused_args, CPS_ARG_COUNT);
|
|
|
|
struct radv_rt_const_arg_info tmin;
|
|
struct radv_rt_const_arg_info tmax;
|
|
|
|
struct radv_rt_const_arg_info sbt_offset;
|
|
struct radv_rt_const_arg_info sbt_stride;
|
|
|
|
struct radv_rt_const_arg_info miss_index;
|
|
|
|
uint32_t set_flags;
|
|
uint32_t unset_flags;
|
|
};
|
|
|
|
struct radv_ray_tracing_stage {
|
|
struct vk_pipeline_cache_object *nir;
|
|
struct radv_shader *shader;
|
|
mesa_shader_stage stage;
|
|
uint32_t stack_size;
|
|
bool needs_nir;
|
|
|
|
struct radv_ray_tracing_stage_info info;
|
|
|
|
uint8_t blake3[BLAKE3_KEY_LEN];
|
|
};
|
|
|
|
struct radv_ray_tracing_state_key {
|
|
uint32_t stage_count;
|
|
struct radv_ray_tracing_stage *stages;
|
|
|
|
uint32_t group_count;
|
|
struct radv_ray_tracing_group *groups;
|
|
|
|
struct radv_shader_stage_key stage_keys[MESA_VULKAN_SHADER_STAGES];
|
|
};
|
|
|
|
void radv_destroy_ray_tracing_pipeline(struct radv_device *device, struct radv_ray_tracing_pipeline *pipeline);
|
|
|
|
void radv_ray_tracing_pipeline_hash(const struct radv_device *device,
|
|
const VkRayTracingPipelineCreateInfoKHR *pCreateInfo,
|
|
const struct radv_ray_tracing_state_key *rt_state, unsigned char *hash);
|
|
|
|
VkResult radv_generate_ray_tracing_state_key(struct radv_device *device,
|
|
const VkRayTracingPipelineCreateInfoKHR *pCreateInfo,
|
|
struct radv_ray_tracing_state_key *rt_state);
|
|
|
|
void radv_ray_tracing_state_key_finish(struct radv_ray_tracing_state_key *rt_state);
|
|
|
|
struct radv_ray_tracing_binary_header {
|
|
uint32_t is_traversal_shader : 1;
|
|
uint32_t has_shader : 1;
|
|
uint32_t has_nir : 1;
|
|
uint8_t stage_blake3[BLAKE3_KEY_LEN];
|
|
uint32_t stack_size;
|
|
struct radv_ray_tracing_stage_info stage_info;
|
|
};
|
|
|
|
#endif /* RADV_PIPELINE_RT */
|