Tinker9 70bd052 (Thu Nov 9 12:11:35 2023 -0800)
Loading...
Searching...
No Matches
parallelacc.h
1#pragma once
2#include <cstddef>
3
4namespace tinker {
5template <class T>
6T reduceSum_acc(const T* gpu_a, size_t nelem, int queue);
7
8template <class HT, size_t HN, class DPTR>
9void reduceSum2_acc(HT (&h_ans)[HN], DPTR v, size_t nelem, int queue);
10
11template <class T>
12void reduceSumOnDevice_acc(T*, const T*, size_t, int);
13
14template <class HT, size_t HN, class DPTR>
15void reduceSum2OnDevice_acc(HT (&)[HN], DPTR, size_t, int);
16
17template <class T>
18T dotProd_acc(const T* a, const T* b, size_t nelem, int queue);
19
20template <class T>
21void dotProd_acc(T* ans, const T* a, const T* b, size_t nelem, int queue);
22
23template <class T>
24void scaleArray_acc(T* dst, T scal, size_t nelem, int queue);
25}
Definition: testrt.h:9
void reduceSum2_acc(HT(&h_ans)[HN], DPTR v, size_t nelem, int queue)
void reduceSumOnDevice_acc(T *, const T *, size_t, int)
T dotProd_acc(const T *a, const T *b, size_t nelem, int queue)
void scaleArray_acc(T *dst, T scal, size_t nelem, int queue)
T reduceSum_acc(const T *gpu_a, size_t nelem, int queue)
void reduceSum2OnDevice_acc(HT(&)[HN], DPTR, size_t, int)