Home
last modified time | relevance | path

Searched refs:d_segment_offsets (Results 1 – 3 of 3) sorted by relevance

/dports/misc/xgboost/xgboost-1.5.1/cub/test/
H A Dtest_device_reduce.cu342 … d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, reduction_op, identity, in Dispatch()
375 d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
408 d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
441 d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
474 d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
507 d_in, d_out, max_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
651 (void)d_segment_offsets; in CnpDispatchKernel()
1006 if (d_segment_offsets) CubDebugExit(g_allocator.DeviceFree(d_segment_offsets)); in TestProblem()
1057 OffsetT *d_segment_offsets = NULL; in TestByBackend() local
1105 d_segment_offsets, in TestByBackend()
[all …]
H A Dtest_device_radix_sort.cu246 const int *d_segment_offsets, in Dispatch() argument
255 num_items, num_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
278 const int *d_segment_offsets, in Dispatch() argument
290 num_items, num_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
318 const int *d_segment_offsets, in Dispatch() argument
327 num_items, num_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
350 const int *d_segment_offsets, in Dispatch() argument
362 num_items, num_segments, d_segment_offsets, d_segment_offsets + 1, in Dispatch()
486 const int *d_segment_offsets, in CnpDispatchKernel() argument
502 (void)d_segment_offsets; in CnpDispatchKernel()
[all …]
/dports/misc/xgboost/xgboost-1.5.1/cub/experimental/defunct/
H A Dtest_device_seg_reduce.cu1535 …SegmentOffsetIterator d_segment_offsets, ///< [in] A sequence of (\… in Dispatch()
1623 SegmentOffsetIterator d_segment_end_offsets = d_segment_offsets + 1; in Dispatch()
1741 d_segment_offsets, in Dispatch()
1813 …SegmentOffsetIterator d_segment_offsets, ///< [in] A sequence of (\p num_se… in Reduce()
1837 d_segment_offsets, in Reduce()
1873 …SegmentOffsetIterator d_segment_offsets, ///< [in] A sequence of (\p num_se… in Sum()
1901 d_segment_offsets, in Sum()
2045 OffsetT *d_segment_offsets = NULL; in Test() local
2048 …CubDebugExit(g_allocator.DeviceAllocate((void**)&d_segment_offsets, sizeof(OffsetT) * (num_segment… in Test()
2051 …CubDebugExit(cudaMemcpy(d_segment_offsets, h_segment_offsets, sizeof(OffsetT) * (num_segments + 1)… in Test()
[all …]