fix bug forward

momo609 · Dec 9, 2024 · e9c17c1 · e9c17c1
1 parent f299001
commit e9c17c1
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/mmcv/ops/modulated_deform_conv.py b/mmcv/ops/modulated_deform_conv.py
@@ -56,7 +56,8 @@ def _npu_forward(ctx, input_tensor, offset, mask, weight, bias):
                 kernel_h, kernel_w, ctx.deform_groups)
         select_offset = offset.index_select(1, sort_index_fp)
         offset_all = torch.cat([select_offset, mask], dim=1)
-        output, offset_out = torch.npu_deformable_conv2d(
+        import torch_npu
+        output, offset_out = torch_npu.npu_deformable_conv2d(
             input_tensor,
             weight,
             offset_all,