Skip to content

Commit 4bbff44

Browse files
committed
Add PyInit__torchaudio
1 parent 5e02d10 commit 4bbff44

1 file changed

Lines changed: 10 additions & 0 deletions

File tree

src/libtorchaudio/_torchaudio.cpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
#include <libtorchaudio/utils.h>
22
#include <torch/csrc/stable/library.h>
33

4+
#include <Python.h>
5+
46
STABLE_TORCH_LIBRARY(_torchaudio, m) {
57
m.def("is_align_available() -> bool");
68
m.def("cuda_version() -> int?");
@@ -10,3 +12,11 @@ STABLE_TORCH_LIBRARY_IMPL(_torchaudio, CompositeExplicitAutograd, m) {
1012
m.impl("is_align_available", TORCH_BOX(&torchaudio::is_align_available));
1113
m.impl("cuda_version", TORCH_BOX(&torchaudio::cuda_version));
1214
}
15+
16+
static struct PyModuleDef _torchaudio_module = {
17+
.m_base = PyModuleDef_HEAD_INIT,
18+
.m_name = "_torchaudio"};
19+
20+
PyMODINIT_FUNC PyInit__torchaudio(void) {
21+
return PyModuleDef_Init(&_torchaudio_module);
22+
}

0 commit comments

Comments
 (0)