From 8ee2f20782635d77def3a8a9ca4e7bf3b1950216 Mon Sep 17 00:00:00 2001
From: Yosuke Shinya <42844407+shinya7y@users.noreply.github.com>
Date: Sat, 8 May 2021 05:14:14 +0000
Subject: [PATCH] [Fix] Fix DCN fp16

---
 mmcv/ops/deform_conv.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/mmcv/ops/deform_conv.py b/mmcv/ops/deform_conv.py
index 5c45bd4ebb..13ff04ea09 100644
--- a/mmcv/ops/deform_conv.py
+++ b/mmcv/ops/deform_conv.py
@@ -70,6 +70,7 @@ def forward(ctx,
         ctx.deform_groups = deform_groups
         ctx.im2col_step = im2col_step
 
+        weight = weight.type_as(input)
         ctx.save_for_backward(input, offset, weight)
 
         output = input.new_empty(