Support torch 1.7.1

.github/workflows/publish_to_pypi.yml

@@ -20,6 +20,8 @@ on:
   push:
     tags:
       - '*'
+    branches:
+      - torch-1.7.1
 
 jobs:
   pypi:
@@ -29,7 +31,7 @@ jobs:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
         cuda: ["10.1"]
-        torch: ["1.8.1"]
+        torch: ["1.7.1"]
         python-version: [3.6, 3.7, 3.8]
     steps:
       - uses: actions/checkout@v2

.github/workflows/run-tests.yml

@@ -32,7 +32,7 @@ jobs:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
         cuda: ["10.1"]
-        torch: ["1.8.1"]
+        torch: ["1.7.1"]
         python-version: [3.6, 3.7, 3.8]
     steps:
       - uses: actions/checkout@v2

.github/workflows/style_check.yml

@@ -30,7 +30,7 @@ jobs:
     strategy:
       matrix:
         os: [ubuntu-18.04, macos-10.15]
-        python-version: [3.7, 3.9]
+        python-version: [3.7, 3.8, 3.9]
       fail-fast: false
 
     steps:

CMakeLists.txt

@@ -4,7 +4,7 @@ cmake_minimum_required(VERSION 3.8 FATAL_ERROR)
 
 project(kaldifeat)
 
-set(kaldifeat_VERSION "1.4")
+set(kaldifeat_VERSION "1.5.1.dev")
 
 set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")
 set(CMAKE_LIBRARY_OUTPUT_DIRECTORY "${CMAKE_BINARY_DIR}/lib")

kaldifeat/csrc/feature-mfcc.cc

@@ -7,6 +7,7 @@
 #include "kaldifeat/csrc/feature-mfcc.h"
 
 #include "kaldifeat/csrc/matrix-functions.h"
+#include "torch/fft.h"
 
 namespace kaldifeat {
 

kaldifeat/csrc/feature-plp.cc

@@ -7,6 +7,7 @@
 #include "kaldifeat/csrc/feature-plp.h"
 
 #include "kaldifeat/csrc/feature-functions.h"
+#include "torch/fft.h"
 #include "torch/torch.h"
 
 namespace kaldifeat {

kaldifeat/csrc/feature-spectrogram.cc

@@ -6,6 +6,8 @@
 
 #include "kaldifeat/csrc/feature-spectrogram.h"
 
+#include "torch/fft.h"
+
 namespace kaldifeat {
 
 std::ostream &operator<<(std::ostream &os, const SpectrogramOptions &opts) {