diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h b/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h
index 7eb2fa8de36930121b3cb2c1f080595abad7873b..d082eac06967fdcc743b4befcf13484d6bdb100a 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h
@@ -336,7 +336,7 @@ struct Tensor* readInputBatch(const char* file_name, int data_type,
   long int num_elems = dim1_size * dim2_size * dim3_size * dim4_size;
   long int size_in_bytes = type_size * dim1_size * dim2_size * dim3_size * dim4_size;
   float* tensor_data = (float*) malloc(sizeof(float) * num_elems);
-  int file_header_size = type_size * start * dim2_size * dim3_size * dim4_size;
+  long int file_header_size = type_size * start * dim2_size * dim3_size * dim4_size;
   
   FILE* file = fopen(file_name, "rb");
   if(file == NULL){