torch-ext/torch_binding.h

#include <torch/torch.h>

torch::Tensor mps_relu(const torch::Tensor &input);