Cleanup lrelu (#44)

* #5424: Cleanup lrelu - wormhole * Fix slope conversion to pass all tests
tenstorrent · Oct 24, 2024 · 38df924 · 38df924
1 parent 56ec2f5
commit 38df924
Showing 1 changed file with 4 additions and 4 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"
 
@@ -17,11 +18,10 @@ namespace ckernel
 namespace sfpu
 {
 
-template <bool APPROXIMATION_MODE, int ITERATIONS>
+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++) {