Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
88486570
MegEngine
项目概览
MegEngine 天元
/
MegEngine
12 个月 前同步成功
通知
393
Star
4704
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看板
体验新版 GitCode,发现更多精彩内容 >>
提交
88486570
编写于
1月 28, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
test(trace): test subtensor on unknown shape
GitOrigin-RevId: 1b5cfa4e0ac098b54e7f4544433310d9ae90c99e
上级
c34a75d0
变更
1
隐藏空白更改
内联
并排
Showing
1 changed file
with
27 addition
and
0 deletion
+27
-0
imperative/python/test/unit/core/test_indexing_op.py
imperative/python/test/unit/core/test_indexing_op.py
+27
-0
未找到文件。
imperative/python/test/unit/core/test_indexing_op.py
浏览文件 @
88486570
...
...
@@ -7,6 +7,8 @@
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import
collections
import
platform
from
tempfile
import
NamedTemporaryFile
import
numpy
as
np
import
pytest
...
...
@@ -16,6 +18,8 @@ import megengine
import
megengine.core.tensor.megbrain_graph
as
G
import
megengine.functional
as
F
import
megengine.jit
as
jit
import
megengine.random
as
rand
import
megengine.utils.comp_graph_tools
as
cgtools
from
megengine.core._imperative_rt.core2
import
apply
from
megengine.core._trace_option
import
use_symbolic_shape
from
megengine.core.ops
import
builtin
...
...
@@ -724,3 +728,26 @@ def test_nd_int_indexing(symbolic):
np
.
testing
.
assert_equal
(
out
.
numpy
(),
npy_out
)
run_test
([
inp
,
idx
],
lambda
inp
,
idx
:
inp
[
idx
])
@
pytest
.
mark
.
skipif
(
platform
.
system
()
==
"Windows"
,
reason
=
"windows temp file issue, fixme later"
)
def
test_subtensor_when_shape_invalid
():
@
jit
.
trace
(
symbolic
=
True
,
capture_as_const
=
True
)
def
fun
(
inp
):
shape
=
inp
.
shape
H
=
shape
[
-
1
]
NH
=
H
*
8
+
4
arr
=
F
.
arange
(
4
,
NH
,
8
)
arr_shape
=
arr
.
shape
return
arr_shape
[
0
]
inp
=
rand
.
uniform
(
size
=
[
1
,
3
,
224
,
224
])
fun
(
inp
)
with
NamedTemporaryFile
()
as
f
:
fun
.
dump
(
f
.
name
,
arg_names
=
[
"data"
],
optimize_for_inference
=
True
)
inp
=
rand
.
uniform
(
size
=
[
1
,
3
,
512
,
512
])
net
=
cgtools
.
GraphInference
(
f
.
name
)
net
.
run
(
inp_dict
=
{
"data"
:
inp
})
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录