fix bug.

momo609 · momo609 · commit c6e04a5beb2a · 2024-05-31T10:43:13.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,7 @@ 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 = NpuUtils::format_contiguous(y);
+  at::Tensor res = y.contiguous();
   grad_input.copy_(res);
 }
 

Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ 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 = NpuUtils::format_contiguous(y);`
	`73`	`+ at::Tensor res = y.contiguous();`
`74`	`74`	`grad_input.copy_(res);`
`75`	`75`	`}`
`76`	`76`