From a3dbbf8da541708e22b3be5a9ce0f54a5360a341 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@miranda.cs.illinois.edu>
Date: Tue, 22 Jun 2021 02:38:24 -0500
Subject: [PATCH] fixing tensorutils definitions to return void*
---
hpvm/include/nvdla/tensorUtils.h | 12 ++++++------
1 file changed, 6 insertions(+), 6 deletions(-)
diff --git a/hpvm/include/nvdla/tensorUtils.h b/hpvm/include/nvdla/tensorUtils.h
index 47441bae9d..ecfd109287 100644
--- a/hpvm/include/nvdla/tensorUtils.h
+++ b/hpvm/include/nvdla/tensorUtils.h
@@ -15,9 +15,9 @@ std::string model_params_path = "../../test/dnn_benchmarks/model_params/";
-__attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_name, int data_type,
- long int dim1_size, long int dim2_size,
- long int dim3_size, long int dim4_size) {
+__attribute__((noinline)) void *readTrainedWeights(const char *file_name, int data_type,
+ long int dim1_size, long int dim2_size,
+ long int dim3_size, long int dim4_size) {
int type_size = 4; // NOTE: Assuming floating point tensors
long int num_elems = dim1_size * dim2_size * dim3_size * dim4_size;
@@ -45,9 +45,9 @@ __attribute__((noinline)) struct Tensor *readTrainedWeights(const char *file_nam
}
-__attribute__((noinline)) struct Tensor *readInputBatch(const char *file_name, long data_type,
- long start, long end,
- long dim2_size, long dim3_size, long dim4_size) {
+__attribute__((noinline)) void * readInputBatch(const char *file_name, long data_type,
+ long start, long end,
+ long dim2_size, long dim3_size, long dim4_size) {
long int dim1_size = end - start;
// FIXIT: Don't assume floating point types
long int type_size = 4; // NOTE: Assuming floating point tensors
--
GitLab