include cstdint for int*_t

facebookincubator · May 12, 2024 · a389c38 · a389c38
1 parent d158982
commit a389c38
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python/aitemplate/backend/cuda/conv2d/common.py b/python/aitemplate/backend/cuda/conv2d/common.py
@@ -364,11 +364,14 @@
 
 PROFILER_MAIN_TEMPLATE = jinja2.Template(
     """
+#include <cstdint>
 #include <iostream>
 #include <string>
 
 #include "cutlass/cutlass.h"
 
+using std::int8_t;
+
 {{benchmark_decls}}
 
 int main(int argc, char** argv) {