[Fix] Call operator __half explicit (open-mmlab#2608)

tyomj · May 8, 2023 · f52641b · f52641b
1 parent 884ef8d
commit f52641b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mmcv/ops/csrc/common/cuda/carafe_cuda_kernel.cuh b/mmcv/ops/csrc/common/cuda/carafe_cuda_kernel.cuh
@@ -59,7 +59,7 @@ __device__ __forceinline__ phalf warpReduceSum(phalf val) {
     __PHALF(val) += __shfl_down(val, offset);
 #else
     __PHALF(val) +=
-        __shfl_down_sync(FULL_MASK, static_cast<__half>(__PHALF(val)), offset);
+        __shfl_down_sync(FULL_MASK, __PHALF(val).operator __half(), offset);
 #endif
   return val;
 }