AI/rocm-systems
2
0 Fork 0
程式碼 問題 合併請求 Actions 套件 專案 版本發布 Wiki 動態

Fix primitives function prototype

瀏覽代碼
此提交包含在：
Sylvain Jeaugey committed 2016-10-13 10:32:42 -07:00
1 父節點 bf7d1514f7
當前提交 b2781d0501
1 file changed
+8 -8
下載 Patch 檔 下載差異檔 展開所有檔案 折疊所有檔案
src/primitives.h
+8 -8
查看文件
@@ -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...);
}
};