Add BUILD_BENCH option for printing the PoW-related info

DLTcollab · Sep 23, 2018 · 22dc766 · 22dc766
1 parent c5dae88
commit 22dc766
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 1 deletion.
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/tests/common.h b/tests/common.h
@@ -19,8 +19,9 @@
 #include "pow_cl.h"
 #endif
 
-#include <stdlib.h>
+#include <inttypes.h>
 #include <stdint.h>
+#include <stdlib.h>
 
 #ifdef NDEBUG
 #undef NDEBUG

diff --git a/tests/test-pow.c b/tests/test-pow.c
@@ -112,6 +112,9 @@ int main()
         doThePoW(PoW_Context_ptr, pow_ctx);
         int8_t *ret_trytes = getPoWResult(PoW_Context_ptr, pow_ctx);
         assert(ret_trytes);
+#if defined(ENABLE_BENCH)
+        printf("Hash count: %"PRIu64"\n", getHashCount(PoW_Context_ptr, pow_ctx));
+#endif
         freePoWContext(PoW_Context_ptr, pow_ctx);
         destroyImplContext(PoW_Context_ptr);