|
|
|
@ -269,11 +269,11 @@ void AscendBackendIRFusionOptimization(const std::shared_ptr<session::KernelGrap
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<BnGradSplit>());
|
|
|
|
|
} else {
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<BatchNormGradSplit>());
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<LayerNormGradSplit>());
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<FusedBatchNormFusion>());
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<FusedBatchNormMixPrecisionFusion0>());
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<FusedBatchNormMixPrecisionFusion1>());
|
|
|
|
|
}
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<LayerNormGradSplit>());
|
|
|
|
|
ir_fusion_pm->AddPass(std::make_shared<InsertPadForNMSWithMask>());
|
|
|
|
|
AddAscendIRFusionRulesPass(ir_fusion_pm.get());
|
|
|
|
|
AddAscendIRFusionPass(ir_fusion_pm.get());
|
|
|
|
|