[Accuracy diff No.153] Fix accuracy diff for paddle.vision.ops.deform_conv2d API

python/paddle/vision/ops.py

@@ -892,6 +892,17 @@ def deform_conv2d(
 
  use_deform_conv2d_v1 = True if mask is None else False
 
+ # cpu not support float16, need to convert dtype.
+ if paddle.device.get_device() == "cpu":
+ if offset.dtype == paddle.float16:
+ offset = offset.astype(x.dtype)
+ if weight.dtype == paddle.float16:
+ weight = weight.astype(x.dtype)
+ if bias is not None and bias.dtype == paddle.float16:
+ bias = bias.astype(x.dtype)
+ if mask is not None and mask.dtype == paddle.float16:
+ mask = mask.astype(x.dtype)
+
  if in_dynamic_or_pir_mode():
  pre_bias = _C_ops.deformable_conv(
  x,

test/legacy_test/test_deformable_conv_op.py

@@ -465,5 +465,37 @@ def test_deform_conv2d_v2():
  test_deform_conv2d_v2()
 
 
+class TestDeformConv2DAPI_CPU_FP16(unittest.TestCase):
+ def setUp(self):
+ self.padding = [1, 1]
+ self.stride = [1, 1]
+ self.dilation = [1, 1]
+ self.groups = 1
+ self.data_format = "NCL"
+
+ def test_cpu_fp16(self):
+ with paddle.base.dygraph.guard(paddle.CPUPlace()):
+ x = paddle.ones([4, 5, 5, 5])
+ offset = paddle.ones([4, 90, 5, 5]).astype(paddle.float16)
+ weight = paddle.ones([5, 5, 3, 3]).astype(paddle.float16)
+ bias = paddle.ones([5]).astype(paddle.float16)
+ mask = paddle.ones([4, 45, 5, 5]).astype(paddle.float16)
+
+ # If there is an error, an error will be thrown.
+ out = paddle.vision.ops.deform_conv2d(
+ x,
+ offset,
+ weight,
+ bias,
+ stride=self.stride,
+ padding=self.padding,
+ dilation=self.dilation,
+ groups=self.groups,
+ deformable_groups=5,
+ mask=mask,
+ )
+ np.testing.assert_allclose(out.shape, [4, 5, 5, 5])
+
+
 if __name__ == '__main__':
  unittest.main()