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)
#include "torch/all.h"
#include "torch/script.h"
static void TestPreemph() {

View File

@ -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_