From 9a1a8d3bdec842f5ab81dd0f39c16b19559ab189 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Fri, 13 Mar 2020 19:02:18 -0500
Subject: [PATCH] Fixing remaining overflow bug
---
llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
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 7eb2fa8de3..d082eac069 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){
--
GitLab