Skip to content

Commit 06d867a

Browse files
hanli0612facebook-github-bot
authored andcommitted
Cleanup tl.constexpr HAS_ATTN_SCALE (#2531)
Summary: X-link: meta-recsys/generative-recommenders#110 Pull Request resolved: #2531 Reviewed By: LinjianMa Differential Revision: D65366330 fbshipit-source-id: 43be8c6a04a8751749c35cd18f79a92046e5d36b
1 parent c64ed1e commit 06d867a

File tree

1 file changed

+1
-2
lines changed
  • torchbenchmark/operators/ragged_attention

1 file changed

+1
-2
lines changed

torchbenchmark/operators/ragged_attention/hstu.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -117,8 +117,7 @@ def forward(
117117
"ALLOW_TF32": True,
118118
"BLOCK_D_Q": DimQ,
119119
"BLOCK_D_V": DimV,
120-
"max_attn_len": 0,
121-
"HAS_MAX_ATTN_LEN": False,
120+
"MAX_ATTN_LEN": 0,
122121
"sort_by_length_indices": False,
123122
"AUTOTUNE_MAX_SEQ_LEN": N,
124123
"contextual_seq_len": 0,

0 commit comments

Comments
 (0)