cupy_backend.h 308 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_CUDA_H
#define INCLUDE_GUARD_CUPY_CUDA_H

#if CUPY_USE_HIP

#include "hip/cupy_hip.h"

#elif !defined(CUPY_NO_CUDA)

#include "cuda/cupy_cuda.h"

#else // #ifndef CUPY_NO_CUDA

#include "stub/cupy_cuda.h"

#endif // #ifndef CUPY_NO_CUDA
#endif // #ifndef INCLUDE_GUARD_CUPY_CUDA_H