Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
dfc3935d
MegEngine
项目概览
MegEngine 天元
/
MegEngine
接近 2 年 前同步成功
通知
414
Star
4708
Fork
583
代码
文件
提交
分支
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看板
提交
dfc3935d
编写于
10月 17, 2020
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
fix(pytest/windows/impertive): fix windows wrong test skip
GitOrigin-RevId: ca0f04cfe7b069af1358cf06fb60eaa09eadb1c9
上级
786399c5
变更
3
隐藏空白更改
内联
并排
Showing
3 changed file
with
5 addition
and
26 deletion
+5
-26
imperative/python/test/unit/distributed/test_distributed.py
imperative/python/test/unit/distributed/test_distributed.py
+4
-4
imperative/python/test/unit/functional/test_tensor.py
imperative/python/test/unit/functional/test_tensor.py
+0
-9
imperative/python/test/unit/module/test_batchnorm.py
imperative/python/test/unit/module/test_batchnorm.py
+1
-13
未找到文件。
imperative/python/test/unit/distributed/test_distributed.py
浏览文件 @
dfc3935d
...
@@ -41,7 +41,7 @@ def _assert_q_val(q, val):
...
@@ -41,7 +41,7 @@ def _assert_q_val(q, val):
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"
do not imp GPU mode at Windows now
"
platform
.
system
()
==
"Windows"
,
reason
=
"
windows disable MGB_ENABLE_OPR_MM
"
)
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
...
@@ -85,7 +85,7 @@ def test_init_process_group():
...
@@ -85,7 +85,7 @@ def test_init_process_group():
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"
do not imp GPU mode at Windows now
"
platform
.
system
()
==
"Windows"
,
reason
=
"
windows disable MGB_ENABLE_OPR_MM
"
)
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
...
@@ -119,7 +119,7 @@ def test_new_group():
...
@@ -119,7 +119,7 @@ def test_new_group():
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"
do not imp GPU mode at Windows now
"
platform
.
system
()
==
"Windows"
,
reason
=
"
windows disable MGB_ENABLE_OPR_MM
"
)
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
...
@@ -155,7 +155,7 @@ def test_group_barrier():
...
@@ -155,7 +155,7 @@ def test_group_barrier():
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"
do not imp GPU mode at Windows now
"
platform
.
system
()
==
"Windows"
,
reason
=
"
windows disable MGB_ENABLE_OPR_MM
"
)
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
...
...
imperative/python/test/unit/functional/test_tensor.py
浏览文件 @
dfc3935d
...
@@ -338,9 +338,6 @@ def copy_test(dst, src):
...
@@ -338,9 +338,6 @@ def copy_test(dst, src):
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
==
0
,
reason
=
"CUDA is disabled"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
==
0
,
reason
=
"CUDA is disabled"
)
def
test_copy_h2d
():
def
test_copy_h2d
():
copy_test
(
"cpu0"
,
"gpu0"
)
copy_test
(
"cpu0"
,
"gpu0"
)
...
@@ -349,9 +346,6 @@ def test_copy_h2d():
...
@@ -349,9 +346,6 @@ def test_copy_h2d():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
==
0
,
reason
=
"CUDA is disabled"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
==
0
,
reason
=
"CUDA is disabled"
)
def
test_copy_d2h
():
def
test_copy_d2h
():
copy_test
(
"gpu0"
,
"cpu0"
)
copy_test
(
"gpu0"
,
"cpu0"
)
...
@@ -360,9 +354,6 @@ def test_copy_d2h():
...
@@ -360,9 +354,6 @@ def test_copy_d2h():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
@
pytest
.
mark
.
skipif
(
get_device_count_by_fork
(
"gpu"
)
<
2
,
reason
=
"need more gpu device"
)
def
test_copy_d2d
():
def
test_copy_d2d
():
copy_test
(
"gpu0"
,
"gpu1"
)
copy_test
(
"gpu0"
,
"gpu1"
)
...
...
imperative/python/test/unit/module/test_batchnorm.py
浏览文件 @
dfc3935d
...
@@ -26,7 +26,7 @@ _assert_allclose = functools.partial(np.testing.assert_allclose, atol=5e-6, rtol
...
@@ -26,7 +26,7 @@ _assert_allclose = functools.partial(np.testing.assert_allclose, atol=5e-6, rtol
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"
do not imp GPU mode at Windows now
"
platform
.
system
()
==
"Windows"
,
reason
=
"
windows disable MGB_ENABLE_OPR_MM
"
)
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
def
test_syncbn
():
def
test_syncbn
():
...
@@ -143,9 +143,6 @@ def test_batchnorm():
...
@@ -143,9 +143,6 @@ def test_batchnorm():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
def
test_syncbn1d
():
def
test_syncbn1d
():
nr_chan
=
8
nr_chan
=
8
...
@@ -237,9 +234,6 @@ def test_batchnorm2d():
...
@@ -237,9 +234,6 @@ def test_batchnorm2d():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
def
test_syncbn2d
():
def
test_syncbn2d
():
nr_chan
=
8
nr_chan
=
8
...
@@ -311,9 +305,6 @@ def test_batchnorm_no_stats():
...
@@ -311,9 +305,6 @@ def test_batchnorm_no_stats():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
def
test_syncbn_no_stats
():
def
test_syncbn_no_stats
():
nr_chan
=
8
nr_chan
=
8
...
@@ -363,9 +354,6 @@ def test_batchnorm2d_no_stats():
...
@@ -363,9 +354,6 @@ def test_batchnorm2d_no_stats():
@
pytest
.
mark
.
skipif
(
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
platform
.
system
()
==
"Darwin"
,
reason
=
"do not imp GPU mode at macos now"
)
)
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"do not imp GPU mode at Windows now"
)
@
pytest
.
mark
.
isolated_distributed
@
pytest
.
mark
.
isolated_distributed
def
test_syncbn2d_no_stats
():
def
test_syncbn2d_no_stats
():
nr_chan
=
8
nr_chan
=
8
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录