sglang_v0.5.2/pytorch_2.8.0/test/cpp_extensions/jit_extension2.cpp

8 lines
118 B
C++

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