Skip to content

Commit 9f56344

Browse files
committed
[RELEASE ONLY CHANGES] Supply the FLA graph flag during export
PyTorch 2.15 wrap_triton export requires the cumulative-sum USE_GRAPH argument explicitly instead of relying on the eager Triton default. Authored with Codex assistance.
1 parent c2da612 commit 9f56344

1 file changed

Lines changed: 1 addition & 0 deletions

File tree

‎backends/cuda/triton/kernels/chunk_gated_delta_rule.py‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -128,6 +128,7 @@ def chunk_gated_delta_rule(
128128
REVERSE=False,
129129
HAS_SCALE=True,
130130
IS_VARLEN=False,
131+
USE_GRAPH=False,
131132
)
132133

133134
# 2. chunk_scaled_dot_kkt: compute beta * K * K^T with gating

0 commit comments

Comments
 (0)