Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
539aac44
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
399
Star
4705
Fork
582
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
MegEngine
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
提交
539aac44
编写于
8月 28, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(dnn/cuda): fix sass loader
GitOrigin-RevId: f7dec65a2a7eaa24382b6311de3c299cb868bdb8
上级
39d43283
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
16 addition
and
15 deletion
+16
-15
src/opr/test/dnn/convolution.cpp
src/opr/test/dnn/convolution.cpp
+16
-15
未找到文件。
src/opr/test/dnn/convolution.cpp
浏览文件 @
539aac44
...
...
@@ -2033,7 +2033,7 @@ TEST(TestOprDNN, HeuristicReproducible) {
#if MGB_CUDA
TEST
(
TestOprDNN
,
ConvolutionMultiCompNode
)
{
REQUIRE_GPU
(
2
);
REQUIRE_GPU
(
1
);
auto
cn0
=
CompNode
::
load
(
"gpu0:0"
),
cn1
=
CompNode
::
load
(
"gpu0:1"
);
cn0
.
activate
();
auto
&&
prop
=
CompNodeEnv
::
from_comp_node
(
cn0
).
cuda_env
().
device_prop
;
...
...
@@ -2106,21 +2106,22 @@ TEST(TestOprDNN, ConvolutionMultiCompNode) {
auto
func0
=
graph0
->
compile
({
make_callback_copy
(
y0
,
host_y0
)});
auto
func1
=
graph1
->
compile
({
make_callback_copy
(
y1
,
host_y1
)});
auto
worker
=
[
&
func0
,
&
func1
](
int
wid
)
{
static
int
cons
t
iter_num
=
1000
;
if
(
wid
==
0
)
{
for
(
int
i
=
0
;
i
<
iter_num
;
++
i
)
func0
->
execute
();
}
else
{
for
(
int
i
=
0
;
i
<
iter_num
;
++
i
)
func1
->
execute
();
}
};
std
::
thread
worker0
(
worker
,
0
);
std
::
thread
worker1
(
worker
,
1
);
worker0
.
join
();
worker1
.
join
();
auto
worker
=
[
&
func0
,
&
func1
](
int
wid
)
{
static
const
in
t
iter_num
=
1000
;
if
(
wid
==
0
)
{
for
(
int
i
=
0
;
i
<
iter_num
;
++
i
)
func0
->
execute
();
}
else
{
for
(
int
i
=
0
;
i
<
iter_num
;
++
i
)
func1
->
execute
();
}
};
std
::
thread
worker0
(
worker
,
0
);
std
::
thread
worker1
(
worker
,
1
);
worker0
.
join
();
worker1
.
join
();
}
#endif
// vim: syntax=cpp.doxygen foldmethod=marker foldmarker=f{{{,f}}}
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录