We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 3052944 commit 090b100Copy full SHA for 090b100
include/flashinfer/attention/hopper/variants.cuh
@@ -28,7 +28,7 @@ struct StandardAttention {
28
29
template <typename MainloopParams, typename BlockCoord>
30
__device__ StandardAttention(const MainloopParams& params, const BlockCoord& block_coord) {
31
- sm_scale_log2 = params.sm_scale * math::log2e;
+ sm_scale_log2 = params.additional_params.sm_scale * math::log2e;
32
}
33
34
template <int NUM_ROWS_PER_THREAD>
0 commit comments