Fix compilation.

This commit is contained in:
Fangjun Kuang 2021-10-11 17:01:43 +08:00
parent 22f2fb7371
commit dccc927428
2 changed files with 2 additions and 0 deletions

View File

@ -2,6 +2,7 @@
// //
// Copyright (c) 2021 Xiaomi Corporation (authors: Fangjun Kuang) // Copyright (c) 2021 Xiaomi Corporation (authors: Fangjun Kuang)
#include "torch/all.h"
#include "torch/script.h" #include "torch/script.h"
static void TestPreemph() { static void TestPreemph() {

View File

@ -6,6 +6,7 @@
#define KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_ #define KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_
#include "pybind11/pybind11.h" #include "pybind11/pybind11.h"
#include "torch/torch.h"
namespace py = pybind11; namespace py = pybind11;
#endif // KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_ #endif // KALDIFEAT_PYTHON_CSRC_KALDIFEAT_H_