Cast to raw reference

NVIDIA · Sep 9, 2024 · 08b32dc · 08b32dc
1 parent 56a06b0
commit 08b32dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cub/cub/device/dispatch/dispatch_transform.cuh b/cub/cub/device/dispatch/dispatch_transform.cuh
@@ -109,7 +109,7 @@ _CCCL_DEVICE void transform_kernel_impl(
   RandomAccessIteratorsIn... ins)
 {
   auto op = [&](Offset i) {
-    out[i] = transform_op(ins[i]...);
+    out[i] = transform_op(THRUST_NS_QUALIFIER::raw_reference_cast(ins[i])...);
   };
   using OpT = decltype(op);