5 lines
151 B
C
5 lines
151 B
C
#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 |