dgl_cub.cuh

/*!
 *  Copyright (c) 2021 by Contributors
 * @file cuda_common.h
 * @brief Wrapper to place cub in dgl namespace.
 */

#ifndef DGL_ARRAY_CUDA_DGL_CUB_CUH_
#define DGL_ARRAY_CUDA_DGL_CUB_CUH_

// This should be defined in CMakeLists.txt
#ifndef THRUST_CUB_WRAPPED_NAMESPACE
static_assert(false, "THRUST_CUB_WRAPPED_NAMESPACE must be defined for DGL.");
#endif

#include "cub/cub.cuh"

#endif  // DGL_ARRAY_CUDA_DGL_CUB_CUH_