mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!21897 add Dynamic input ops set 

Merge pull request !21897 from yanzhenxiang2020/add_dynamic_input_map
This commit is contained in:
i-robot 2021-08-23 07:47:15 +00:00 committed by Gitee
parent 5297c760f5 d00c93885c
commit bf759339cd
2 changed files with 4 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_metadata.cc
View File

@ -38,9 +38,7 @@ void AicpuMetadataInfo(const CNodePtr &kernel_node, std::vector<std::shared_ptr<
return;
}
// For compatibility with the current framework
if (op_name == kPrint || op_name == kGetNext || op_name == kPack || op_name == kMeshgrid ||
op_name == kStackInitOpName || op_name == kStackDestroyOpName || op_name == kStackPushOpName ||
op_name == kStackPopOpName || op_name == kDynamicStitch) {
if (op_name == kGetNext || kDynamicInputOps.find(op_name) != kDynamicInputOps.end()) {
AicpuMetadataInfoForSpecialNodes(kernel_node, kernel_info_list);
return;
}
@ -55,9 +53,7 @@ void AicpuMetadataInfoForSpecialNodes(const CNodePtr &kernel_node,
std::vector<std::string> inputs_format{};
std::vector<TypeId> inputs_type{};
auto op_name = AnfAlgo::GetCNodeName(kernel_node);
if (op_name == kPrint || op_name == kPack || op_name == kMeshgrid || op_name == kStackInitOpName ||
op_name == kStackDestroyOpName || op_name == kStackPushOpName || op_name == kStackPopOpName ||
op_name == kDynamicStitch) {
if (kDynamicInputOps.find(op_name) != kDynamicInputOps.end()) {
size_t input_num = AnfAlgo::GetInputTensorNum(kernel_node);
for (size_t input_index = 0; input_index < input_num; ++input_index) {
inputs_format.emplace_back(kOpFormat_DEFAULT);

3
mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_util.h
View File

@ -68,7 +68,8 @@ const std::set<std::string> kCustAiCpuKernelOps{kIdentity, kMaskedSelect,
kSearchSorted, kResizeBilinear, kResizeBilinearGrad};
const std::set<std::string> kCacheKernelOps{kUpdateCache, kCacheSwapTable, kSubAndFilter,
kPadAndShift, kDropout3D, kDropout2D};
const std::set<std::string> kDynamicInputOps{
kPrint, kPack, kMeshgrid, kStackInitOpName, kStackDestroyOpName, kStackPushOpName, kStackPopOpName, kDynamicStitch};
struct AicpuParamHead {
uint32_t length; // Total length: include cunstom message
uint32_t ioAddrNum; // Input and output address number