forked from arrayfire/arrayfire
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathcuda.cpp
More file actions
42 lines (36 loc) · 1.3 KB
/
cuda.cpp
File metadata and controls
42 lines (36 loc) · 1.3 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/*******************************************************
* Copyright (c) 2019, ArrayFire
* All rights reserved.
*
* This file is distributed under 3-clause BSD license.
* The complete license agreement can be obtained at:
* http://arrayfire.com/licenses/BSD-3-Clause
********************************************************/
#include <af/backend.h>
#include "symbol_manager.hpp"
#define AF_DEFINE_CUDA_TYPES
#include <af/cuda.h>
af_err afcu_get_stream(cudaStream_t* stream, int id) {
af_backend backend;
af_get_active_backend(&backend);
if (backend == AF_BACKEND_CUDA) { CALL(afcu_get_stream, stream, id); }
return AF_ERR_NOT_SUPPORTED;
}
af_err afcu_get_native_id(int* nativeid, int id) {
af_backend backend;
af_get_active_backend(&backend);
if (backend == AF_BACKEND_CUDA) { CALL(afcu_get_native_id, nativeid, id); }
return AF_ERR_NOT_SUPPORTED;
}
af_err afcu_set_native_id(int nativeid) {
af_backend backend;
af_get_active_backend(&backend);
if (backend == AF_BACKEND_CUDA) { CALL(afcu_set_native_id, nativeid); }
return AF_ERR_NOT_SUPPORTED;
}
af_err afcu_cublasSetMathMode(cublasMath_t mode) {
af_backend backend;
af_get_active_backend(&backend);
if (backend == AF_BACKEND_CUDA) { CALL(afcu_cublasSetMathMode, mode); }
return AF_ERR_NOT_SUPPORTED;
}