From 7883bed5997f430219077202c84af7bf80ada2b7 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Tue, 18 Mar 2025 23:37:56 +0100 Subject: arith stats: Improve output format This makes it nicer to process using unix tools. Try: $ sed -n '/ox-arrays-arith-stats start/,/ox-arrays-arith-stats end/ !d; /===/ !p' | sort -n -k4,4 -k6,6 --- cbits/arith.c | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/cbits/arith.c b/cbits/arith.c index c3e34ad..c7bb0bf 100644 --- a/cbits/arith.c +++ b/cbits/arith.c @@ -139,11 +139,13 @@ static size_t stats_print_unary(uint8_t *buf) { i64 *strides = (i64*)buf; buf += rank * 8; double secs = *(double*)buf; buf += 8; - printf("unary %d sh=[", (int)id); + i64 shsize = 1; for (i32 i = 0; i < rank; i++) shsize *= shape[i]; + + printf("unary %d sz %" PRIi64 " ms %lf sh=[", (int)id, shsize, secs * 1000); for (i32 i = 0; i < rank; i++) { if (i > 0) putchar(','); printf("%" PRIi64, shape[i]); } printf("] str=["); for (i32 i = 0; i < rank; i++) { if (i > 0) putchar(','); printf("%" PRIi64, strides[i]); } - printf("] ms=%lf\n", secs * 1000); + printf("]\n"); return buf - orig_buf; } @@ -158,13 +160,15 @@ static size_t stats_print_binary(uint8_t *buf) { i64 *strides2 = (i64*)buf; buf += rank * 8; double secs = *(double*)buf; buf += 8; - printf("binary %d sh=[", (int)id); + i64 shsize = 1; for (i32 i = 0; i < rank; i++) shsize *= shape[i]; + + printf("binary %d sz %" PRIi64 " ms %lf sh=[", (int)id, shsize, secs * 1000); for (i32 i = 0; i < rank; i++) { if (i > 0) putchar(','); printf("%" PRIi64, shape[i]); } printf("] str1=["); for (i32 i = 0; i < rank; i++) { if (i > 0) putchar(','); printf("%" PRIi64, strides1[i]); } printf("] str2=["); for (i32 i = 0; i < rank; i++) { if (i > 0) putchar(','); printf("%" PRIi64, strides2[i]); } - printf("] ms=%lf\n", secs * 1000); + printf("]\n"); return buf - orig_buf; } -- cgit v1.2.3-70-g09d2