From 79b722484ac31fecd10c4f0e8baddcf47987ceb6 Mon Sep 17 00:00:00 2001
From: "-T.K.-" <t_k_233@outlook.com>
Date: Thu, 11 Jul 2024 02:32:12 -0700
Subject: [PATCH] FIX: minor fix

---
 nn/functional/nn_relu.c | 2 +-
 nn/impl/minimum1.h      | 2 ++
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/nn/functional/nn_relu.c b/nn/functional/nn_relu.c
index a7c016d..15addb0 100644
--- a/nn/functional/nn_relu.c
+++ b/nn/functional/nn_relu.c
@@ -9,7 +9,7 @@ void NN_relu(Tensor *y, Tensor *x) {
 
   switch (y->dtype) {
     case DTYPE_F16:
-      NN__maximum1_f16(y->size, (float *)y->data, 1, (float *)x->data, 1, 0.0f);
+      NN__maximum1_f16(y->size, (float16_t *)y->data, 1, (float16_t *)x->data, 1, 0.0f);
       return;
     case DTYPE_F32:
       NN__maximum1_f32(y->size, (float *)y->data, 1, (float *)x->data, 1, 0.0f);
diff --git a/nn/impl/minimum1.h b/nn/impl/minimum1.h
index b867ec1..327c256 100644
--- a/nn/impl/minimum1.h
+++ b/nn/impl/minimum1.h
@@ -4,6 +4,8 @@
 #include <stddef.h>
 #include <stdint.h>
 
+#include "nn_float16.h"
+
 
 void NN__minimum1_f32(size_t n,
     float *y, size_t incy,