From 9106993eeb2036f1dc5165535e1f2be77c273f59 Mon Sep 17 00:00:00 2001 From: Tom Smeding Date: Tue, 15 Apr 2025 17:18:34 +0200 Subject: arith: Better typing of enum stats_binary_id --- cbits/arith.c | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'cbits/arith.c') diff --git a/cbits/arith.c b/cbits/arith.c index a6de4c9..f19b01e 100644 --- a/cbits/arith.c +++ b/cbits/arith.c @@ -109,7 +109,7 @@ try_again: ; } __attribute__((unused)) -static void stats_record_unary(uint8_t id, i32 rank, const i64 *shape, const i64 *strides, double secs) { +static void stats_record_unary(enum stats_binary_id id, i32 rank, const i64 *shape, const i64 *strides, double secs) { if (!atomic_load(&stats_enabled)) return; uint8_t *buf = stats_alloc(1 + 1 + 4 + 2*rank*8 + 8); *buf = 1; buf += 1; @@ -121,7 +121,7 @@ static void stats_record_unary(uint8_t id, i32 rank, const i64 *shape, const i64 } __attribute__((unused)) -static void stats_record_binary(uint8_t id, i32 rank, const i64 *shape, const i64 *strides1, const i64 *strides2, double secs) { +static void stats_record_binary(enum stats_binary_id id, i32 rank, const i64 *shape, const i64 *strides1, const i64 *strides2, double secs) { if (!atomic_load(&stats_enabled)) return; uint8_t *buf = stats_alloc(1 + 1 + 4 + 3*rank*8 + 8); *buf = 2; buf += 1; @@ -144,7 +144,7 @@ static void stats_record_binary(uint8_t id, i32 rank, const i64 *shape, const i6 static size_t stats_print_unary(uint8_t *buf) { uint8_t *orig_buf = buf; - uint8_t id = *buf; buf += 1; + enum stats_binary_id id = *buf; buf += 1; i32 rank = *(i32*)buf; buf += 4; i64 *shape = (i64*)buf; buf += rank * 8; i64 *strides = (i64*)buf; buf += rank * 8; @@ -164,7 +164,7 @@ static size_t stats_print_unary(uint8_t *buf) { static size_t stats_print_binary(uint8_t *buf) { uint8_t *orig_buf = buf; - uint8_t id = *buf; buf += 1; + enum stats_binary_id id = *buf; buf += 1; i32 rank = *(i32*)buf; buf += 4; i64 *shape = (i64*)buf; buf += rank * 8; i64 *strides1 = (i64*)buf; buf += rank * 8; -- cgit v1.2.3-70-g09d2