#include <af/defines.h>
#include <af/exception.h>
#include <cuda.h>
#include <cuda_runtime.h>
Go to the source code of this file.
|
AFAPI af_err | afcu_get_stream (cudaStream_t *stream, int id) |
| Get the stream for the CUDA device with id in ArrayFire context.
|
|
AFAPI af_err | afcu_get_native_id (int *nativeid, int id) |
| Get the native device id of the CUDA device with id in ArrayFire context.
|
|
AFAPI af_err | afcu_set_native_id (int nativeid) |
| Set the CUDA device with given native id as the active device for ArrayFire.
|
|
static cudaStream_t | getStream (int id) |
| Get the stream for the CUDA device with id in ArrayFire context.
|
|
static int | getNativeId (int id) |
| Get the native device id of the CUDA device with id in ArrayFire context.
|
|
static void | setNativeId (int nativeId) |
| Set the CUDA device with given native id as the active device for ArrayFire.
|
|