Kgj:rSrSSKrSSKJr SSKJr \"S/S9SS\R S\R S-S\R S-S \R S-S \R S-S \S-S \S-S \S-S\R S-S\RS-S\RS-S\S-4Sjj5r g)z4Main interface for causal 1D convolution operations.N) FLACPContext) input_guardx)no_guard_contiguousweightbiasresidual initial_stateoutput_final_state activationbackend cu_seqlenscu_seqlens_cpu chunk_indices cp_contextc  bSSKJn SSKJnJn SSKJn U b%UbS5eUSLdS5eU "UUUUU U S 9nUS4$US :Xa!URUUUUUUUUU U 5 unnUU4$US :Xa!U RS 5nU"UUUUUUUUU U US 9 $US:XaU"UUUUUUUU4SU 0U D6$[SU35e)uv A causal 1D convolution implementation that powers Mamba/Mamba2 and DeltaNet architectures. When a residual connection is provided, this implements the Canon operation described in the paper at https://papers.ssrn.com/sol3/papers.cfm?abstract_id=5240330. Args: x (torch.Tensor): Input tensor of shape [B, T, D]. weight (Optional[torch.Tensor]): Weight tensor of shape [D, W]. Default: `None`. bias (Optional[torch.Tensor]): Bias tensor of shape [D]. Default: `None`. residual (Optional[torch.Tensor]): Residual tensor of shape [B, T, D]. Default: `None`. initial_state (Optional[torch.Tensor]): Initial state tensor of shape [N, D, W], where `N` is the number of sequences in the batch and `W` is the kernel size. If provided, the initial state is used to initialize the cache. Default: `None`. output_final_state (Optional[bool]): Whether to output the final state of shape [N, D, W]. Default: `False`. activation (Optional[str]): Activations applied to output, only `swish`/`silu` or `None` (i.e., no activation) are supported. Default: `None`. backend (Optional[str]): Specifies the backend to use for the convolution operation. Supported values are `'cuda'` 、 `'triton'` and `'mix'`. Default: `'triton'`. cu_seqlens (Optional[torch.Tensor]): Cumulative sequence lengths (optional) chunk_indices (Optional[torch.LongTensor]): Chunk indices for variable-length sequences (optional) Returns: Tuple of (output, final_state). If `output_final_state` is `False`, the final state is `None`. r)causal_conv1d_cp)causal_conv1d_cudafast_causal_conv1d_fn)CausalConv1dFunctionNz%Initial state is not supported for CPFz*Output final state is not supported for CP)rrrr rrtritonmixseq_idx)rrrcudarzUnsupported backend: ) fla.modules.conv.cprfla.modules.conv.cudarrfla.modules.conv.tritonrapplyget ValueError)rrrr r r r r rrrrkwargsrrrroutputy final_staters ڬC:\Users\johnn\AppData\Local\Packages\PythonSoftwareFoundation.Python.3.13_qbz5n2kfra8p0\LocalCache\local-packages\Python313\site-packages\fla/modules/conv/causal_conv1d.py causal_conv1dr&sDj5O<$M&MM$!U*X,XX*!!'!  t|(-33           ;+~ E **Y'$       )'   F !         *    0 :;;) NNNNFNrNNNN) __doc__torch fla.ops.cpr fla.utilsrTensorboolstr LongTensorr&r'r%r1s+; #! #'#' $$()-&+!"&*.2-1&*p< ||p< LL4 p< ,, p<llT! p< <<$& p< t p<d p<4Zp< t#p<$$t+p<##d*p<t#p<(p