p47068915
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

add call parameter and populate call

This commit is contained in:
mengyuanli 2021-11-30 10:16:12 +08:00
parent 17fcbaae94
commit 0da2c8d145
2 changed files with 38 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/call_parameter.h Normal file
View File

@ -0,0 +1,28 @@
/**
* Copyright 2021 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_NNACL_CALL_PARAMETER_H_
#define MINDSPORE_NNACL_CALL_PARAMETER_H_
#include "nnacl/op_base.h"
#include "nnacl/int8/quantize.h"
typedef struct CallParameter {
OpParameter op_parameter_;
bool is_tail_call = true;
} CallParameter;
#endif // MINDSPORE_NNACL_CALL_PARAMETER_H_

13
mindspore/lite/src/ops/populate/call_populate.cc
View File

@ -14,6 +14,7 @@
* limitations under the License.
*/
#include "src/ops/populate/populate_register.h"
#include "nnacl/call_parameter.h"
using mindspore::schema::PrimitiveType_Call;
namespace mindspore {
@ -21,15 +22,21 @@ namespace lite {
OpParameter *PopulateCallParameter(const void *prim) {
MS_CHECK_TRUE_RET(prim != nullptr, nullptr);
auto primitive = static_cast<const schema::Primitive *>(prim);
auto value = primitive->value_as_Call();
if (value == nullptr) {
MS_LOG(ERROR) << "call param is nullptr";
return nullptr;
}
auto *param = reinterpret_cast<OpParameter *>(malloc(sizeof(OpParameter)));
auto *param = reinterpret_cast<CallParameter *>(malloc(sizeof(CallParameter)));
if (param == nullptr) {
MS_LOG(ERROR) << "malloc CallParameter failed.";
return nullptr;
}
memset(param, 0, sizeof(OpParameter));
memset(param, 0, sizeof(CallParameter));
param->type_ = primitive->value_type();
param->op_parameter_.type_ = primitive->value_type();
param->is_tail_call = value->is_tail_call();
return reinterpret_cast<OpParameter *>(param);
}