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)