cuda.h

Go to the documentation of this file.

/*******************************************************
 * Copyright (c) 2014, 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
 ********************************************************/

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


#ifdef __NVCC__
#include <cuda.h>
#include <cuda_runtime.h>
#include <cublas_v2.h>
#else
#ifdef AF_DEFINE_CUDA_TYPES
typedef struct CUstream_st *cudaStream_t;

/*Enum for default math mode/tensor operation*/
typedef enum {
  CUBLAS_DEFAULT_MATH = 0,
  CUBLAS_TENSOR_OP_MATH = 1
} cublasMath_t;
#endif
#endif

#ifdef __cplusplus
extern "C" {
#endif


#if AF_API_VERSION >= 31

AFAPI af_err afcu_get_stream(cudaStream_t* stream, int id);
#endif

#if AF_API_VERSION >= 31

AFAPI af_err afcu_get_native_id(int* nativeid, int id);
#endif

#if AF_API_VERSION >= 32

AFAPI af_err afcu_set_native_id(int nativeid);
#endif

#if AF_API_VERSION >= 37

AFAPI af_err afcu_cublasSetMathMode(cublasMath_t mode);
#endif

#ifdef __cplusplus
}
#endif

#ifdef __cplusplus

namespace afcu
{

#if AF_API_VERSION >= 31

static inline cudaStream_t getStream(int id)
{
    cudaStream_t retVal;
    af_err err = afcu_get_stream(&retVal, id);
    if (err!=AF_SUCCESS)
        throw af::exception("Failed to get CUDA stream from ArrayFire");
    return retVal;
}
#endif

#if AF_API_VERSION >= 31

static inline int getNativeId(int id)
{
    int retVal;
    af_err err = afcu_get_native_id(&retVal, id);
    if (err!=AF_SUCCESS)
        throw af::exception("Failed to get CUDA device native id from ArrayFire");
    return retVal;
}
#endif

#if AF_API_VERSION >= 32

static inline void setNativeId(int nativeId)
{
    af_err err = afcu_set_native_id(nativeId);
    if (err!=AF_SUCCESS)
        throw af::exception("Failed to change active CUDA device to the device with given native id");
}
#endif

}
#endif