From 23f534691db5e1af6835511ec129dfc3bd850504 Mon Sep 17 00:00:00 2001
From: Han Li <hanli@meta.com>
Date: Wed, 20 Nov 2024 12:21:30 -0800
Subject: [PATCH] Cleanup tl.constexpr HAS_CONTEXTUAL_SEQ_LEN (#139)

Summary: Pull Request resolved: https://github.com/facebookresearch/generative-recommenders/pull/139

Reviewed By: zhaozhul, LinjianMa

Differential Revision: D66220248

fbshipit-source-id: 94837917fedf0c8b0a64af0f802a96edbfe19ade
---
 tritonbench/operators/ragged_attention/hstu.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/tritonbench/operators/ragged_attention/hstu.py b/tritonbench/operators/ragged_attention/hstu.py
index 7059f057..896c5ee6 100644
--- 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,
         }