SyTen
cublas_support.h File Reference

Support functions for using CuBLAS. More...

#include "inc/util/cuda_support.h"
#include <cublas_v2.h>
#include "inc/util/cuda_span.h"
+ Include dependency graph for cublas_support.h:
+ This graph shows which files directly or indirectly include this file:

Classes

struct  syten::Cuda::HandleBox
 RAII wrapper around a handle, putting it out of use on destruction. More...
 

Namespaces

namespace  syten
 Syten namespace.
 
namespace  syten::Cuda
 Support functions (memory allocation etc.) for CUDA-based GPUs.
 

Functions

HandleBox syten::Cuda::get_handle (CudaStream const &str)
 Returns a cublas handle associated to the device device and switches the current CUDA device to this device. More...
 
cublasStatus_t syten::Cuda::handle_error (cublasStatus_t err)
 Checks the error code and asserts if it is not CUBLAS_STATUS_SUCCESS. More...
 
void syten::Cuda::initialise_handles ()
 Initialises the cublas handles for the currently allowed devices. More...
 

Detailed Description

Support functions for using CuBLAS.