Cleanup tl.constexpr HAS_CONTEXTUAL_SEQ_LEN (#139)

Summary: Pull Request resolved: facebookresearch/generative-recommenders#139 Reviewed By: zhaozhul, LinjianMa Differential Revision: D66220248 fbshipit-source-id: 94837917fedf0c8b0a64af0f802a96edbfe19ade
pytorch-labs · Nov 20, 2024 · 23f5346 · 23f5346
1 parent 6091289
commit 23f5346
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tritonbench/operators/ragged_attention/hstu.py b/tritonbench/operators/ragged_attention/hstu.py
@@ -140,8 +140,7 @@ def forward(
             "BLOCK_D_Q": DimQ,
             "BLOCK_D_V": DimV,
             "MAX_ATTN_LEN": 0,
-            "HAS_CONTEXTUAL_SEQ_LEN": False,
-            "contextual_seq_len": 0,
+            "CONTEXTUAL_SEQ_LEN": 0,
             "HAS_SORT_BY_LENGTH_INDICES": False,
             "sort_by_length_indices": None,
         }