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