mult_kernel_impl.h File Reference
#include <TiledArray/external/cuda.h>
#include <thrust/device_vector.h>
#include <thrust/execution_policy.h>
Include dependency graph for mult_kernel_impl.h:

Namespaces

 TiledArray
 

Functions

template<typename T >
void TiledArray::mult_to_cuda_kernel_impl (T *result, const T *arg, std::size_t n, cudaStream_t stream, int device_id)
 result[i] = result[i] * arg[i] More...
 
template<typename T >
void TiledArray::mult_cuda_kernel_impl (T *result, const T *arg1, const T *arg2, std::size_t n, cudaStream_t stream, int device_id)
 result[i] = arg1[i] * arg2[i] More...