jit_extension2.cpp

#include <torch/extension.h>

using namespace at;

Tensor exp_add(Tensor x, Tensor y) {
  return x.exp() + y.exp();
}