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 27d777d7ba52b5bf85316d5121c598c329ecde13..2aa97bc4d91b616e2fbe2e6c73ef3fa996fa5b35 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); }