perf(mesh): release per-mesh scratch buffer for static BLAS (#66) #104
2 changed files with 33 additions and 0 deletions
|
|
@ -152,6 +152,23 @@ namespace {
|
|||
|
||||
self.buildFlags = flags;
|
||||
self.builtPrimitiveCount = primitiveCount;
|
||||
|
||||
// Static (!allowUpdate) meshes can never refit, so their per-mesh
|
||||
// scratch is dead weight the moment this build's GPU work completes —
|
||||
// release it instead of keeping a persistent DEVICE_LOCAL allocation
|
||||
// per mesh (the waste that dominates many-mesh scenes, issue #66). The
|
||||
// fresh build recorded above still reads scratchBuffer.address from
|
||||
// this command buffer, so a plain Clear() would be a use-after-free;
|
||||
// DeferredClear() retires the allocation only once the build's frame
|
||||
// has passed its fence (the queue from #101/#102). A later Refit on a
|
||||
// static mesh takes the rebuild fallback, whose Resize sees the nulled
|
||||
// handle and re-Creates the scratch. Refit-capable meshes keep theirs:
|
||||
// an in-place UPDATE reuses it every frame (build scratch ≥ update
|
||||
// scratch, so no resize). Only applies to a fresh build — an UPDATE
|
||||
// never reaches here for a static mesh.
|
||||
if (!update && !self.allowUpdate) {
|
||||
self.scratchBuffer.DeferredClear();
|
||||
}
|
||||
}
|
||||
|
||||
void RecordBLASBuild(Mesh& self, std::uint32_t vertexCount, std::uint32_t indexCount, VkBuildAccelerationStructureFlagsKHR flags, bool update, VkCommandBuffer cmd) {
|
||||
|
|
|
|||
|
|
@ -128,6 +128,10 @@ int main() {
|
|||
Check((tri.buildFlags & VK_BUILD_ACCELERATION_STRUCTURE_ALLOW_UPDATE_BIT_KHR) != 0,
|
||||
"allowUpdate=true → ALLOW_UPDATE bit set");
|
||||
Check(tri.builtPrimitiveCount == 12, "triangle BLAS reports 12 primitives");
|
||||
// allowUpdate=true keeps the scratch buffer alive so the in-place UPDATE
|
||||
// refit below can reuse it (build scratch ≥ update scratch, no resize).
|
||||
Check(tri.scratchBuffer.buffer != VK_NULL_HANDLE,
|
||||
"allowUpdate=true → scratch retained for refit (issue #66)");
|
||||
|
||||
const VkDeviceAddress triAddrBefore = tri.blasAddr;
|
||||
const VkAccelerationStructureKHR triHandleBefore = tri.accelerationStructure;
|
||||
|
|
@ -157,6 +161,12 @@ int main() {
|
|||
"FastBuild preference → PREFER_FAST_BUILD bit set");
|
||||
Check((triFast.buildFlags & VK_BUILD_ACCELERATION_STRUCTURE_ALLOW_UPDATE_BIT_KHR) == 0,
|
||||
"allowUpdate=false → ALLOW_UPDATE bit clear");
|
||||
// A static mesh can never refit, so its scratch is released right after
|
||||
// the build completes rather than kept as persistent VRAM (issue #66).
|
||||
// DeferredClear nulls the handle immediately (the allocation itself is
|
||||
// retired by the fence-keyed queue once the build's frame passes).
|
||||
Check(triFast.scratchBuffer.buffer == VK_NULL_HANDLE,
|
||||
"allowUpdate=false → scratch released after build (issue #66)");
|
||||
|
||||
// Refit without allowUpdate must still succeed via the rebuild fallback.
|
||||
{
|
||||
|
|
@ -167,6 +177,10 @@ int main() {
|
|||
SubmitWait(cmd);
|
||||
}
|
||||
Check(triFast.blasAddr != 0, "Refit fallback rebuild still produced a valid BLAS");
|
||||
// The fallback rebuild re-Created the scratch (Resize saw the nulled
|
||||
// handle) and, still being a static mesh, released it again afterwards.
|
||||
Check(triFast.scratchBuffer.buffer == VK_NULL_HANDLE,
|
||||
"static-mesh refit rebuild re-released its scratch (issue #66)");
|
||||
|
||||
// ── Procedural (AABB) BLAS: build with options, then refit. ─────────
|
||||
Mesh proc;
|
||||
|
|
@ -182,6 +196,8 @@ int main() {
|
|||
}
|
||||
Check(proc.blasAddr != 0, "procedural Build produced a non-zero blasAddr");
|
||||
Check(proc.builtPrimitiveCount == 2, "procedural BLAS reports 2 primitives");
|
||||
Check(proc.scratchBuffer.buffer != VK_NULL_HANDLE,
|
||||
"procedural allowUpdate=true → scratch retained for refit (issue #66)");
|
||||
|
||||
const VkDeviceAddress procAddrBefore = proc.blasAddr;
|
||||
const VkAccelerationStructureKHR procHandleBefore = proc.accelerationStructure;
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue