Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
MegEngine 天元
MegEngine
提交
8dd69e87
MegEngine
项目概览
MegEngine 天元
/
MegEngine
大约 1 年 前同步成功
通知
394
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,发现更多精彩内容 >>
提交
8dd69e87
编写于
12月 11, 2022
作者:
M
Megvii Engine Team
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
perf(imperative): speed up the hot code of reshape
GitOrigin-RevId: 97cc37198240b284b106f9614280801351853ce2
上级
022dbea8
变更
1
显示空白变更内容
内联
并排
Showing
1 changed file
with
48 addition
and
0 deletion
+48
-0
imperative/python/src/tensor_utils.cpp
imperative/python/src/tensor_utils.cpp
+48
-0
未找到文件。
imperative/python/src/tensor_utils.cpp
浏览文件 @
8dd69e87
...
...
@@ -1747,6 +1747,54 @@ PyObject* broadcast_cpp(PyObject* self, PyObject* const* args, size_t nargs) {
PyObject
*
reshape_cpp
(
PyObject
*
self
,
PyObject
*
const
*
args
,
size_t
nargs
)
{
try
{
mgb_assert
(
nargs
==
2
,
"reshape should have 2 args but give %zu"
,
nargs
);
PyObject
*
pyinp
=
args
[
0
];
PyObject
*
pyshp
=
args
[
1
];
if
((
PyList_CheckExact
(
pyshp
)
||
PyTuple_CheckExact
(
pyshp
)
||
PyLong_Check
(
pyshp
))
&&
enable_fastpath
(
pyinp
))
{
using
OptionalAxisV1
=
::
megdnn
::
param
::
OptionalAxisV1
;
int32_t
unspec_axis
=
OptionalAxisV1
::
INVALID_AXIS
;
std
::
vector
<
int32_t
>
shape
;
bool
is_int_arrays
=
true
;
// judge whether the target shape is int arrays like (int,), [int], int
if
(
PyLong_Check
(
pyshp
))
{
shape
.
push_back
(
static_cast
<
int32_t
>
(
PyLong_AsLong
(
pyshp
)));
if
(
shape
[
0
]
==
-
1
)
{
unspec_axis
=
0
;
}
}
else
{
size_t
len
=
PySequence_Fast_GET_SIZE
(
pyshp
);
shape
.
resize
(
len
);
for
(
size_t
i
=
0
;
i
<
len
;
++
i
)
{
auto
obj
=
PySequence_Fast_GET_ITEM
(
pyshp
,
i
);
if
(
!
PyLong_Check
(
obj
))
{
is_int_arrays
=
false
;
break
;
}
shape
[
i
]
=
PyLong_AsLong
(
obj
);
if
(
shape
[
i
]
==
-
1
)
{
mgb_assert
(
unspec_axis
==
OptionalAxisV1
::
INVALID_AXIS
);
unspec_axis
=
i
;
}
}
}
// if the target shape is the int arrays, we dispatch directly
if
(
is_int_arrays
)
{
std
::
shared_ptr
<
OpDef
>
op
=
Reshape
::
make
(
unspec_axis
,
shape
);
std
::
vector
<
PyObject
*>
packed_param
(
2
);
py
::
object
Op
=
py
::
cast
(
op
);
packed_param
[
0
]
=
Op
.
ptr
();
packed_param
[
1
]
=
pyinp
;
py
::
tuple
ret
=
py
::
reinterpret_steal
<
py
::
object
>
(
py_apply
(
NULL
,
packed_param
.
data
(),
packed_param
.
size
()));
return
py
::
object
(
ret
[
0
]).
release
().
ptr
();
}
}
// fallback
return
_reshape_cpp
(
args
[
0
],
args
[
1
]).
release
().
ptr
();
}
PYEXT17_TRANSLATE_EXC_RET
(
nullptr
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录