mirror of https://github.com/hpcaitech/ColossalAI
13 lines
296 B
Plaintext
13 lines
296 B
Plaintext
|
// copied from https://github.com/dmlc/dgl/pull/2758
|
||
|
|
#ifndef DGL_ARRAY_CUDA_DGL_CUB_CUH_
|
||
|
|
#define DGL_ARRAY_CUDA_DGL_CUB_CUH_
|
||
|
|
|
||
|
|
#define CUB_NS_PREFIX namespace ls {
|
||
|
|
#define CUB_NS_POSTFIX }
|
||
|
|
#include "cub/cub.cuh"
|
||
|
|
#include "cub/util_allocator.cuh"
|
||
|
|
#undef CUB_NS_POSTFIX
|
||
|
|
#undef CUB_NS_PREFIX
|
||
|
|
|
||
|
|
#endif
|