torch-ext/torch_bindings.h

#pragma once

#include <torch/torch.h>

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