From 9ae8f93ea4a62b7da8350ce62bc9966844ae9775 Mon Sep 17 00:00:00 2001
From: 103yiran <1039105206@qq.com>
Date: Thu, 26 Jun 2025 11:58:44 +0800
Subject: [PATCH] fix grad_output shape error

$N \times G \times \frac{C_o}{G} \times H_o \times W_o \to N \times C_o \times H_o \times W_o$
---
 mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp b/mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp
index b0f080cdac..efd201bff0 100644
--- a/mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp
+++ b/mmcv/ops/csrc/pytorch/modulated_deform_conv.cpp
@@ -245,8 +245,8 @@ void modulated_deform_conv_backward_fallthrough(
     if (with_bias)
       grad_bias = grad_bias.view({grad_bias.size(0) * grad_bias.size(1)});
   }
-  grad_output = grad_output.view({grad_output.size(0) * grad_output.size(1),
-                                  grad_output.size(2), grad_output.size(3),
+  grad_output = grad_output.view({grad_output.size(0), grad_output.size(1)
+                                  * grad_output.size(2), grad_output.size(3),
                                   grad_output.size(4)});
 }