Skip to content

Commit fccab79

Browse files
committed
Use proclaim_return_type to wrap extended device lambda
1 parent b7e99d9 commit fccab79

File tree

1 file changed

+7
-6
lines changed

1 file changed

+7
-6
lines changed

source/pbat/gpu/impl/geometry/SweepAndPrune.cu

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -104,12 +104,13 @@ void RunSweepAndPruneTests()
104104
gpu::impl::geometry::SweepAndPrune sap{};
105105
sap.SortAndSweep(
106106
aabbs,
107-
[nEdges, o = overlaps.Raw()] PBAT_DEVICE(GpuIndex si, GpuIndex sj) mutable -> void {
108-
if (si < nEdges and sj >= nEdges)
109-
o.Append(OverlapType{si, sj - nEdges});
110-
if (si >= nEdges and sj < nEdges)
111-
o.Append(OverlapType{sj, si - nEdges});
112-
});
107+
cuda::proclaim_return_type<void>(
108+
[nEdges, o = overlaps.Raw()] PBAT_DEVICE(GpuIndex si, GpuIndex sj) mutable {
109+
if (si < nEdges and sj >= nEdges)
110+
o.Append(OverlapType{si, sj - nEdges});
111+
if (si >= nEdges and sj < nEdges)
112+
o.Append(OverlapType{sj, si - nEdges});
113+
}));
113114
std::vector<OverlapType> overlapsCpu = overlaps.Get();
114115
// Assert
115116
for (OverlapType overlap : overlapsCpu)

0 commit comments

Comments
 (0)