From 24e612735b64ba5b771a8048a8e716ae85242faf Mon Sep 17 00:00:00 2001
From: Ilia Kuznetcov <diablomobile.modest@gmail.com>
Date: Fri, 20 Aug 2021 21:18:46 +0400
Subject: [PATCH] Fix clang compilation

---
 ctcdecode/src/binding.cpp       | 4 ++--
 ctcdecode/src/decoder_utils.cpp | 4 +++-
 2 files changed, 5 insertions(+), 3 deletions(-)

diff --git a/ctcdecode/src/binding.cpp b/ctcdecode/src/binding.cpp
index c14a8221..c42cd63f 100644
--- a/ctcdecode/src/binding.cpp
+++ b/ctcdecode/src/binding.cpp
@@ -201,8 +201,8 @@ std::pair<torch::Tensor, torch::Tensor> beam_decode_with_given_state(at::Tensor
         }
         }
     
-    torch::Tensor output_tokens_tensor = torch::randint(1, {batch_results.size(), max_result_size, max_output_tokens_size});
-    torch::Tensor output_timesteps_tensor = torch::randint(1, {batch_results.size(), max_result_size, max_output_tokens_size});
+    torch::Tensor output_tokens_tensor = torch::randint(1, {static_cast<long long>(batch_results.size()), max_result_size, max_output_tokens_size});
+    torch::Tensor output_timesteps_tensor = torch::randint(1, {static_cast<long long>(batch_results.size()), max_result_size, max_output_tokens_size});
 
 
     auto scores_accessor =  th_scores.accessor<float, 2>();
diff --git a/ctcdecode/src/decoder_utils.cpp b/ctcdecode/src/decoder_utils.cpp
index 2a35fe48..70a4a83a 100644
--- a/ctcdecode/src/decoder_utils.cpp
+++ b/ctcdecode/src/decoder_utils.cpp
@@ -3,7 +3,9 @@
 #include <algorithm>
 #include <cmath>
 #include <limits>
-#include <bits/stdc++.h>
+#include <vector>
+#include <unordered_map>
+
 using namespace std;