diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp index 26c1164d32..a5e5830180 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_pipeline.hpp @@ -472,7 +472,7 @@ struct HstuAttentionWithSoftmaxFwdPipelineQRKSVS { // use the m_old[i] as the m-for-stablization if m_old[i] - m[i] >= -8.0f // and still keep the m-for-stablization in m[] - if(m_old[i_idx] - m[i_idx] >= -8.0f) + if(m_old[i_idx] - m[i_idx] >= type_convert(-8.0f)) m(i_idx) = m_old[i_idx]; sweep_tile_span(p_spans[number<1>{}], [&](auto idx1) { diff --git a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp index 2dd879231b..2b4cd4b4b4 100644 --- a/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp +++ b/example/ck_tile/18_hstu_attention/hstu_attention_with_softmax_fwd_trload_pipeline.hpp @@ -478,7 +478,7 @@ struct HstuAttentionWithSoftmaxFwdPipelineQRKSVSTrLoad { // use the m_old[i] as the m-for-stablization if m_old[i] - m[i] >= -8.0f // and still keep the m-for-stablization in m[] - if(m_old[i_idx] - m[i_idx] >= -8.0f) + if(m_old[i_idx] - m[i_idx] >= type_convert(-8.0f)) m(i_idx) = m_old[i_idx]; sweep_tile_span(p_spans[number<1>{}], [&](auto idx1) {