Update HSTU pin. (#294)

htyu · web-flow · commit 608f961fb1e8 · 2025-07-18T15:20:22.000-04:00
diff --git a/submodules/generative-recommenders b/submodules/generative-recommenders
@@ -1 +1 @@
-Subproject commit 21ceb7e001a0a3f8235efc0c11c9e75cec498bd4
+Subproject commit 88512dbd71b053226bc4ef8ec1630e3db53e55e5
diff --git a/tritonbench/operators/ragged_attention/operator.py b/tritonbench/operators/ragged_attention/operator.py
@@ -87,6 +87,7 @@ def hstu(self, q, k, v, seq_offsets, num_targets, max_seq_len):
             max_attn_len=self.max_attn_len,
             contextual_seq_len=self.contextual_seq_len,
             sort_by_length=True,
+            enable_tma=True,
         )
 
     # TODO: remove B200 hacks like these.

Original file line number	Diff line number	Diff line change
`@@ -87,6 +87,7 @@ def hstu(self, q, k, v, seq_offsets, num_targets, max_seq_len):`
`87`	`87`	`max_attn_len=self.max_attn_len,`
`88`	`88`	`contextual_seq_len=self.contextual_seq_len,`
`89`	`89`	`sort_by_length=True,`
	`90`	`+ enable_tma=True,`
`90`	`91`	`)`
`91`	`92`
`92`	`93`	`# TODO: remove B200 hacks like these.`