"vscode:/vscode.git/clone" did not exist on "d34d88e41744bdcc36a425299af85ce762f3d30e"
dgl_cub.cuh 423 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
/*!
 *  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_

// include cub in a safe manner
#define CUB_NS_PREFIX namespace dgl {
#define CUB_NS_POSTFIX }
13
#define CUB_NS_QUALIFIER ::dgl::cub
14
#include "cub/cub.cuh"
15
#undef CUB_NS_QUALIFIER
16
17
18
19
#undef CUB_NS_POSTFIX
#undef CUB_NS_PREFIX

#endif