Fix half conversions

NVIDIA · Oct 27, 2021 · 143259c · 143259c
1 parent ef752cb
commit 143259c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cub/agent/agent_sub_warp_merge_sort.cuh b/cub/agent/agent_sub_warp_merge_sort.cuh
@@ -122,15 +122,15 @@ class AgentSubWarpSort
 #if defined(__CUDA_FP16_TYPES_EXIST__) && (CUB_PTX_ARCH < 530)
     __device__ bool operator()(__half lhs, __half rhs)
     {
-      return (*this)(static_cast<float>(lhs), static_cast<float>(rhs));
+      return (*this)(__half2float(lhs), __half2float(rhs));
     }
 #endif
   };
 
 #if defined(__CUDA_FP16_TYPES_EXIST__) && (CUB_PTX_ARCH < 530)
   __device__ static bool equal(__half lhs, __half rhs)
   {
-    return static_cast<float>(lhs) == static_cast<float>(rhs);
+    return __half2float(lhs) == __half2float(rhs);
   }
 #endif