Graphite
Loading...
Searching...
No Matches
vector.hpp File Reference
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

template<typename T >
__global__ void graphite::ops::axpy_kernel (size_t n, T *z, const T a, const T *x, T *y)
 
template<typename T >
void graphite::ops::axpy_async (cudaStream_t stream, size_t n, T *z, const T a, const T *x, T *y)
 
template<typename T >
__global__ void graphite::ops::damping_kernel (size_t n, T *z, const T a, const T *diag, const T *x)
 
template<typename T >
void graphite::ops::damp_by_factor_async (cudaStream_t stream, size_t n, T *z, const T a, const T *diag, const T *x)
 
template<typename T >
__global__ void graphite::ops::clamp_kernel (size_t n, T min_val, T max_val, T *x)
 
template<typename T >
void graphite::ops::clamp_async (cudaStream_t stream, size_t n, T min_val, T max_val, T *x)
 
template<typename T >
__global__ void graphite::ops::rescale_vec_kernel (size_t n, T *out, const T scale, const T *x)
 
template<typename T >
void graphite::ops::rescale_vec_async (cudaStream_t stream, size_t n, T *out, const T scale, const T *x)