A Simple Example

#include<cub/cub.cuh>// Block-sorting CUDA kernel__global__voidBlockSortKernel(int *d_in,int *d_out){usingnamespacecub;// Specialize BlockRadixSort, BlockLoad, and BlockStore for 128 threads// owning 16 integer items eachtypedef BlockRadixSort<int,128,16>                     BlockRadixSort;typedef BlockLoad<int,128,16, BLOCK_LOAD_TRANSPOSE>   BlockLoad;typedef BlockStore<int,128,16, BLOCK_STORE_TRANSPOSE> BlockStore;// Allocate shared memory     __shared__union {typename BlockRadixSort::TempStorage  sort;typename BlockLoad::TempStorage       load;typename BlockStore::TempStorage      store;     } temp_storage;int block_offset = blockIdx.x * (128 *16);// OffsetT for this block's ment// Obtain a segment of 2048 consecutive keys that are blocked across threadsint thread_keys[16];BlockLoad(temp_storage.load).Load(d_in + block_offset, thread_keys);__syncthreads();// Collectively sort the keysBlockRadixSort(temp_storage.sort).Sort(thread_keys);__syncthreads();// Store the sorted segmentBlockStore(temp_storage.store).Store(d_out + block_offset, thread_keys);}

Supported Compilers

Releases

Development Process

# Clone Thrust and CUB from Github. CUB is located in Thrust's# `dependencies/cub` submodule.git clone --recursive https://github.com/NVIDIA/thrust.gitcd thrust# Create build directory:mkdir buildcd build# Configure -- use one of the following:cmake -DTHRUST_INCLUDE_CUB_CMAKE=ON ..   # Command line interface.ccmake -DTHRUST_INCLUDE_CUB_CMAKE=ON ..  # ncurses GUI (Linux only)cmake-gui  # Graphical UI, set source/build directories and options in the app# Build:cmake --build . -j <num jobs>   # invokes make (or ninja, etc)# Run tests and examples:ctest

Open Source License

Copyright (c) 2010-2011, Duane Merrill.  All rights reserved.Copyright (c) 2011-2018, NVIDIA CORPORATION.  All rights reserved.Redistribution and use in source and binary forms, with or withoutmodification, are permitted provided that the following conditions are met:   *  Redistributions of source code must retain the above copyright      notice, this list of conditions and the following disclaimer.   *  Redistributions in binary form must reproduce the above copyright      notice, this list of conditions and the following disclaimer in the      documentation and/or other materials provided with the distribution.   *  Neither the name of the NVIDIA CORPORATION nor the      names of its contributors may be used to endorse or promote products      derived from this software without specific prior written permission.THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" ANDANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIEDWARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE AREDISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANYDIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED ANDON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THISSOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.