fla/modules/convolution.py

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


from fla.modules.conv import (
    ImplicitLongConvolution,
    LongConvolution,
    PositionalEmbedding,
    ShortConvolution,
    causal_conv1d,
    fft_conv,
)
from fla.modules.conv.cp import CausalConv1dFunctionCP, causal_conv1d_cp
from fla.modules.conv.cuda import FastCausalConv1dFn, fast_causal_conv1d_fn
from fla.modules.conv.triton import (
    CausalConv1dFunction,
    causal_conv1d_bwd,
    causal_conv1d_fwd,
    causal_conv1d_update,
    causal_conv1d_update_states,
)

__all__ = [
    'CausalConv1dFunction',
    'CausalConv1dFunctionCP',
    'FastCausalConv1dFn',
    'ImplicitLongConvolution',
    'LongConvolution',
    'PositionalEmbedding',
    'ShortConvolution',
    'causal_conv1d',
    'causal_conv1d_bwd',
    'causal_conv1d_cp',
    'causal_conv1d_fwd',
    'causal_conv1d_update',
    'causal_conv1d_update_states',
    'fast_causal_conv1d_fn',
    'fft_conv',
]