cupy_rand.h 312 Bytes
Newer Older
root's avatar
root committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
#ifndef INCLUDE_GUARD_CUPY_CURAND_H
#define INCLUDE_GUARD_CUPY_CURAND_H

#if CUPY_USE_HIP

#include "hip/cupy_hiprand.h"

#elif !defined(CUPY_NO_CUDA)

#include <curand.h>

#else // #ifndef CUPY_NO_CUDA

#include "stub/cupy_curand.h"

#endif // #ifndef CUPY_NO_CUDA
#endif // #ifndef INCLUDE_GUARD_CUPY_CURAND_H