torch_ext/matrix_add.h

5 lines
151 B
C
Raw Normal View History

2024-11-16 19:26:54 +08:00
#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