From e6e04d09b8bc8217c506c39af20dbe1b017ae9b4 Mon Sep 17 00:00:00 2001
From: Jonathan Wong <131846439+nvjonwong@users.noreply.github.com>
Date: Fri, 16 Jan 2026 18:24:40 +0000
Subject: [PATCH] Fix to ensure freeing comes back clean

---
 cuBQL/builder/cuda/gpu_builder.h | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/cuBQL/builder/cuda/gpu_builder.h b/cuBQL/builder/cuda/gpu_builder.h
index 9197ef9..6e1e45c 100644
--- a/cuBQL/builder/cuda/gpu_builder.h
+++ b/cuBQL/builder/cuda/gpu_builder.h
@@ -62,9 +62,10 @@ namespace cuBQL {
               cudaStream_t       s,
               GpuMemoryResource &memResource)
     {
+      CUBQL_CUDA_CALL(StreamSynchronize(s));  // Sync BEFORE freeing
       gpuBuilder_impl::_FREE(bvh.primIDs,s,memResource);
       gpuBuilder_impl::_FREE(bvh.nodes,s,memResource);
-      CUBQL_CUDA_CALL(StreamSynchronize(s));
+      CUBQL_CUDA_CALL(StreamSynchronize(s));  // And AFTER
       bvh.primIDs = 0;
     }
   }