From 36a4749e952ca6511cc5066e637b86b1422ba63d Mon Sep 17 00:00:00 2001
From: laurent <laurent.mazare@gmail.com>
Date: Mon, 26 Jun 2023 22:05:31 +0100
Subject: [PATCH] Add the f16 affine kernel.

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

diff --git a/kernels/src/affine.cu b/kernels/src/affine.cu
index 986c94a5..3ce5b8a7 100644
--- a/kernels/src/affine.cu
+++ b/kernels/src/affine.cu
@@ -28,6 +28,10 @@ extern "C" __global__ void FN_NAME(  \
     } \
 } \
 
+#if __CUDA_ARCH__ >= 530
+AFFINE_OP(__half, affine_f16)
+#endif
+
 AFFINE_OP(float, affine_f32)
 AFFINE_OP(double, affine_f64)
 AFFINE_OP(uint32_t, affine_u32)