diff --git a/include/flashinfer/attention/hopper/variants.cuh b/include/flashinfer/attention/hopper/variants.cuh index e11bd75d9..69ad56eeb 100644 --- a/include/flashinfer/attention/hopper/variants.cuh +++ b/include/flashinfer/attention/hopper/variants.cuh @@ -28,7 +28,7 @@ struct StandardAttention { template __device__ StandardAttention(const MainloopParams& params, const BlockCoord& block_coord) { - sm_scale_log2 = params.sm_scale * math::log2e; + sm_scale_log2 = params.additional_params.sm_scale * math::log2e; } template