diff --git a/hpvm/test/dnn_benchmarks/include/tensorUtils.h b/hpvm/test/dnn_benchmarks/include/tensorUtils.h index a627f83e6b2aa9f38b09d82ee94ce35da1a6bafe..71e1c268726e1fb77b0713599928262b95bd64f5 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);