Kernels:

drbh
/

first-kernel

Kernel card Files Files and versions

first-kernel / torch-ext /first_kernel /__init__.py

drbh

feat: push full template and build to repo

f9791fd 7 months ago

Raw History Blame Contribute Delete

252 Bytes

	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