Skip to content

Commit

Permalink
Use argument elision with logical_token
Browse files Browse the repository at this point in the history
  • Loading branch information
@caugonnet
caugonnet committed Jan 18, 2025
1 parent 0c8497f commit 0eb223a
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 10 deletions.
10 changes: 5 additions & 5 deletions cpp/src/prims/detail/transform_v_frontier_e.cuh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -538,7 +538,7 @@ auto transform_v_frontier_e(raft::handle_t const& handle,
raft::grid_1d_block_t update_grid(high_size,
detail::transform_v_frontier_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tv_buffers[0].write())->*[&](cudaStream_t stream, auto /* aggregate_value_buffer */) {
cudastf_ctx.task(l_tv_buffers[0].write())->*[&](cudaStream_t stream) {


detail::transform_v_frontier_e_high_degree<GraphViewType>
Expand All @@ -562,7 +562,7 @@ auto transform_v_frontier_e(raft::handle_t const& handle,
raft::grid_1d_warp_t update_grid(mid_size,
detail::transform_v_frontier_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tv_buffers[1].write())->*[&](cudaStream_t stream, auto ) {
cudastf_ctx.task(l_tv_buffers[1].write())->*[&](cudaStream_t stream) {
detail::transform_v_frontier_e_mid_degree<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -584,7 +584,7 @@ auto transform_v_frontier_e(raft::handle_t const& handle,
raft::grid_1d_thread_t update_grid(low_size,
detail::transform_v_frontier_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tv_buffers[2].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tv_buffers[2].write())->*[&](cudaStream_t stream) {
detail::transform_v_frontier_e_hypersparse_or_low_degree<false, GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -606,7 +606,7 @@ auto transform_v_frontier_e(raft::handle_t const& handle,
raft::grid_1d_thread_t update_grid(hypersparse_size,
detail::transform_v_frontier_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tv_buffers[3].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tv_buffers[3].write())->*[&](cudaStream_t stream) {
detail::transform_v_frontier_e_hypersparse_or_low_degree<true, GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -627,7 +627,7 @@ auto transform_v_frontier_e(raft::handle_t const& handle,
detail::transform_v_frontier_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);

cudastf_ctx.task(l_tv_buffers[4].write())->*[&,i](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tv_buffers[4].write())->*[&,i](cudaStream_t stream) {
detail::transform_v_frontier_e_hypersparse_or_low_degree<false, GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand Down
10 changes: 5 additions & 5 deletions cpp/src/prims/transform_reduce_e.cuh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -525,7 +525,7 @@ T transform_reduce_e(raft::handle_t const& handle,
raft::grid_1d_block_t update_grid((*segment_offsets)[1],
detail::transform_reduce_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tr_buffers[0].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tr_buffers[0].write())->*[&](cudaStream_t stream) {
detail::transform_reduce_e_high_degree<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -543,7 +543,7 @@ T transform_reduce_e(raft::handle_t const& handle,
raft::grid_1d_warp_t update_grid((*segment_offsets)[2] - (*segment_offsets)[1],
detail::transform_reduce_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tr_buffers[1].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tr_buffers[1].write())->*[&](cudaStream_t stream) {
detail::transform_reduce_e_mid_degree<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -561,7 +561,7 @@ T transform_reduce_e(raft::handle_t const& handle,
raft::grid_1d_thread_t update_grid((*segment_offsets)[3] - (*segment_offsets)[2],
detail::transform_reduce_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tr_buffers[2].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tr_buffers[2].write())->*[&](cudaStream_t stream) {
detail::transform_reduce_e_low_degree<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -579,7 +579,7 @@ T transform_reduce_e(raft::handle_t const& handle,
raft::grid_1d_thread_t update_grid(*(edge_partition.dcs_nzd_vertex_count()),
detail::transform_reduce_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);
cudastf_ctx.task(l_tr_buffers[3].write())->*[&](cudaStream_t stream, auto) {
cudastf_ctx.task(l_tr_buffers[3].write())->*[&](cudaStream_t stream) {
detail::transform_reduce_e_hypersparse<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
edge_partition,
Expand All @@ -597,7 +597,7 @@ T transform_reduce_e(raft::handle_t const& handle,
detail::transform_reduce_e_kernel_block_size,
handle.get_device_properties().maxGridSize[0]);

cudastf_ctx.task(l_tr_buffers[4].write())->*[&](cudaStream_t stream, auto ) {
cudastf_ctx.task(l_tr_buffers[4].write())->*[&](cudaStream_t stream) {

detail::transform_reduce_e_low_degree<GraphViewType>
<<<update_grid.num_blocks, update_grid.block_size, 0, stream>>>(
Expand Down

0 comments on commit 0eb223a

Please # to comment.