diff options
| -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; |