From 8dcd46acbf4f64c93059d99739d570fe6dbee18c Mon Sep 17 00:00:00 2001 From: hemaohua Date: Mon, 28 Nov 2022 15:35:01 +0800 Subject: [PATCH] add Transpose to fusion_type_name_maps --- mindspore/ccsrc/kernel/common_utils.cc | 1 + mindspore/ccsrc/kernel/kernel.h | 1 + 2 files changed, 2 insertions(+) diff --git a/mindspore/ccsrc/kernel/common_utils.cc b/mindspore/ccsrc/kernel/common_utils.cc index d30f82afec2..250463bd0d2 100644 --- a/mindspore/ccsrc/kernel/common_utils.cc +++ b/mindspore/ccsrc/kernel/common_utils.cc @@ -99,6 +99,7 @@ const std::unordered_map 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) { diff --git a/mindspore/ccsrc/kernel/kernel.h b/mindspore/ccsrc/kernel/kernel.h index b4ca9f5a3d3..90f0472dc22 100644 --- 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, };