File size: 236 Bytes
0482f3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import torch
# Just so we use it
import nvidia_cutlass_dsl
from ._ops import ops
from .silu_and_mul import _silu_and_mul
def silu_and_mul(x: torch.Tensor) -> torch.Tensor:
return ops.silu_and_mul(x)
__all__ = ["silu_and_mul"]
|