Add windows typedef for ssize_t (#368)

triton-inference-server · Jun 17, 2024 · 6c17c1f · 6c17c1f
1 parent ebc8c6c
commit 6c17c1f
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/pb_tensor.cc b/src/pb_tensor.cc
@@ -35,6 +35,11 @@ namespace py = pybind11;
 #endif
 #include "pb_tensor.h"
 
+// WAR for undefined ssize_t on Windows: https://stackoverflow.com/a/35368387
+#if defined(_MSC_VER)
+#include <BaseTsd.h>
+typedef SSIZE_T ssize_t;
+#endif
 
 namespace triton { namespace backend { namespace python {