[Rocm] fix tests of inplace_abn_op & grid_sampler_op

Author: Ray2020BD

python/paddle/fluid/layers/nn.py

@@ -3000,8 +3000,6 @@ def inplace_abn(input,
                              'inplace_abn')
     dtype = helper.input_dtype()
 
-    if core.is_compiled_with_rocm():
-        dtype = 'float32'
     input_shape = input.shape
     if data_layout == 'NCHW':
         channel_num = input_shape[1]

python/paddle/nn/functional/vision.py

@@ -265,10 +265,9 @@ def grid_sample(x,
 
     cudnn_version = get_cudnn_version()
     use_cudnn = False
-    if (
+    if not core.is_compiled_with_rocm() and (
             cudnn_version is not None
-    ) and align_corners and mode == 'bilinear' and padding_mode == 'zeros' and not core.is_compiled_with_rocm(
-    ):
+    ) and align_corners and mode == 'bilinear' and padding_mode == 'zeros':
         use_cudnn = True
         # CUDNN always computes gradients for all inputs
         x.stop_gradient = False