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

Open
wants to merge 2 commits into
base: master
Choose a base branch
from
+69 −2
Open

Add ray tracing spirv intrinsics #914

Changes from all commits
Commits
Show all changes
2 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
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
71 changes: 69 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,81 @@ static const float32_t3 HitTriangleVertexPositionsKHR[3];
[[vk::ext_instruction(spv::OpRayQueryGetIntersectionTriangleVertexPositionsKHR)]]
float3 rayQueryGetIntersectionTriangleVertexPositionsKHR([[vk::ext_reference]] RayQueryKHR query, uint32_t committed)[3];

[[vk::ext_builtin_input(spv::BuiltInLaunchIdKHR)]]
static const uint32_t3 LaunchIdKHR;

[[vk::ext_builtin_input(spv::BuiltInLaunchSizeKHR)]]
static const uint32_t3 LaunchSizeKHR;

//! Ray Tracing Pipeline stuff (TODO: @kevyu)
[[vk::ext_builtin_input(spv::BuiltInInstanceCustomIndexKHR)]]
static const uint32_t InstanceCustomIndexKHR;

[[vk::ext_builtin_input(spv::BuiltInRayGeometryIndexKHR)]]
static const uint32_t RayGeometryIndexKHR;

[[vk::ext_builtin_input(spv::BuiltInWorldRayOriginKHR)]]
static const float32_t3 WorldRayOriginKHR;

[[vk::ext_builtin_input(spv::BuiltInWorldRayDirectionKHR)]]
static const float32_t3 WorldRayDirectionKHR;

[[vk::ext_builtin_input(spv::BuiltInObjectRayOriginKHR)]]
static const float32_t3 ObjectRayOriginKHR;

[[vk::ext_builtin_input(spv::BuiltInObjectRayDirectionKHR)]]
static const float32_t3 ObjectRayDirectionKHR;

[[vk::ext_builtin_input(spv::BuiltInRayTminKHR)]]
static const float32_t RayTminKHR;

[[vk::ext_builtin_input(spv::BuiltInRayTmaxKHR)]]
static const float32_t RayTmaxKHR;

[[vk::ext_builtin_input(spv::BuiltInObjectToWorldKHR)]]
static const float32_t4x3 ObjectToWorldKHR;

[[vk::ext_builtin_input(spv::BuiltInWorldToObjectKHR)]]
static const float32_t4x3 WorldToObjectKHR;

[[vk::ext_builtin_input(spv::BuiltInHitKindKHR)]]
static const uint32_t HitKindKHR;

[[vk::ext_builtin_input(spv::BuiltInIncomingRayFlagsKHR)]]
static const uint32_t IncomingRayFlagsKHR;
Comment on lines +89 to +129
Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

can you add capabilities and extensions to builtins?

Copy link
Member

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

We've definitely added capabilities to builtins

Nabla/include/nbl/builtin/hlsl/spirv_intrinsics/subgroup_ballot.hlsl

Line 27 in 33d2a57

[[vk::ext_capability(spv::CapabilityGroupNonUniformBallot)]]

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Done.


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