cupy_sparse.h 439 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
18
19
20
21
#ifndef INCLUDE_GUARD_CUPY_CUSPARSE_H
#define INCLUDE_GUARD_CUPY_CUSPARSE_H

#if !defined(CUPY_NO_CUDA) && !defined(CUPY_USE_HIP)

#include "cuda/cupy_cusparse.h"

#elif defined(CUPY_USE_HIP)

#include "hip/cupy_hip_common.h"
#include "hip/cupy_hipsparse.h"

#else

#include "stub/cupy_cuda_common.h"
#include "stub/cupy_cusparse.h"

#endif  // #if defined(CUPY_NO_CUDA) || defined(CUPY_USE_HIP)


#endif  // INCLUDE_GUARD_CUPY_CUSPARSE_H