Fix

NVIDIA · Jan 27, 2025 · f99c90c · f99c90c
1 parent 11552b5
commit f99c90c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/c/parallel/src/scan.cu b/c/parallel/src/scan.cu
@@ -302,7 +302,8 @@ struct scan_kernel_source
   }
   scan_tile_state TileState()
   {
-    return *(reinterpret_cast<scan_tile_state*>(build.tile_state));
+    auto result = (reinterpret_cast<scan_tile_state*>(build.tile_state));
+    return *result;
   }
 };
 
@@ -477,7 +478,7 @@ extern "C" CCCL_C_API CUresult cccl_device_scan_build(
     }
 
     bool is_primitive = not(accum_t.type == cccl_type_enum::STORAGE);
-    auto tile_state   = std::make_unique<scan::scan_tile_state>(status_size, is_primitive, uninitialized_size);
+    auto tile_state   = std::make_unique<scan::scan_tile_state>(is_primitive, status_size, uninitialized_size);
 
     build->cc               = cc;
     build->cubin            = (void*) result.cubin.release();