Searched refs:col_scan (Results 1 – 2 of 2) sorted by relevance
66 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in concat_variable_kernel() local81 smem_col_scan[i] = col_scan[i]; in concat_variable_kernel()86 col_scan = smem_col_scan; in concat_variable_kernel()93 cuda_helper::upper_bound<IntType>(col_scan, num_inputs, gidx) - 1; in concat_variable_kernel()95 IntType curr_offset = col_scan[segment]; in concat_variable_kernel()99 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in concat_variable_kernel()
121 IntType* col_scan = GetGpuDeviceArrayOnDevice(&output_scan); in split_v_kernel() local136 smem_col_scan[i] = col_scan[i]; in split_v_kernel()141 col_scan = smem_col_scan; in split_v_kernel()148 cuda_helper::upper_bound<IntType>(col_scan, num_outputs, gidx) - 1; in split_v_kernel()150 IntType curr_offset = col_scan[segment]; in split_v_kernel()154 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in split_v_kernel()