causal_conv1d#

causal_conv1d(x, weight, bias=None, activation='identity')#

Depthwise causal 1D conv via the subq_ops CUDA kernel.

Parameters:

x (Tensor) – Input tensor [B, C, L].
weight (Tensor) – Depthwise weight [C, K].
bias (Tensor | None) – Optional per-channel bias [C].
activation (str) – "identity" (default) or "silu".

Returns:

Output tensor [B, C, L] (same shape as input).

Return type: