Skip to content

Commit ac51200

Browse files
committed
Refactor the NVRTC build list helper
1 parent 9e60fb9 commit ac51200

File tree

12 files changed

+315
-291
lines changed

12 files changed

+315
-291
lines changed

c/parallel/src/for.cu

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -104,25 +104,25 @@ CUresult cccl_device_for_build(
104104
std::string lowered_name;
105105

106106
auto cl =
107-
make_nvrtc_command_list()
108-
.add_program(nvrtc_translation_unit{device_for_kernel, name})
109-
.add_expression({for_kernel_name})
110-
.compile_program({args, num_args})
111-
.get_name({for_kernel_name, lowered_name})
112-
.cleanup_program()
113-
.add_link({op.ltoir, op.ltoir_size});
107+
begin_linking_nvrtc_program(num_lto_args, lopts)
108+
->add_program(nvrtc_translation_unit{device_for_kernel, name})
109+
->add_expression({for_kernel_name})
110+
->compile_program({args, num_args})
111+
->get_name({for_kernel_name, lowered_name})
112+
->link_program()
113+
->add_link({op.ltoir, op.ltoir_size});
114114

115115
nvrtc_link_result result{};
116116

117117
if (cccl_iterator_kind_t::CCCL_ITERATOR == d_data.type)
118118
{
119-
result = cl.add_link({d_data.advance.ltoir, d_data.advance.ltoir_size})
120-
.add_link({d_data.dereference.ltoir, d_data.dereference.ltoir_size})
121-
.finalize_program(num_lto_args, lopts);
119+
result = cl->add_link({d_data.advance.ltoir, d_data.advance.ltoir_size})
120+
->add_link({d_data.dereference.ltoir, d_data.dereference.ltoir_size})
121+
->finalize_program();
122122
}
123123
else
124124
{
125-
result = cl.finalize_program(num_lto_args, lopts);
125+
result = cl->finalize_program();
126126
}
127127

128128
cuLibraryLoadData(&build_ptr->library, result.data.get(), nullptr, nullptr, 0, nullptr, nullptr, 0);

c/parallel/src/merge_sort.cu

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -427,18 +427,18 @@ struct device_merge_sort_vsmem_helper {{
427427
list_appender.add_iterator_definition(output_items_it);
428428

429429
nvrtc_link_result result =
430-
make_nvrtc_command_list()
431-
.add_program(nvrtc_translation_unit{src.c_str(), name})
432-
.add_expression({block_sort_kernel_name})
433-
.add_expression({partition_kernel_name})
434-
.add_expression({merge_kernel_name})
435-
.compile_program({args, num_args})
436-
.get_name({block_sort_kernel_name, block_sort_kernel_lowered_name})
437-
.get_name({partition_kernel_name, partition_kernel_lowered_name})
438-
.get_name({merge_kernel_name, merge_kernel_lowered_name})
439-
.cleanup_program()
440-
.add_link_list(ltoir_list)
441-
.finalize_program(num_lto_args, lopts);
430+
begin_linking_nvrtc_program(num_lto_args, lopts)
431+
->add_program(nvrtc_translation_unit{src.c_str(), name})
432+
->add_expression({block_sort_kernel_name})
433+
->add_expression({partition_kernel_name})
434+
->add_expression({merge_kernel_name})
435+
->compile_program({args, num_args})
436+
->get_name({block_sort_kernel_name, block_sort_kernel_lowered_name})
437+
->get_name({partition_kernel_name, partition_kernel_lowered_name})
438+
->get_name({merge_kernel_name, merge_kernel_lowered_name})
439+
->link_program()
440+
->add_link_list(ltoir_list)
441+
->finalize_program();
442442

443443
cuLibraryLoadData(&build_ptr->library, result.data.get(), nullptr, nullptr, 0, nullptr, nullptr, 0);
444444
check(

0 commit comments

Comments
 (0)