yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix primitives function prototype

Browse Source
This commit is contained in:
Sylvain Jeaugey 2016-10-13 10:32:42 -07:00
parent bf7d1514f7
commit b2781d0501
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
src/primitives.h
View File

@ -177,29 +177,29 @@ class Primitives {
template <typename... SYNC_Ts>
static __device__ __forceinline__ void
Copy(const T* src, T* dst,
int len, int step, SYNC_Ts... flags) {
GenericOp(src, nullptr, dst, nullptr, len, step, flags...);
int len, int maxOffset, int step, SYNC_Ts... flags) {
GenericOp(src, nullptr, dst, nullptr, len, maxOffset, step, flags...);
}
template <typename... SYNC_Ts>
static __device__ __forceinline__ void
DoubleCopy(const T* src, T* dst1, T* dst2,
int len, int step, SYNC_Ts... flags) {
GenericOp(src, nullptr, dst1, dst2, len, step, flags...);
int len, int maxOffset, int step, SYNC_Ts... flags) {
GenericOp(src, nullptr, dst1, dst2, len, maxOffset, step, flags...);
}
template <typename... SYNC_Ts>
static __device__ __forceinline__ void
Reduce(const T* src1, const T* src2, T* dst,
int len, int step, SYNC_Ts... flags) {
GenericOp(src1, src2, dst, nullptr, len, step, flags...);
int len, int maxOffset, int step, SYNC_Ts... flags) {
GenericOp(src1, src2, dst, nullptr, len, maxOffset, step, flags...);
}
template <typename... SYNC_Ts>
static __device__ __forceinline__ void
ReduceCopy(const T* src1, const T* src2, T* dst1, T* dst2,
int len, int step, SYNC_Ts... flags) {
GenericOp(src1, src2, dst1, dst2, len, step, flags...);
int len, int maxOffset, int step, SYNC_Ts... flags) {
GenericOp(src1, src2, dst1, dst2, len, maxOffset, step, flags...);
}
};