Lines Matching refs:set_arg
349 count_kernel.set_arg(0, *input_buffer); in radix_sort_impl()
350 count_kernel.set_arg(1, input_offset); in radix_sort_impl()
351 count_kernel.set_arg(2, static_cast<uint_>(count)); in radix_sort_impl()
352 count_kernel.set_arg(3, counts); in radix_sort_impl()
353 count_kernel.set_arg(4, offsets); in radix_sort_impl()
354 count_kernel.set_arg(5, block_size * sizeof(uint_), 0); in radix_sort_impl()
355 count_kernel.set_arg(6, i * k); in radix_sort_impl()
395 scan_kernel.set_arg(0, counts); in radix_sort_impl()
396 scan_kernel.set_arg(1, offsets); in radix_sort_impl()
397 scan_kernel.set_arg(2, block_count); in radix_sort_impl()
401 scatter_kernel.set_arg(0, *input_buffer); in radix_sort_impl()
402 scatter_kernel.set_arg(1, input_offset); in radix_sort_impl()
403 scatter_kernel.set_arg(2, static_cast<uint_>(count)); in radix_sort_impl()
404 scatter_kernel.set_arg(3, i * k); in radix_sort_impl()
405 scatter_kernel.set_arg(4, counts); in radix_sort_impl()
406 scatter_kernel.set_arg(5, offsets); in radix_sort_impl()
407 scatter_kernel.set_arg(6, *output_buffer); in radix_sort_impl()
408 scatter_kernel.set_arg(7, output_offset); in radix_sort_impl()
410 scatter_kernel.set_arg(8, *values_input_buffer); in radix_sort_impl()
411 scatter_kernel.set_arg(9, values_input_offset); in radix_sort_impl()
412 scatter_kernel.set_arg(10, *values_output_buffer); in radix_sort_impl()
413 scatter_kernel.set_arg(11, values_output_offset); in radix_sort_impl()