Kernels:

drbh
/

first-kernel

drbh

feat: push full template and build to repo

f9791fd 7 months ago

107 Bytes

	#pragma once

	#include <torch/torch.h>

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