Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
magicwindyyd
mindspore
提交
fa83fc3e
M
mindspore
项目概览
magicwindyyd
/
mindspore
与 Fork 源项目一致
Fork自
MindSpore / mindspore
通知
1
Star
1
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看板
提交
fa83fc3e
编写于
5月 13, 2020
作者:
D
dengwentao
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
rename kernel/mng to kernel/rts
上级
25b2424f
变更
30
隐藏空白更改
内联
并排
Showing
30 changed file
with
149 addition
and
149 deletion
+149
-149
mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
+1
-1
mindspore/ccsrc/kernel/CMakeLists.txt
mindspore/ccsrc/kernel/CMakeLists.txt
+1
-1
mindspore/ccsrc/kernel/kernel_query.cc
mindspore/ccsrc/kernel/kernel_query.cc
+1
-1
mindspore/ccsrc/kernel/rts/assign.cc
mindspore/ccsrc/kernel/rts/assign.cc
+1
-1
mindspore/ccsrc/kernel/rts/assign.h
mindspore/ccsrc/kernel/rts/assign.h
+5
-5
mindspore/ccsrc/kernel/rts/label_goto.cc
mindspore/ccsrc/kernel/rts/label_goto.cc
+1
-1
mindspore/ccsrc/kernel/rts/label_goto.h
mindspore/ccsrc/kernel/rts/label_goto.h
+5
-5
mindspore/ccsrc/kernel/rts/label_set.cc
mindspore/ccsrc/kernel/rts/label_set.cc
+1
-1
mindspore/ccsrc/kernel/rts/label_set.h
mindspore/ccsrc/kernel/rts/label_set.h
+5
-5
mindspore/ccsrc/kernel/rts/label_switch.cc
mindspore/ccsrc/kernel/rts/label_switch.cc
+1
-1
mindspore/ccsrc/kernel/rts/label_switch.h
mindspore/ccsrc/kernel/rts/label_switch.h
+5
-5
mindspore/ccsrc/kernel/rts/memcpy_async.cc
mindspore/ccsrc/kernel/rts/memcpy_async.cc
+1
-1
mindspore/ccsrc/kernel/rts/memcpy_async.h
mindspore/ccsrc/kernel/rts/memcpy_async.h
+5
-5
mindspore/ccsrc/kernel/rts/profiling_kernel_mod.cc
mindspore/ccsrc/kernel/rts/profiling_kernel_mod.cc
+1
-1
mindspore/ccsrc/kernel/rts/profiling_kernel_mod.h
mindspore/ccsrc/kernel/rts/profiling_kernel_mod.h
+4
-4
mindspore/ccsrc/kernel/rts/recv.cc
mindspore/ccsrc/kernel/rts/recv.cc
+1
-1
mindspore/ccsrc/kernel/rts/recv.h
mindspore/ccsrc/kernel/rts/recv.h
+5
-5
mindspore/ccsrc/kernel/rts/rt_kernel.cc
mindspore/ccsrc/kernel/rts/rt_kernel.cc
+1
-1
mindspore/ccsrc/kernel/rts/rt_kernel.h
mindspore/ccsrc/kernel/rts/rt_kernel.h
+3
-3
mindspore/ccsrc/kernel/rts/rt_kernel_build.cc
mindspore/ccsrc/kernel/rts/rt_kernel_build.cc
+2
-2
mindspore/ccsrc/kernel/rts/rt_kernel_build.h
mindspore/ccsrc/kernel/rts/rt_kernel_build.h
+3
-3
mindspore/ccsrc/kernel/rts/rt_kernel_info.cc
mindspore/ccsrc/kernel/rts/rt_kernel_info.cc
+1
-1
mindspore/ccsrc/kernel/rts/rt_kernel_info.h
mindspore/ccsrc/kernel/rts/rt_kernel_info.h
+75
-75
mindspore/ccsrc/kernel/rts/send.cc
mindspore/ccsrc/kernel/rts/send.cc
+1
-1
mindspore/ccsrc/kernel/rts/send.h
mindspore/ccsrc/kernel/rts/send.h
+5
-5
mindspore/ccsrc/kernel/rts/stream_active.cc
mindspore/ccsrc/kernel/rts/stream_active.cc
+1
-1
mindspore/ccsrc/kernel/rts/stream_active.h
mindspore/ccsrc/kernel/rts/stream_active.h
+5
-5
mindspore/ccsrc/kernel/rts/stream_switch.cc
mindspore/ccsrc/kernel/rts/stream_switch.cc
+1
-1
mindspore/ccsrc/kernel/rts/stream_switch.h
mindspore/ccsrc/kernel/rts/stream_switch.h
+5
-5
tests/ut/cpp/CMakeLists.txt
tests/ut/cpp/CMakeLists.txt
+2
-2
未找到文件。
mindspore/ccsrc/device/ascend/kernel_build_ascend.cc
浏览文件 @
fa83fc3e
...
...
@@ -28,7 +28,7 @@
#include "kernel/tbe/tbe_kernel_parallel_build.h"
#include "kernel/aicpu/aicpu_kernel_build.h"
#include "kernel/hccl/hccl_kernel_build.h"
#include "kernel/
mng
/rt_kernel_build.h"
#include "kernel/
rts
/rt_kernel_build.h"
#include "kernel/tbe/tbe_utils.h"
#include "operator/ops.h"
#include "session/anf_runtime_algorithm.h"
...
...
mindspore/ccsrc/kernel/CMakeLists.txt
浏览文件 @
fa83fc3e
...
...
@@ -11,7 +11,7 @@ if (ENABLE_D)
"kernel_fusion.cc"
"tbe/*.cc"
"aicpu/*.cc"
"
mng
/*.cc"
"
rts
/*.cc"
"hccl/*.cc"
)
add_compile_definitions
(
ENABLE_D
)
...
...
mindspore/ccsrc/kernel/kernel_query.cc
浏览文件 @
fa83fc3e
...
...
@@ -18,7 +18,7 @@
#include <memory>
#include <algorithm>
#include "kernel/aicpu/aicpu_kernel_metadata.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel_info.h"
#include "kernel/hccl/hccl_kernel_metadata.h"
#include "kernel/tbe/tbe_kernel_select.h"
#include "session/anf_runtime_algorithm.h"
...
...
mindspore/ccsrc/kernel/
mng
/assign.cc
→
mindspore/ccsrc/kernel/
rts
/assign.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/assign.h"
#include "kernel/
rts
/assign.h"
#include <memory>
...
...
mindspore/ccsrc/kernel/
mng
/assign.h
→
mindspore/ccsrc/kernel/
rts
/assign.h
浏览文件 @
fa83fc3e
...
...
@@ -14,12 +14,12 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_ASSIGN_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_ASSIGN_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_ASSIGN_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_ASSIGN_H
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -38,4 +38,4 @@ MS_REG_RTKERNEL(assign, AssignKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_ASSIGN_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_ASSIGN_H
mindspore/ccsrc/kernel/
mng
/label_goto.cc
→
mindspore/ccsrc/kernel/
rts
/label_goto.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/label_goto.h"
#include "kernel/
rts
/label_goto.h"
#include <asm-generic/param.h>
#include <memory>
#include "runtime/stream.h"
...
...
mindspore/ccsrc/kernel/
mng
/label_goto.h
→
mindspore/ccsrc/kernel/
rts
/label_goto.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_GOTO_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_GOTO_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_GOTO_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_GOTO_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -44,4 +44,4 @@ MS_REG_RTKERNEL(labelgoto, LabelGotoKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_GOTO_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_GOTO_H
mindspore/ccsrc/kernel/
mng
/label_set.cc
→
mindspore/ccsrc/kernel/
rts
/label_set.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/label_set.h"
#include "kernel/
rts
/label_set.h"
#include <asm-generic/param.h>
#include <memory>
#include "runtime/stream.h"
...
...
mindspore/ccsrc/kernel/
mng
/label_set.h
→
mindspore/ccsrc/kernel/
rts
/label_set.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SET_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SET_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SET_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SET_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -44,4 +44,4 @@ MS_REG_RTKERNEL(labelset, LabelSetKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SET_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SET_H
mindspore/ccsrc/kernel/
mng
/label_switch.cc
→
mindspore/ccsrc/kernel/
rts
/label_switch.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/label_switch.h"
#include "kernel/
rts
/label_switch.h"
#include <asm-generic/param.h>
#include <memory>
#include "runtime/stream.h"
...
...
mindspore/ccsrc/kernel/
mng
/label_switch.h
→
mindspore/ccsrc/kernel/
rts
/label_switch.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SWITCH_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SWITCH_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SWITCH_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SWITCH_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -46,4 +46,4 @@ MS_REG_RTKERNEL(labelswitch, LabelSwitchKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_LABEL_SWITCH_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_LABEL_SWITCH_H
mindspore/ccsrc/kernel/
mng
/memcpy_async.cc
→
mindspore/ccsrc/kernel/
rts
/memcpy_async.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/memcpy_async.h"
#include "kernel/
rts
/memcpy_async.h"
#include <memory>
#include <string>
...
...
mindspore/ccsrc/kernel/
mng
/memcpy_async.h
→
mindspore/ccsrc/kernel/
rts
/memcpy_async.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_MEMCPY_ASYNC_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_MEMCPY_ASYNC_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_MEMCPY_ASYNC_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_MEMCPY_ASYNC_H
#include <vector>
#include <memory>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -53,4 +53,4 @@ MS_REG_RTKERNEL(memcpy_async, MemCpyAsyncKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_MEMCPY_ASYNC_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_MEMCPY_ASYNC_H
mindspore/ccsrc/kernel/
mng
/profiling_kernel_mod.cc
→
mindspore/ccsrc/kernel/
rts
/profiling_kernel_mod.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/profiling_kernel_mod.h"
#include "kernel/
rts
/profiling_kernel_mod.h"
#include <vector>
#include <string>
...
...
mindspore/ccsrc/kernel/
mng
/profiling_kernel_mod.h
→
mindspore/ccsrc/kernel/
rts
/profiling_kernel_mod.h
浏览文件 @
fa83fc3e
...
...
@@ -13,10 +13,10 @@
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#ifndef MINDSPORE_MINDSPORE_CCSRC_KERNEL_
PROFILING
_PROFILING_KERNEL_MOD_H_
#define MINDSPORE_MINDSPORE_CCSRC_KERNEL_
PROFILING
_PROFILING_KERNEL_MOD_H_
#ifndef MINDSPORE_MINDSPORE_CCSRC_KERNEL_
RTS
_PROFILING_KERNEL_MOD_H_
#define MINDSPORE_MINDSPORE_CCSRC_KERNEL_
RTS
_PROFILING_KERNEL_MOD_H_
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel.h"
namespace
mindspore
{
namespace
kernel
{
class
ProfilingKernelMod
:
public
RtKernel
{
...
...
@@ -37,4 +37,4 @@ class ProfilingKernelMod : public RtKernel {
MS_REG_RTKERNEL
(
profiling
,
ProfilingKernelMod
);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_MINDSPORE_CCSRC_KERNEL_
PROFILING
_PROFILING_KERNEL_MOD_H_
#endif // MINDSPORE_MINDSPORE_CCSRC_KERNEL_
RTS
_PROFILING_KERNEL_MOD_H_
mindspore/ccsrc/kernel/
mng
/recv.cc
→
mindspore/ccsrc/kernel/
rts
/recv.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/recv.h"
#include "kernel/
rts
/recv.h"
#include <memory>
#include "runtime/stream.h"
#include "utils/context/ms_context.h"
...
...
mindspore/ccsrc/kernel/
mng
/recv.h
→
mindspore/ccsrc/kernel/
rts
/recv.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_RECV_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_RECV_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_RECV_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_RECV_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -43,4 +43,4 @@ MS_REG_RTKERNEL(recv, RecvKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_RECV_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_RECV_H
mindspore/ccsrc/kernel/
mng
/rt_kernel.cc
→
mindspore/ccsrc/kernel/
rts
/rt_kernel.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel.h"
namespace
mindspore
{
namespace
kernel
{
...
...
mindspore/ccsrc/kernel/
mng
/rt_kernel.h
→
mindspore/ccsrc/kernel/
rts
/rt_kernel.h
浏览文件 @
fa83fc3e
...
...
@@ -14,8 +14,8 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_H
#include <vector>
#include <utility>
...
...
@@ -74,4 +74,4 @@ class _RtKernelRegister {
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_H
mindspore/ccsrc/kernel/
mng
/rt_kernel_build.cc
→
mindspore/ccsrc/kernel/
rts
/rt_kernel_build.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,14 +14,14 @@
* limitations under the License.
*/
#include "kernel/
mng
/rt_kernel_build.h"
#include "kernel/
rts
/rt_kernel_build.h"
#include <string>
#include <memory>
#include <utility>
#include <algorithm>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel.h"
#include "session/anf_runtime_algorithm.h"
namespace
mindspore
{
...
...
mindspore/ccsrc/kernel/
mng
/rt_kernel_build.h
→
mindspore/ccsrc/kernel/
rts
/rt_kernel_build.h
浏览文件 @
fa83fc3e
...
...
@@ -14,8 +14,8 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_BUILD_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_BUILD_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_BUILD_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_BUILD_H
#include <vector>
#include <memory>
...
...
@@ -26,4 +26,4 @@ KernelModPtr RtOpBuild(const AnfNodePtr &anf_node);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_RT_KERNEL_BUILD_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_RT_KERNEL_BUILD_H
mindspore/ccsrc/kernel/
mng
/rt_kernel_info.cc
→
mindspore/ccsrc/kernel/
rts
/rt_kernel_info.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel_info.h"
#include <unordered_map>
#include <algorithm>
#include "utils/convert_utils.h"
...
...
mindspore/ccsrc/kernel/
mng
/rt_kernel_info.h
→
mindspore/ccsrc/kernel/
rts
/rt_kernel_info.h
浏览文件 @
fa83fc3e
/**
* Copyright 2019 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_CCSRC_KERNEL_
MNG_RT_KERNEL_INFO_H
#define MINDSPORE_CCSRC_KERNEL_
MNG_RT_KERNEL_INFO_H
#include <memory>
#include <functional>
#include <map>
#include <string>
#include <set>
#include <vector>
#include <utility>
#include "ir/dtype.h"
#include "kernel/kernel_build_info.h"
#include "kernel/kernel.h"
#include "utils/utils.h"
namespace
mindspore
{
namespace
kernel
{
class
RtKerDesc
{
public:
virtual
~
RtKerDesc
()
{}
virtual
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
GetKernelInfo
()
{
return
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
{};
}
};
using
RtKerDescCreater
=
std
::
function
<
std
::
shared_ptr
<
RtKerDesc
>
()
>
;
class
RtKerDescFactory
{
RtKerDescFactory
()
=
default
;
~
RtKerDescFactory
()
=
default
;
public:
static
RtKerDescFactory
&
Get
();
void
Register
(
const
std
::
string
&
name
,
RtKerDescCreater
&&
fun
);
static
std
::
shared_ptr
<
RtKerDesc
>
Create
(
const
std
::
string
&
name
);
private:
std
::
map
<
std
::
string
,
RtKerDescCreater
>
fmap_
;
};
class
_RtKerDescRegister
{
public:
_RtKerDescRegister
(
const
std
::
string
&
name
,
RtKerDescCreater
&&
fun
)
{
RtKerDescFactory
::
Get
().
Register
(
name
,
std
::
move
(
fun
));
}
~
_RtKerDescRegister
()
=
default
;
};
#define _MS_REG_RTKERNEL_DESC_REG(KNAME, clazz) \
static_assert
(
std
::
is_base_of
<
RtKerDesc
,
clazz
>::
value
,
" must be base of RtKerDesc"
);
\
static
const
_RtKerDescRegister
g_
##
KNAME
##
_
##
_rtkernel_desc_reg
(
#
KNAME
,
[]()
{
return
std
::
make_shared
<
clazz
>
();
});
#define MS_REG_RTKERNEL_DESC(KNAME, clazz) _MS_REG_RTKERNEL_DESC_REG(KNAME, clazz)
void
GetRtKelInfo
(
const
CNodePtr
&
kernel_node
,
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
*
kernel_info_list
);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG_RT_KERNEL_INFO_H
/**
* Copyright 2019 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_CCSRC_KERNEL_
RTS_RT_KERNEL_INFO_H
#define MINDSPORE_CCSRC_KERNEL_
RTS_RT_KERNEL_INFO_H
#include <memory>
#include <functional>
#include <map>
#include <string>
#include <set>
#include <vector>
#include <utility>
#include "ir/dtype.h"
#include "kernel/kernel_build_info.h"
#include "kernel/kernel.h"
#include "utils/utils.h"
namespace
mindspore
{
namespace
kernel
{
class
RtKerDesc
{
public:
virtual
~
RtKerDesc
()
{}
virtual
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
GetKernelInfo
()
{
return
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
{};
}
};
using
RtKerDescCreater
=
std
::
function
<
std
::
shared_ptr
<
RtKerDesc
>
()
>
;
class
RtKerDescFactory
{
RtKerDescFactory
()
=
default
;
~
RtKerDescFactory
()
=
default
;
public:
static
RtKerDescFactory
&
Get
();
void
Register
(
const
std
::
string
&
name
,
RtKerDescCreater
&&
fun
);
static
std
::
shared_ptr
<
RtKerDesc
>
Create
(
const
std
::
string
&
name
);
private:
std
::
map
<
std
::
string
,
RtKerDescCreater
>
fmap_
;
};
class
_RtKerDescRegister
{
public:
_RtKerDescRegister
(
const
std
::
string
&
name
,
RtKerDescCreater
&&
fun
)
{
RtKerDescFactory
::
Get
().
Register
(
name
,
std
::
move
(
fun
));
}
~
_RtKerDescRegister
()
=
default
;
};
#define _MS_REG_RTKERNEL_DESC_REG(KNAME, clazz) \
static_assert(std::is_base_of<RtKerDesc, clazz>::value, " must be base of RtKerDesc"); \
static const _RtKerDescRegister g_##KNAME##_##_rtkernel_desc_reg(#KNAME, []() { return std::make_shared<clazz>(); });
#define MS_REG_RTKERNEL_DESC(KNAME, clazz) _MS_REG_RTKERNEL_DESC_REG(KNAME, clazz)
void
GetRtKelInfo
(
const
CNodePtr
&
kernel_node
,
std
::
vector
<
std
::
shared_ptr
<
kernel
::
KernelBuildInfo
>>
*
kernel_info_list
);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
RTS_RT_KERNEL_INFO_H
mindspore/ccsrc/kernel/
mng
/send.cc
→
mindspore/ccsrc/kernel/
rts
/send.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/send.h"
#include "kernel/
rts
/send.h"
#include <memory>
#include "runtime/event.h"
#include "framework/ge_runtime/task_info.h"
...
...
mindspore/ccsrc/kernel/
mng
/send.h
→
mindspore/ccsrc/kernel/
rts
/send.h
浏览文件 @
fa83fc3e
...
...
@@ -14,12 +14,12 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_SEND_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_SEND_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_SEND_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_SEND_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -41,4 +41,4 @@ MS_REG_RTKERNEL(send, SendKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_SEND_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_SEND_H
mindspore/ccsrc/kernel/
mng
/stream_active.cc
→
mindspore/ccsrc/kernel/
rts
/stream_active.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/stream_active.h"
#include "kernel/
rts
/stream_active.h"
#include <asm-generic/param.h>
#include <memory>
#include "runtime/stream.h"
...
...
mindspore/ccsrc/kernel/
mng
/stream_active.h
→
mindspore/ccsrc/kernel/
rts
/stream_active.h
浏览文件 @
fa83fc3e
...
...
@@ -14,12 +14,12 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_ACTIVE_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_ACTIVE_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_ACTIVE_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_ACTIVE_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -43,4 +43,4 @@ MS_REG_RTKERNEL(streamactive, StreamActiveKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_ACTIVE_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_ACTIVE_H
mindspore/ccsrc/kernel/
mng
/stream_switch.cc
→
mindspore/ccsrc/kernel/
rts
/stream_switch.cc
浏览文件 @
fa83fc3e
...
...
@@ -14,7 +14,7 @@
* limitations under the License.
*/
#include "kernel/
mng
/stream_switch.h"
#include "kernel/
rts
/stream_switch.h"
#include <memory>
#include <vector>
...
...
mindspore/ccsrc/kernel/
mng
/stream_switch.h
→
mindspore/ccsrc/kernel/
rts
/stream_switch.h
浏览文件 @
fa83fc3e
...
...
@@ -14,13 +14,13 @@
* limitations under the License.
*/
#ifndef MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_SWITCH_H
#define MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_SWITCH_H
#ifndef MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_SWITCH_H
#define MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_SWITCH_H
#include <memory>
#include <vector>
#include "kernel/
mng
/rt_kernel.h"
#include "kernel/
mng
/rt_kernel_info.h"
#include "kernel/
rts
/rt_kernel.h"
#include "kernel/
rts
/rt_kernel_info.h"
namespace
mindspore
{
namespace
kernel
{
...
...
@@ -46,4 +46,4 @@ MS_REG_RTKERNEL(streamswitch, StreamSwitchKernel);
}
// namespace kernel
}
// namespace mindspore
#endif // MINDSPORE_CCSRC_KERNEL_
MNG
_STREAM_SWITCH_H
#endif // MINDSPORE_CCSRC_KERNEL_
RTS
_STREAM_SWITCH_H
tests/ut/cpp/CMakeLists.txt
浏览文件 @
fa83fc3e
...
...
@@ -70,7 +70,7 @@ file(GLOB_RECURSE MINDSPORE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"../../../mindspore/ccsrc/kernel/akg/*.cc"
"../../../mindspore/ccsrc/kernel/kash/*.cc"
"../../../mindspore/ccsrc/kernel/cce/*.cc"
"../../../mindspore/ccsrc/kernel/
mng
/*.cc"
"../../../mindspore/ccsrc/kernel/
rts
/*.cc"
"../../../mindspore/ccsrc/kernel/hccl/*.cc"
"../../../mindspore/ccsrc/kernel/kernel_query.cc"
"../../../mindspore/ccsrc/kernel/kernel_build_info.cc"
...
...
@@ -80,7 +80,7 @@ file(GLOB_RECURSE MINDSPORE_SRC_LIST RELATIVE ${CMAKE_CURRENT_SOURCE_DIR}
"../../../mindspore/ccsrc/pre_activate/mem_reuse/*.cc"
"../../../mindspore/ccsrc/pre_activate/pass/*.cc"
"../../../mindspore/ccsrc/kernel/aicpu/aicpu_kernel_metadata.cc"
"../../../mindspore/ccsrc/kernel/
mng
/rt_kernel_info.cc"
"../../../mindspore/ccsrc/kernel/
rts
/rt_kernel_info.cc"
"../../../mindspore/ccsrc/kernel/common_utils.cc"
"../../../mindspore/ccsrc/kernel/oplib/*.cc"
"../../../mindspore/ccsrc/kernel/tbe/*.cc"
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录