fix roi_pool bug.

momo609 · momo609 · commit b35f773fed24 · 2024-06-11T09:31:42.000+08:00
diff --git a/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp b/mmcv/ops/csrc/pytorch/npu/roi_pool_npu.cpp
@@ -70,7 +70,8 @@ void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,
       .Attr("spatial_scale_w", spatial_scale)
       .Attr("pool_channel", pooled_channel)
       .Run();
-  at::Tensor res = y.contiguous();
+  at::Tensor result = y.transpose(2, 3).transpose(1, 2);
+  at::Tensor res = result.contiguous();
   grad_input.copy_(res);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,8 @@ void roi_pool_backward_npu(Tensor grad_output, Tensor rois, Tensor argmax,`
`70`	`70`	`.Attr("spatial_scale_w", spatial_scale)`
`71`	`71`	`.Attr("pool_channel", pooled_channel)`
`72`	`72`	`.Run();`
`73`		`- at::Tensor res = y.contiguous();`
	`73`	`+ at::Tensor result = y.transpose(2, 3).transpose(1, 2);`
	`74`	`+ at::Tensor res = result.contiguous();`
`74`	`75`	`grad_input.copy_(res);`
`75`	`76`	`}`
`76`	`77`