diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_relu.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_relu.h index e6facb348149..08bfb4411dda 100644 --- a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_relu.h +++ b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/ckernel_sfpu_relu.h @@ -54,26 +54,9 @@ inline void relu_max(uint uint_threshold) } template -inline void calculate_lrelu(uint slope) +inline void calculate_lrelu(const uint slope, const int iterations) { - // SFPU microcode - Converter c_slope; - c_slope.u = slope; - vFloat s = c_slope.f; - - #pragma GCC unroll 0 - for (int d = 0; d < 8; d++) { - vFloat v = dst_reg[0]; - - v_if (v < 0.0f) { - v *= s; - } - v_endif; - - dst_reg[0] = v; - - dst_reg++; - } + _calculate_lrelu_(iterations, slope); } } // namespace sfpu diff --git a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_relu.h b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_relu.h index 6e4589e0836c..f2a61c199190 100644 --- a/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_relu.h +++ b/tt_metal/hw/ckernels/wormhole_b0/metal/llk_api/llk_sfpu/llk_math_eltwise_unary_sfpu_relu.h @@ -33,13 +33,14 @@ inline void llk_math_eltwise_unary_sfpu_relu_min_init() { } -template -inline void llk_math_eltwise_unary_sfpu_lrelu(uint dst_index, uint param0 = 0) { +template +inline void llk_math_eltwise_unary_sfpu_lrelu(uint dst_index, uint param0 = 0, uint param1 = ITERATIONS) { llk_math_eltwise_unary_sfpu_params( ckernel::sfpu::calculate_lrelu, dst_index, (int)VectorMode::RC, - param0); + param0, + param1); } template diff --git a/tt_metal/third_party/tt_llk_wormhole_b0 b/tt_metal/third_party/tt_llk_wormhole_b0 index 56ec2f52c817..26273afd4458 160000 --- a/tt_metal/third_party/tt_llk_wormhole_b0 +++ b/tt_metal/third_party/tt_llk_wormhole_b0 @@ -1 +1 @@ -Subproject commit 56ec2f52c817053116058c7ca4ed22da11c236ae +Subproject commit 26273afd4458855360a30db9fb49ce1149f550ff