mirror of
https://github.com/saymrwulf/pytorch.git
synced 2026-05-14 20:57:59 +00:00
12 lines
174 B
C++
12 lines
174 B
C++
|
#include <torch/extension.h>
|
||
|
|
|
||
|
|
void foo() { }
|
||
|
|
|
||
|
|
TORCH_LIBRARY_IMPL(__test, CPU, m) {
|
||
|
|
m.impl("foo", foo);
|
||
|
|
}
|
||
|
|
|
||
|
|
PYBIND11_MODULE(TORCH_EXTENSION_NAME, m) {
|
||
|
|
m.def("bar", foo);
|
||
|
|
}
|