From ab92cad01b33efb309e30f1409246142b074eef0 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Fri, 29 Mar 2019 02:32:54 -0500
Subject: [PATCH] Updating pipeline tuner file
---
.../dnn_sources/src/pipeline_GEOM.cc | 18 ++++++++++--------
1 file changed, 10 insertions(+), 8 deletions(-)
diff --git a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
index dd9298aa7c..677411fce0 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/src/pipeline_GEOM.cc
@@ -27,14 +27,19 @@ void testPipeline(){
// FIXIT: Extend this to batch of images - currently 5 images
//long int test_batch_size = 9145;
- long int test_batch_size = 2000;
+ long int test_batch_size = 4572;
long int H = 240;
long int W = 300;
printf("Reading input\n");
- void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
+ /*void* input = readTrainedWeights("../model_params/pipeline/dataset/caltech101_255_float32.bin",
float_type,
- test_batch_size, 1, H, W);
+ test_batch_size, 1, H, W); */
+
+ void* input = readTrainedWeights("../model_params/pipeline/dataset/calibration_4572.bin",
+ float_type,
+ test_batch_size, 1, H, W);
+
printf("Reading golden output\n");
void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/GEOM-2000.bin",
//void* golden_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
@@ -114,12 +119,9 @@ void testPipeline(){
// NOTE-IMP: Important to include this call always before doing dumpOutput and computePSNViolation
hpvm_request_tensor(result, 0);
- dumpOutput(result);
+ dumpOutput(result, "GEOM_calib.bin");
- //void* psnr_output = readTrainedWeights("../model_params/pipeline/golden_output/caltech-gaussian.bin",
- //float_type,
- //test_batch_size, 1, H, W);
- computePSNRViolation(result, golden_output, 30);
+ //computePSNRViolation(result, golden_output, 30);
dumpAccuracyNorms();
--
GitLab