Skip to content

Add ray tracing spirv intrinsics #914

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 5 commits into from
Closed

Add ray tracing spirv intrinsics #914

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
5 commits
Select commit Hold shift + click to select a range
a73929d
Add ray tracing spirv intrinsics
Aug 15, 2025
33d2a57
Fix IncomingRayFlags intrinsics.
Aug 15, 2025
3f0509f
Add capability and extension for raytracing spirv builtin
Aug 18, 2025
26edc4a
Add PrimitiveId spirv intrinsics
Aug 20, 2025
892c868
Merge branch 'master' into ray_tracing_spirv_intrinsics
Aug 25, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
2 changes: 2 additions & 0 deletions include/nbl/builtin/hlsl/spirv_intrinsics/core.hlsl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -67,6 +67,8 @@ static const bool HelperInvocation;
static const uint32_t VertexIndex;
[[vk::ext_builtin_input(spv::BuiltInInstanceIndex)]]
static const uint32_t InstanceIndex;
[[vk::ext_builtin_input(spv::BuiltInPrimitiveId)]]
static const uint32_t PrimitiveId;

//! Vertex and friends
[[vk::ext_builtin_output(spv::BuiltInPosition)]]
Expand Down
99 changes: 97 additions & 2 deletions include/nbl/builtin/hlsl/spirv_intrinsics/raytracing.hlsl
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -86,14 +86,109 @@ static const float32_t3 HitTriangleVertexPositionsKHR[3];
[[vk::ext_instruction(spv::OpRayQueryGetIntersectionTriangleVertexPositionsKHR)]]
float3 rayQueryGetIntersectionTriangleVertexPositionsKHR([[vk::ext_reference]] RayQueryKHR query, uint32_t committed)[3];

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInLaunchIdKHR)]]
static const uint32_t3 LaunchIdKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInLaunchSizeKHR)]]
static const uint32_t3 LaunchSizeKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInInstanceCustomIndexKHR)]]
static const uint32_t InstanceCustomIndexKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInRayGeometryIndexKHR)]]
static const uint32_t RayGeometryIndexKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInWorldRayOriginKHR)]]
static const float32_t3 WorldRayOriginKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInWorldRayDirectionKHR)]]
static const float32_t3 WorldRayDirectionKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInObjectRayOriginKHR)]]
static const float32_t3 ObjectRayOriginKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInObjectRayDirectionKHR)]]
static const float32_t3 ObjectRayDirectionKHR;

//! Ray Tracing Pipeline stuff (TODO: @kevyu)
[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInRayTminKHR)]]
static const float32_t RayTminKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInRayTmaxKHR)]]
static const float32_t RayTmaxKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInObjectToWorldKHR)]]
static const float32_t4x3 ObjectToWorldKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInWorldToObjectKHR)]]
static const float32_t4x3 WorldToObjectKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInHitKindKHR)]]
static const uint32_t HitKindKHR;

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_builtin_input(spv::BuiltInIncomingRayFlagsKHR)]]
static const uint32_t IncomingRayFlagsKHR;

template <typename PayloadT>
[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpTraceRayKHR)]]
void traceRayKHR(AccelerationStructureKHR AS, uint32_t rayFlags, uint32_t cullMask, uint32_t sbtOffset, uint32_t sbtStride, uint32_t missIndex, float32_t3 rayOrigin, float32_t rayTmin, float32_t3 rayDirection, float32_t rayTmax, [[vk::ext_reference]] PayloadT payload);

template <typename PayloadT>
[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpTraceRayKHR)]]
void traceRayKHR(RaytracingAccelerationStructure AS, uint32_t rayFlags, uint32_t cullMask, uint32_t sbtOffset, uint32_t sbtStride, uint32_t missIndex, float32_t3 rayOrigin, float32_t rayTmin, float32_t3 rayDirection, float32_t rayTmax, [[vk::ext_reference]] PayloadT payload);

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpReportIntersectionKHR)]]
bool reportIntersectionKHR(float32_t hit, uint32_t hitKind);

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpIgnoreIntersectionKHR)]]
void ignoreIntersectionKHR();

[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpTerminateRayKHR)]]
void terminateRay();
void terminateRayKHR();

template <typename T>
[[vk::ext_capability(spv::CapabilityRayTracingKHR)]]
[[vk::ext_extension("SPV_KHR_ray_tracing")]]
[[vk::ext_instruction(spv::OpExecuteCallableKHR)]]
void executeCallable(uint32_t sbtIndex, [[vk::ext_reference]] T payload);


}
}
Expand Down
Loading