torch-ext/torch_binding.h

#pragma once

#include <torch/torch.h>

torch::Tensor rmsnorm_forward(torch::Tensor &input, torch::Tensor &gamma);