From 85ef70381ae2ff51cf401768638996f19acbe670 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Tue, 2 Feb 2021 16:54:53 -0600
Subject: [PATCH] Fixing Integer overflow bug in Input reading routine
---
hpvm/test/dnn_benchmarks/include/tensorUtils.h | 10 +++++-----
1 file changed, 5 insertions(+), 5 deletions(-)
diff --git a/hpvm/test/dnn_benchmarks/include/tensorUtils.h b/hpvm/test/dnn_benchmarks/include/tensorUtils.h
index a627f83e6b..71e1c26872 100644
--- a/hpvm/test/dnn_benchmarks/include/tensorUtils.h
+++ b/hpvm/test/dnn_benchmarks/include/tensorUtils.h
@@ -318,13 +318,13 @@ struct Tensor* readInputBatch(const char* file_name, int data_type,
int start, int end,
int dim2_size, int dim3_size, int dim4_size){
- int dim1_size = end - start;
+ long int dim1_size = end - start;
// FIXIT: Don't assume floating point types
- int type_size = 4; // NOTE: Assuming floating point tensors
+ long int type_size = 4; // NOTE: Assuming floating point tensors
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){
@@ -335,9 +335,9 @@ struct Tensor* readInputBatch(const char* file_name, int data_type,
fseek(file, file_header_size, SEEK_SET); // Skipping the file header
size_t bytes_read = fread(tensor_data, 1, size_in_bytes, file);
-
fclose(file);
-
+
+ //printf ("FIXED input BATCH read \n");
struct Tensor* weights = (struct Tensor*) create4DTensor(data_type, nchw, dim1_size, dim2_size,
dim3_size, dim4_size);
--
GitLab