Kernels:

drbh
/

first-kernel

Kernel card Files Files and versions

first-kernel / torch-ext /torch_binding.cpp

drbh

feat: push full template and build to repo

f9791fd 7 months ago

Raw History Blame Contribute Delete

324 Bytes

	#include <torch/library.h>

	#include "registration.h"
	#include "torch_binding.h"

	TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
	ops.def("first_kernel(Tensor! out, Tensor input) -> ()");
	#if defined(METAL_KERNEL)
	ops.impl("first_kernel", torch::kMPS, first_kernel);
	#endif
	}

	REGISTER_EXTENSION(TORCH_EXTENSION_NAME)