From 4d19889acc0bc72d9680aff19602e0ea812c9a57 Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Mon, 26 Jun 2023 22:14:32 +0100
Subject: [PATCH] where_cond for f16.

---
 kernels/src/ternary.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/kernels/src/ternary.cu b/kernels/src/ternary.cu
index 8f8b3ac5..8f51526b 100644
--- a/kernels/src/ternary.cu
+++ b/kernels/src/ternary.cu
@@ -32,6 +32,10 @@ extern "C" __global__ void FN_NAME(  \
     } \
 } \
 
+#if __CUDA_ARCH__ >= 530
+WHERE_OP(__half, where_f16)
+#endif
+
 WHERE_OP(float, where_f32)
 WHERE_OP(double, where_f64)
 WHERE_OP(uint32_t, where_u32)