Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
d919aaeb
MegEngine
项目概览
MegEngine 天元
/
MegEngine
1 年多 前同步成功
通知
404
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看板
提交
d919aaeb
编写于
2月 11, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
test(imperative): reopen special interpolate test and sync when test rng
GitOrigin-RevId: e3d03b4d1d981ebf7afc8be6fe79fb5928f81aac
上级
ca2deebc
变更
2
隐藏空白更改
内联
并排
Showing
2 changed file
with
4 addition
and
4 deletion
+4
-4
imperative/python/test/unit/functional/test_functional.py
imperative/python/test/unit/functional/test_functional.py
+1
-1
imperative/python/test/unit/random/test_rng.py
imperative/python/test/unit/random/test_rng.py
+3
-3
未找到文件。
imperative/python/test/unit/functional/test_functional.py
浏览文件 @
d919aaeb
...
...
@@ -507,7 +507,7 @@ def test_interpolate_fastpath():
[(
1
,
1
,
10
,
10
),
(
5
,
5
)],
[(
1
,
3
,
10
,
10
),
(
20
,
20
)],
[(
10
,
1
,
10
,
10
),
(
1
,
1
)],
# [(10, 10, 1, 1), (10, 10)], # FIXME, it causes random CI failure
[(
10
,
10
,
1
,
1
),
(
10
,
10
)],
]
for
inp_shape
,
target_shape
in
test_cases
:
x
=
tensor
(
np
.
random
.
randn
(
*
inp_shape
),
dtype
=
np
.
float32
)
...
...
imperative/python/test/unit/random/test_rng.py
浏览文件 @
d919aaeb
...
...
@@ -39,6 +39,8 @@ from megengine.random import uniform
get_device_count
(
"xpu"
)
<=
2
,
reason
=
"xpu counts need > 2"
,
)
def
test_gaussian_op
():
# FIXME: remove this sync
mge
.
core
.
set_option
(
"async_level"
,
0
)
set_global_seed
(
1024
)
shape
=
(
8
,
...
...
@@ -210,12 +212,9 @@ def test_permutation_op():
assert
str
(
output
.
device
)
==
str
(
cn
)
assert
output
.
dtype
==
dtype
# FIXME: remove this sync
mge
.
core
.
set_option
(
"async_level"
,
0
)
test_permutation_op_dtype
(
np
.
float32
)
test_permutation_op_dtype
(
np
.
int32
)
test_permutation_op_dtype
(
np
.
int16
)
mge
.
core
.
set_option
(
"async_level"
,
2
)
@
pytest
.
mark
.
skipif
(
...
...
@@ -517,3 +516,4 @@ def test_rng_empty_tensor(is_symbolic):
np
.
testing
.
assert_equal
(
out
.
numpy
().
shape
,
(
0
,))
if
is_symbolic
is
None
:
break
mge
.
core
.
set_option
(
"async_level"
,
2
)
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录