forked from Theano/libgpuarray
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathextension.h
More file actions
33 lines (26 loc) · 686 Bytes
/
extension.h
File metadata and controls
33 lines (26 loc) · 686 Bytes
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
#ifndef GPUARRAY_EXTENSIONS_H
#define GPUARRAY_EXTENSIONS_H
/** \file extension.h
* \brief Extensions access.
*/
#include <gpuarray/config.h>
#ifdef __cplusplus
extern "C" {
#endif
#ifdef CONFUSE_EMACS
}
#endif
/* Keep in sync with the flags in private_cuda.h */
#define GPUARRAY_CUDA_CTX_NOFREE 0x10000000 /* DONTFREE */
#define GPUARRAY_CUDA_WAIT_READ 0x10000 /* CUDA_WAIT_READ */
#define GPUARRAY_CUDA_WAIT_WRITE 0x20000 /* CUDA_WAIT_WRITE */
/**
* Obtain a function pointer for an extension.
*
* \returns A function pointer or NULL if the extension was not found.
*/
GPUARRAY_PUBLIC void * gpuarray_get_extension(const char *name);
#ifdef __cplusplus
}
#endif
#endif