#ifndef __MATRIX_ADD_H
#define __MATRIX_ADD_H
#include <torch/extension.h>
void matrix_add(torch::Tensor a, torch::Tensor b, torch::Tensor out);
#endif