| 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"] | |
| 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"] | |