Kernels:

---

drbh

/

first-kernel

like 5

from typing import Optional

import torch

from ._ops import ops


def first_kernel(x: torch.Tensor, out: Optional[torch.Tensor] = None) -> torch.Tensor:
    if out is None:
        out = torch.empty_like(x)
    ops.first_kernel(out, x)
    return out