[LegNeato]

We use the cuda device allocator for allocations on the GPU via Rust's default allocator.

[saagarjha]

Have you considered “allocating” out of shared memory instead?