Skip to content

Commit 090b100

Browse files
authored
hotfix: follow up of #772 (#773)
Some commits are missing in #772 , this PR fix the issue.
1 parent 3052944 commit 090b100

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

include/flashinfer/attention/hopper/variants.cuh

+1-1
Original file line numberDiff line numberDiff line change
@@ -28,7 +28,7 @@ struct StandardAttention {
2828

2929
template <typename MainloopParams, typename BlockCoord>
3030
__device__ StandardAttention(const MainloopParams& params, const BlockCoord& block_coord) {
31-
sm_scale_log2 = params.sm_scale * math::log2e;
31+
sm_scale_log2 = params.additional_params.sm_scale * math::log2e;
3232
}
3333

3434
template <int NUM_ROWS_PER_THREAD>

0 commit comments

Comments
 (0)