From a389c38e3d375714192f02b5d3f1aff9d877b8af Mon Sep 17 00:00:00 2001
From: chengscott <60510scott@gmail.com>
Date: Wed, 1 May 2024 18:27:40 +0800
Subject: [PATCH] include cstdint for int*_t

---
 python/aitemplate/backend/cuda/conv2d/common.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/python/aitemplate/backend/cuda/conv2d/common.py b/python/aitemplate/backend/cuda/conv2d/common.py
index c6bfbdb8b..76822e244 100644
--- 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) {