Fix compilation.

csukuangfj · Oct 11, 2021 · dccc927 · dccc927
1 parent 22f2fb7
commit dccc927
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/kaldifeat/csrc/test_kaldifeat.cc b/kaldifeat/csrc/test_kaldifeat.cc
@@ -2,6 +2,7 @@
 //
 // Copyright (c)  2021  Xiaomi Corporation (authors: Fangjun Kuang)
 
+#include "torch/all.h"
 #include "torch/script.h"
 
 static void TestPreemph() {

diff --git a/kaldifeat/python/csrc/kaldifeat.h b/kaldifeat/python/csrc/kaldifeat.h
@@ -6,6 +6,7 @@
 #define KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_
 
 #include "pybind11/pybind11.h"
+#include "torch/torch.h"
 namespace py = pybind11;
 
 #endif  // KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_