FIX: minor fix

ucb-bar · Jul 11, 2024 · 79b7224 · 79b7224
1 parent aa71c8a
commit 79b7224
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git 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
@@ -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,