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
杨迪
NiuTrans.Tensor
Commits
16ab02c5
Commit
16ab02c5
authored
5 years ago
by
xiaotong
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
add USE_CUDA to protect the CPU execution
parent
7180a4a3
隐藏空白字符变更
内嵌
并排
正在显示
1 个修改的文件
包含
8 行增加
和
0 行删除
+8
-0
source/tensor/core/getandset/SetData.cpp
+8
-0
没有找到文件。
source/tensor/core/getandset/SetData.cpp
查看文件 @
16ab02c5
...
...
@@ -250,7 +250,11 @@ void _SetDataFixedCond(XTensor * tensor, XTensor * condition, DTYPE p)
}
}
else
{
#ifdef USE_CUDA
_CudaSetDataFixedCondFloat
(
tensor
,
condition
,
p
);
#else
ShowNTErrors
(
"Please specify USE_CUDA and recompile the code"
);
#endif
}
}
else
{
...
...
@@ -283,7 +287,11 @@ void _SetDataFixedCondInt(XTensor * tensor, XTensor * condition, int p)
}
}
else
{
#ifdef USE_CUDA
_CudaSetDataFixedCondInt
(
tensor
,
condition
,
p
);
#else
ShowNTErrors
(
"Please specify USE_CUDA and recompile the code"
);
#endif
}
}
else
{
...
...
This diff is collapsed.
Click to expand it.
编写
预览
Markdown
格式
0%
重试
或
添加新文件
添加附件
取消
您添加了
0
人
到此讨论。请谨慎行事。
请先完成此评论的编辑!
取消
请
注册
或者
登录
后发表评论