Kernels:

---

drbh

/

first-kernel

Like 5

#pragma once

#include <torch/torch.h>

void first_kernel(torch::Tensor &out, torch::Tensor const &input);