arith: Better typing of enum stats_binary_idHEADmaster

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;