drbh
/

template

Model card Files Files and versions

template / torch-ext /torch_binding.h

drbh

feat: basic template

dda32b8 6 months ago

History Blame Contribute Delete

121 Bytes

	#pragma once

	#include <torch/torch.h>

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