Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
正统之独孤求败
mindspore
提交
b667193f
M
mindspore
项目概览
正统之独孤求败
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
M
mindspore
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
b667193f
编写于
7月 22, 2020
作者:
M
mindspore-ci-bot
提交者:
Gitee
7月 22, 2020
浏览文件
操作
浏览文件
下载
差异文件
!3260 check tbe kernel property in kernel select process
Merge pull request !3260 from jjfeing/master
上级
0b21dd0f
0d5f1fba
变更
4
隐藏空白更改
内联
并排
Showing
4 changed file
with
92 addition
and
0 deletion
+92
-0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc
...ernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc
+5
-0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
...el_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
+53
-0
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.h
...nel_compiler/tbe/tbe_kernel_select/tbe_property_checker.h
+33
-0
mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h
mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h
+1
-0
未找到文件。
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_select.cc
浏览文件 @
b667193f
...
...
@@ -31,6 +31,7 @@
#include "backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_broadcast_selecter.h"
#include "backend/kernel_compiler/tbe/tbe_kernel_select/tbe_kernel_reduce_selecter.h"
#include "backend/kernel_compiler/tbe/tbe_kernel_select/common_utils.h"
#include "backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -59,6 +60,10 @@ void TbeKernelSelect::TbeMetadataInfoEx() {
MS_LOG
(
INFO
)
<<
"Warning: Cann't find tbe core opinfo, node type: "
<<
node_name_
;
return
;
}
if
(
!
TbePropertyChecker
::
CheckTbeProperties
(
cnode_ptr_
))
{
MS_LOG
(
INFO
)
<<
"Warning: node("
<<
cnode_ptr_
->
fullname_with_scope
()
<<
") not support tbe aicore."
;
return
;
}
MS_LOG
(
INFO
)
<<
"Start to tbe metadata info. node type: "
<<
node_name_
<<
", node name: "
<<
cnode_ptr_
->
fullname_with_scope
();
OpPattern
pattern
=
op_info_ptr
->
op_pattern
();
...
...
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.cc
0 → 100644
浏览文件 @
b667193f
/**
* Copyright 2020 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#include "backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.h"
#include <map>
#include <string>
#include <vector>
#include "backend/session/anf_runtime_algorithm.h"
#include "frontend/parallel/ops_info/ops_utils.h"
namespace
mindspore
{
namespace
kernel
{
using
CheckSupportFun
=
bool
(
*
)(
const
CNodePtr
&
cnode
);
constexpr
char
kAttrStrides
[]
=
"strides"
;
static
bool
CheckStridedSlice
(
const
CNodePtr
&
cnode
)
{
// check stride[-1] != 1 TODO
if
(
AnfAlgo
::
HasNodeAttr
(
kAttrStrides
,
cnode
))
{
auto
strides
=
AnfAlgo
::
GetNodeAttr
<
std
::
vector
<
int
>>
(
cnode
,
kAttrStrides
);
if
(
!
strides
.
empty
()
&&
strides
[
strides
.
size
()
-
1
]
==
1
)
{
return
true
;
}
}
// last tensor TODO
return
true
;
}
bool
TbePropertyChecker
::
CheckTbeProperties
(
const
mindspore
::
CNodePtr
&
cnode
)
{
MS_EXCEPTION_IF_NULL
(
cnode
);
static
std
::
map
<
std
::
string
,
CheckSupportFun
>
tbe_property_checker
=
{{
parallel
::
KStridedSlice
,
CheckStridedSlice
}};
auto
cnode_type
=
AnfAlgo
::
GetCNodeName
(
cnode
);
auto
find_iter
=
tbe_property_checker
.
find
(
cnode_type
);
if
(
find_iter
!=
tbe_property_checker
.
end
())
{
return
find_iter
->
second
(
cnode
);
}
return
true
;
}
}
// namespace kernel
}
// namespace mindspore
mindspore/ccsrc/backend/kernel_compiler/tbe/tbe_kernel_select/tbe_property_checker.h
0 → 100644
浏览文件 @
b667193f
/**
* Copyright 2020 Huawei Technologies Co., Ltd
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MINDSPORE_BACKEND_KERNEL_COMPILER_TBE_TBE_KERNEL_SELECT_TBE_PROPERTY_CHECKER_H
#define MINDSPORE_BACKEND_KERNEL_COMPILER_TBE_TBE_KERNEL_SELECT_TBE_PROPERTY_CHECKER_H
#include "mindspore/core/ir/anf.h"
namespace
mindspore
{
namespace
kernel
{
class
TbePropertyChecker
{
public:
TbePropertyChecker
()
=
default
;
~
TbePropertyChecker
()
=
default
;
static
bool
CheckTbeProperties
(
const
mindspore
::
CNodePtr
&
cnode
);
};
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_BACKEND_KERNEL_COMPILER_TBE_TBE_KERNEL_SELECT_TBE_PROPERTY_CHECKER_H
mindspore/ccsrc/frontend/parallel/ops_info/ops_utils.h
浏览文件 @
b667193f
...
...
@@ -232,6 +232,7 @@ constexpr char UNSORTEF_SEGMENT_PRODD[] = "UnsortedSegmentProdD";
constexpr
char
DEPTHWISE_CONV2D_NATIVE
[]
=
"DepthwiseConv2dNative"
;
constexpr
char
DEPTHWISE_CONV2D
[]
=
"DepthwiseConv2D"
;
constexpr
char
ADD
[]
=
"Add"
;
constexpr
char
KStridedSlice
[]
=
"StridedSlice"
;
// Parallel don't care
constexpr
char
TUPLE_GETITEM
[]
=
"tuple_getitem"
;
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录