From d869308b1106941114bcd0180d26e15301a9f035 Mon Sep 17 00:00:00 2001
From: Hashim Sharif <hsharif3@tyler.cs.illinois.edu>
Date: Fri, 29 Mar 2019 02:54:05 -0500
Subject: [PATCH] Updating dumpOutput in utils
---
llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
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 27d777d7ba..2aa97bc4d9 100644
--- a/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h
+++ b/llvm/projects/hpvm-tensor-rt/dnn_sources/include/utils.h
@@ -586,14 +586,14 @@ float computePSNRViolation(void* gold_ptr, void* approx_ptr, float PSNR_threshol
}
-void dumpOutput(void* output_ptr){
+void dumpOutput(void* output_ptr, const char* file_name){
struct Tensor* out_tensor = (struct Tensor*) output_ptr;
size_t size_in_bytes = out_tensor->size_in_bytes;
printf ("** Output size = %lu \n", size_in_bytes);
float* host_data = (float*) out_tensor->host_data;
- FILE* fd = fopen("output.bin", "w+");
+ FILE* fd = fopen(file_name, "w+");
fwrite(host_data, 1, size_in_bytes, fd);
fclose(fd);
}
--
GitLab