cuda.h File Reference

(2022-02-24 14:07:56 -0500, (build 8c2561559))

#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...