danieldk's picture
danieldk HF Staff
Add Python invalid dependency test kernel
bb1e912
import torch
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"]