Add BUILD_BENCH option for printing the PoW-related info

DLTcollab · Sep 22, 2018 · 5b71fe3 · 5b71fe3
1 parent 3703822
commit 5b71fe3
Show file tree

Hide file tree

Showing 7 changed files with 17 additions and 0 deletions.
diff --git a/Makefile b/Makefile
@@ -47,6 +47,10 @@ ifeq ("$(BUILD_JNI)","1")
 include mk/java.mk
 endif
 
+ifeq ("$(BUILD_BENCH)","1")
+CFLAGS += -DENABLE_BENCH
+endif
+
 TESTS = \
 	trinary \
 	curl \

diff --git a/mk/defs.mk b/mk/defs.mk
@@ -15,3 +15,6 @@ BUILD_COMPAT ?= 0
 
 # Build FPGA backend or not
 BUILD_FPGA_ACCEL ?= 0
+
+# Show the PoW-related messages or not
+BUILD_BENCH ?= 0
diff --git a/src/pow_avx.c b/src/pow_avx.c
@@ -592,8 +592,10 @@ bool PowAVX(void *pow_ctx)
 
     nonce_to_result(tx_tryte, nonce_tryte, ctx->output_trytes);
 
+#if defined(ENABLE_BENCH)
     /* PoW-related information */
     printf("Hash count: %"PRIu64"\n", ctx->hash_count);
+#endif
 
 fail:
     /* Free memory */

diff --git a/src/pow_c.c b/src/pow_c.c
@@ -353,8 +353,10 @@ bool PowC(void *pow_ctx)
 
     nonce_to_result(tx_tryte, nonce_tryte, ctx->output_trytes);
 
+#if defined(ENABLE_BENCH)
     /* PoW-related information */
     printf("Hash count: %"PRIu64"\n", ctx->hash_count);
+#endif
 
 fail:
     /* Free memory */

diff --git a/src/pow_cl.c b/src/pow_cl.c
@@ -233,8 +233,10 @@ bool PowCL(void *pow_ctx)
 
     ctx->hash_count = ctx->clctx->hash_count;
 
+#if defined(ENABLE_BENCH)
     /* PoW-related information */
     printf("Hash count: %"PRIu64"\n", ctx->hash_count);
+#endif
 
 fail:
     freeTrobject(tx_trit);

diff --git a/src/pow_fpga_accel.c b/src/pow_fpga_accel.c
@@ -92,8 +92,10 @@ static bool PoWFPGAAccel(void *pow_ctx)
     memcpy(ctx->output_trytes + ((NonceTrinaryOffset) / 3), nonce_trytes->data,
            ((transactionTrinarySize) - (NonceTrinaryOffset)) / 3);
 
+#if defined(ENABLE_BENCH)
     /* PoW-related information */
     printf("Hash count: %"PRIu64"\n", ctx->hash_count);
+#endif
 
 fail:
     freeTrobject(object_tryte);

diff --git a/src/pow_sse.c b/src/pow_sse.c
@@ -369,8 +369,10 @@ bool PowSSE(void *pow_ctx)
 
     nonce_to_result(tx_tryte, nonce_tryte, ctx->output_trytes);
 
+#if defined(ENABLE_BENCH)
     /* PoW-related information */
     printf("Hash count: %"PRIu64"\n", ctx->hash_count);
+#endif
 
 fail:
     /* Free memory */