From 19050f5c0d74b3a5fb4dd14a81a1ebd748f48cf2 Mon Sep 17 00:00:00 2001
From: pritamsoni-hsr <pritamsoni.hsr@gmail.com>
Date: Wed, 20 Nov 2019 02:47:25 +0530
Subject: [PATCH] dtype error on MacOS 10.14 fixed

---
 src/loadwordemb.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/loadwordemb.cpp b/src/loadwordemb.cpp
index 4cba8db..eedab5b 100644
--- a/src/loadwordemb.cpp
+++ b/src/loadwordemb.cpp
@@ -61,7 +61,7 @@ VocabAndTensor load_glove(const char * filename){
         dim    = get_tokens_in_first_line(fp) - 1;
         n_word = get_line_numbers(fp);
 
-        dest.resize_(torch::IntArrayRef{n_word, dim});
+        dest.resize_(torch::IntArrayRef{static_cast<long long>(n_word), static_cast<long long>(dim)});
 
         char buf[512];
         float * data = static_cast<float*>(dest.storage().data());
@@ -98,7 +98,7 @@ VocabAndTensor load_word2vec(const char * filename){
         size_t dim, n_word;
         fscanf(fp, "%zu %zu", &n_word, &dim);
 
-        dest.resize_(torch::IntArrayRef{n_word, dim});
+        dest.resize_(torch::IntArrayRef{static_cast<long long>(n_word), static_cast<long long>(dim)});
 
         char buf[512];
         float * data = static_cast<float*>(dest.storage().data());
@@ -136,7 +136,7 @@ VocabAndTensor load_word2vec_bin(const char * filename){
         fscanf(fp, "%zu %zu\n", &n_word, &dim);
         //printf("%zu %zu\n", n_word, dim );
 
-        dest.resize_(torch::IntArrayRef{n_word, dim});
+        dest.resize_(torch::IntArrayRef{static_cast<long long>(n_word), static_cast<long long>(dim)});
 
         char buf[512];
         float * data = static_cast<float*>(dest.storage().data());