From d153b5aa0af0cd2290d65b5e83c002ec53d6a075 Mon Sep 17 00:00:00 2001
From: AllentDan <41138331+AllentDan@users.noreply.github.com>
Date: Thu, 29 Sep 2022 18:35:51 +0800
Subject: [PATCH] add rewrite for ort (#1138)

---
 mmdeploy/codebase/mmdet/models/backbones.py | 7 +++----
 1 file changed, 3 insertions(+), 4 deletions(-)

diff --git a/mmdeploy/codebase/mmdet/models/backbones.py b/mmdeploy/codebase/mmdet/models/backbones.py
index daa8b2a59..122a362b9 100644
--- a/mmdeploy/codebase/mmdet/models/backbones.py
+++ b/mmdeploy/codebase/mmdet/models/backbones.py
@@ -175,10 +175,9 @@ def shift_window_msa__window_partition__tensorrt(ctx, self, x):
 
 
 @FUNCTION_REWRITER.register_rewriter(
-    func_name='mmdet.models.backbones.swin.ShiftWindowMSA.forward',
-    backend='tensorrt')
-def shift_window_msa__forward__tensorrt(ctx, self, query, hw_shape):
-    """Rewrite forward function of ShiftWindowMSA class for TensorRT.
+    func_name='mmdet.models.backbones.swin.ShiftWindowMSA.forward')
+def shift_window_msa__forward__default(ctx, self, query, hw_shape):
+    """Rewrite forward function of ShiftWindowMSA class.
 
     1. replace dynamic padding with static padding and dynamic slice.
     2. always do slice `x = x[:, :H, :W, :].contiguous()` for stability.