"examples/pytorch/vscode:/vscode.git/clone" did not exist on "bb542066202b34fe0ea72ec0644ec1735334db38"
dgl_cub.cuh 424 Bytes
Newer Older
1
2
3
/*!
 *  Copyright (c) 2021 by Contributors
 * \file cuda_common.h
4
 * \brief Wrapper to place cub in dgl namespace.
5
6
7
8
9
 */

#ifndef DGL_ARRAY_CUDA_DGL_CUB_CUH_
#define DGL_ARRAY_CUDA_DGL_CUB_CUH_

10
11
12
13
14
// 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

15
16
#include "cub/cub.cuh"

17
#endif  // DGL_ARRAY_CUDA_DGL_CUB_CUH_