diff options
| author | Tom Smeding <tom@tomsmeding.com> | 2025-04-15 17:18:34 +0200 |
|---|---|---|
| committer | Tom Smeding <tom@tomsmeding.com> | 2025-04-15 17:18:34 +0200 |
| commit | 9106993eeb2036f1dc5165535e1f2be77c273f59 (patch) | |
| tree | 94b117ab12346d6d9999db7114a41bf5ec027014 /cbits | |
| parent | 8210378510b92f8ec224c6adcda3ecc77625f1a0 (diff) | |
arith: Better typing of enum stats_binary_id
Diffstat (limited to 'cbits')
| -rw-r--r-- | cbits/arith.c | 8 |
1 files changed, 4 insertions, 4 deletions
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; |