Fix slope conversion to pass all tests

tenstorrent · Oct 18, 2024 · 32388ed · 32388ed
1 parent 26273af
commit 32388ed
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/common/inc/sfpu/ckernel_sfpu_relu.h b/common/inc/sfpu/ckernel_sfpu_relu.h
@@ -4,9 +4,10 @@
 
 #pragma once
 
-#include "ckernel_defs.h"
 #include "ckernel.h"
+#include "ckernel_defs.h"
 #include "noc_nonblocking_api.h"
+#include "ckernel_sfpu_converter.h"
 
 #include "sfpi.h"
 
@@ -20,8 +21,7 @@ namespace sfpu
 template <bool APPROXIMATION_MODE>
 inline void _calculate_lrelu_(const int iterations, uint slope)
 {
-    // SFPU microcode
-    vFloat s = s2vFloat16b(slope);
+    vFloat s = Converter::to_float(slope);
 
     #pragma GCC unroll 0
     for (int d = 0; d < iterations; d++) {