Skip to content
项目
群组
代码片段
帮助
当前项目
正在载入...
登录 / 注册
切换导航面板
N
NiuTrans.Tensor
概览
Overview
Details
Activity
Cycle Analytics
版本库
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
问题
0
Issues
0
列表
Board
标记
里程碑
合并请求
0
Merge Requests
0
CI / CD
CI / CD
流水线
作业
日程表
图表
维基
Wiki
代码片段
Snippets
成员
Collapse sidebar
Close sidebar
活动
图像
聊天
创建新问题
作业
提交
Issue Boards
Open sidebar
Emmay
NiuTrans.Tensor
Commits
f74b1c17
Commit
f74b1c17
authored
6 years ago
by
xiaotong
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
better way of memory copy
parent
ae990819
隐藏空白字符变更
内嵌
并排
正在显示
3 个修改的文件
包含
5 行增加
和
1 行删除
+5
-1
.gitignore
+3
-0
source/tensor/core/arithmetic/Sum.cu
+1
-0
source/tensor/core/utilities/FlushToMem.cu
+1
-1
没有找到文件。
.gitignore
查看文件 @
f74b1c17
NiuTrans.Tensor.vcxproj
NiuTrans.Tensor.vcxproj.filters
x64/
vc140.pdb
NiuTrans.Tensor.vcxproj.user
NiuTrans.Tensor.aps
This diff is collapsed.
Click to expand it.
source/tensor/core/arithmetic/Sum.cu
查看文件 @
f74b1c17
...
...
@@ -20,6 +20,7 @@
*/
#include "../../XDevice.h"
#include "../../XUtility.h"
#include "Sum.cuh"
namespace nts { // namespace nts(NiuTrans.Tensor)
...
...
This diff is collapsed.
Click to expand it.
source/tensor/core/utilities/FlushToMem.cu
查看文件 @
f74b1c17
...
...
@@ -117,7 +117,7 @@ void CudaGPUToCPUFlush(XTensor * tensor)
else {
tensor->dataHost = new char[tensor->unitNum * tensor->unitSize];
if (tensor->data != NULL)
cudaMemcpy(tensor->dataHost, tensor->data, tensor->unitNum * tensor->unitSize, cudaMemcpyDeviceToHost
);
XMemCopy(tensor->dataHost, -1, tensor->data, tensor->devID, tensor->unitNum * tensor->unitSize
);
else
memset(tensor->dataHost, 0, tensor->unitNum * tensor->unitSize);
}
...
...
This diff is collapsed.
Click to expand it.
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论