Coverage for src/flag_gems/fused/FLA/__init__.py: 100%

1# This file contains code copied from the flash-linear-attention project. 

2# The original source code was licensed under the MIT license and included 

3# the following copyright notice: 

4# Copyright (c) 2023-2025, Songlin Yang, Yu Zhang 

5from flag_gems.fused.FLA.chunk import chunk_gated_delta_rule_fwd 

6from flag_gems.fused.FLA.fused_recurrent import fused_recurrent_gated_delta_rule_fwd 

7 

8__all__ = [ 

9 "chunk_gated_delta_rule_fwd", 

10 "fused_recurrent_gated_delta_rule_fwd", 

11]