add Transpose to fusion_type_name_maps

2022-11-28 15:35:01 +08:00 · 2022-11-28 15:35:01 +08:00 · 8dcd46acbf
parent 0f1ef33250
commit 8dcd46acbf
2 changed files with 2 additions and 0 deletions
--- a/mindspore/ccsrc/kernel/common_utils.cc
+++ b/mindspore/ccsrc/kernel/common_utils.cc
@ -99,6 +99,7 @@ const std::unordered_map<FusionType, std::string> fusion_type_name_maps = {
  {FusionType::DROPOUT_DOMASKV3D, "DropOutDoMaskV3D"},
  {FusionType::TRANSDATA, "Transdata"},
  {FusionType::NORM, "Norm"},
+  {FusionType::TRANSPOSE, "Transpose"},
  {FusionType::UNKNOWN_FUSION_TYPE, ""}};

 abstract::BaseShapePtr GetValidShapeFromAbstract(const abstract::AbstractBasePtr &abs) {
--- a/mindspore/ccsrc/kernel/kernel.h
+++ b/mindspore/ccsrc/kernel/kernel.h
@ -106,6 +106,7 @@ enum FusionType {
  DROPOUT_DOMASKV3D,
  TRANSDATA,
  NORM,
+  TRANSPOSE,
  UNKNOWN_FUSION_TYPE = -1,
 };