From 917df77946750ff96ad6a67ab58bd2552c37501b Mon Sep 17 00:00:00 2001
From: Fangjun Kuang <csukuangfj@gmail.com>
Date: Sat, 27 Feb 2021 00:33:48 +0800
Subject: [PATCH] add benchmark code.

It's about 3x faster than kaldi for a wave with duration 320 seconds on
CPU.
---
 kaldifeat/python/csrc/kaldifeat.cc | 14 ++++++++++++++
 kaldifeat/python/tests/test.py     |  6 +++---
 2 files changed, 17 insertions(+), 3 deletions(-)

diff --git a/kaldifeat/python/csrc/kaldifeat.cc b/kaldifeat/python/csrc/kaldifeat.cc
index b7c8ac9..77641bc 100644
--- a/kaldifeat/python/csrc/kaldifeat.cc
+++ b/kaldifeat/python/csrc/kaldifeat.cc
@@ -4,6 +4,8 @@
 
 #include "kaldifeat/python/csrc/kaldifeat.h"
 
+#include <chrono>
+
 #include "kaldifeat/csrc/feature-fbank.h"
 #include "torch/torch.h"
 
@@ -19,7 +21,19 @@ PYBIND11_MODULE(_kaldifeat, m) {
     Fbank fbank(fbank_opts);
     float vtln_warp = 1.0f;
 
+    std::chrono::steady_clock::time_point begin =
+        std::chrono::steady_clock::now();
+
     torch::Tensor ans = fbank.ComputeFeatures(tensor, vtln_warp);
+    std::chrono::steady_clock::time_point end =
+        std::chrono::steady_clock::now();
+    std::cout << "Time difference = "
+              << std::chrono::duration_cast<std::chrono::microseconds>(end -
+                                                                       begin)
+                         .count() /
+                     1000000.
+              << "[s]" << std::endl;
+
     return ans;
   });
 }
diff --git a/kaldifeat/python/tests/test.py b/kaldifeat/python/tests/test.py
index f8953d8..1b8c922 100755
--- a/kaldifeat/python/tests/test.py
+++ b/kaldifeat/python/tests/test.py
@@ -22,9 +22,9 @@ def main():
         a *= 32768
         tensor = torch.from_numpy(a)
         ans = _kaldifeat.test(tensor)
-        torch.set_printoptions(profile="full")
-        print(ans.shape)
-        print(ans)
+        #  torch.set_printoptions(profile="full")
+        #  print(ans.shape)
+        #  print(ans)
 
 
 if __name__ == '__main__':