cuda.h File Reference

#include <af/defines.h>
#include <af/exception.h>

Go to the source code of this file.

Namespaces
namespace	afcu

Functions
AFAPI af_err	afcu_get_stream (cudaStream_t *stream, int id)
	This file contain functions that apply only to the CUDA backend. More...
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. More...
AFAPI af_err	afcu_set_native_id (int nativeid)
	Set the CUDA device with given native id as the active device for ArrayFire. More...
AFAPI af_err	afcu_cublasSetMathMode (cublasMath_t mode)
	Sets the cuBLAS math mode for the internal handle. More...
static cudaStream_t	getStream (int id)
	Get the stream for the CUDA device with id in ArrayFire context. More...
static int	getNativeId (int id)
	Get the native device id of the CUDA device with id in ArrayFire context. More...
static void	setNativeId (int nativeId)
	Set the CUDA device with given native id as the active device for ArrayFire. More...