pytorch/jit_extension2.cpp at 22fe542b8e99679f44386a9327b6f11dde336b34 - pytorch - Carlos Sousa's Git

OSSForks/pytorch

mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00

Peter Goldsborough 22fe542b8e Use TORCH_EXTENSION_NAME macro to avoid mismatched module/extension name (#5277 )

* Warn users about mismatched module/extension name

* Define TORCH_EXTENSION_NAME macro

2018-02-16 22:31:04 -05:00

8 lines

114 B

C++

Raw Blame History

 #include <torch/torch.h>
 using namespace at;
 Tensor exp_add(Tensor x, Tensor y) {
   return x.exp() + y.exp();
 }