diff options
Diffstat (limited to 'src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp')
| -rw-r--r-- | src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp b/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp index 98ea03ac3c..796797728e 100644 --- a/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp +++ b/src/backends/tosaCommon/operatorMappings/AdditionOperator.cpp @@ -6,7 +6,8 @@ #include "AdditionOperator.hpp" TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const std::vector<const TensorInfo*>& inputs, - const std::vector<const TensorInfo*>& outputs) + const std::vector<const TensorInfo*>& outputs, + bool isMain) { // A helper function with static global variables ensures uniqueness // for dynamically generating input, output and block names @@ -15,6 +16,12 @@ TosaSerializationBasicBlock* ConvertAdditionToTosaOperator(const std::vector<con std::string outputName = std::string("Op_ADD_output0_") + GetUniqueTosaMappingID(); std::string blockName = std::string("Op_ADD_block_") + GetUniqueTosaMappingID(); + // If it's the first block, overwrite block name with main. + if (isMain) + { + blockName = std::string("main"); + } + TosaSerializationOperator* op = new TosaSerializationOperator(Op_ADD, Attribute_NONE, nullptr, |