support v2.1.0

mmcv/ops/csrc/pytorch/cuda/upfirdn2d_kernel.cu

@@ -734,12 +734,13 @@ torch::Tensor upfirdn2d_op(torch::Tensor x, torch::Tensor f, int upx, int upy,
 
   // Launch CUDA kernel.
   void *args[] = {&p};
-#ifndef MMCV_WITH_HIP
-  AT_CUDA_CHECK(cudaLaunchKernel(spec.kernel, gridSize, blockSize, args, 0,
-                                 at::cuda::getCurrentCUDAStream()));
-#else
+#ifdef MMCV_WITH_HIP
   AT_CUDA_CHECK(hipLaunchKernel(spec.kernel, gridSize, blockSize, args, 0,
+                                at::cuda::getCurrentCUDAStream()));
+#else
+  AT_CUDA_CHECK(cudaLaunchKernel(spec.kernel, gridSize, blockSize, args, 0,
                                  at::cuda::getCurrentCUDAStream()));
 #endif
+
   return y;
 }