Mac compatibility (#416)

Co-authored-by: Hang Zhang <[email protected]>
zhanghang1989 · Jun 2, 2024 · 96e71b0 · 96e71b0
1 parent c959dab
commit 96e71b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/encoding/lib/cpu/operator.h b/encoding/lib/cpu/operator.h
@@ -93,7 +93,7 @@ py::array_t<float> apply_transform(int H, int W, int C, py::array_t<float> img,
   auto ctm_buf = ctm.request();
 
   // printf("H: %d, W: %d, C: %d\n", H, W, C);
-  py::array_t<float> result{(unsigned long)img_buf.size};
+  py::array_t<float> result{static_cast<ssize_t>(img_buf.size)};
   auto res_buf = result.request();
 
   float *img_ptr = (float *)img_buf.ptr;