cupy_complex.h 358 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_COMPLEX_H
#define INCLUDE_GUARD_CUPY_COMPLEX_H

#ifdef CUPY_USE_HIP

#include "hip/cupy_cuComplex.h"

#elif !defined(CUPY_NO_CUDA)

#include <cuComplex.h>

#else // #if !defined(CUPY_NO_CUDA) || !defined(CUPY_USE_HIP)

#include "stub/cupy_cuComplex.h"

#endif // #ifndef CUPY_NO_CUDA
#endif // #ifndef INCLUDE_GUARD_CUPY_COMPLEX_H