mmcv编译不过修改

mmcv/ops/csrc/common/cuda/correlation_cuda.cuh

@@ -78,7 +78,11 @@ __global__ void correlation_forward_cuda_kernel(
       }
       // accumulate
       for (int offset = 16; offset > 0; offset /= 2)
-        prod_sum += __shfl_down_sync(FULL_MASK, float(prod_sum), offset);
+        #ifdef HIP_DIFF
+          prod_sum += __shfl_down(FULL_MASK, prod_sum, offset);
+        #else
+          prod_sum += __shfl_down_sync(FULL_MASK, float(prod_sum), offset);
+        #endif
       if (thread == 0) {
         output[n][ph][pw][h][w] = prod_sum;
       }