From 5f321b3b8c7524831c0dd6d81a0fcb81930a1e16 Mon Sep 17 00:00:00 2001
From: JihaoXin <46529913+JihaoXin@users.noreply.github.com>
Date: Thu, 19 Oct 2023 17:26:09 +0800
Subject: [PATCH] Include stdio.h

---
 src/fastertransformer/utils/cuda_fp8_utils.h | 1 +
 src/fastertransformer/utils/cuda_utils.h     | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/fastertransformer/utils/cuda_fp8_utils.h b/src/fastertransformer/utils/cuda_fp8_utils.h
index 5b681171c..6b3b737c2 100644
--- a/src/fastertransformer/utils/cuda_fp8_utils.h
+++ b/src/fastertransformer/utils/cuda_fp8_utils.h
@@ -20,6 +20,7 @@
 #include <cuda_fp8.h>
 #include <cuda_runtime.h>
 #include <stdint.h>
+#include <stdio.h>
 
 // #define FP8_MHA
 #if defined(__CUDA_ARCH__) && __CUDA_ARCH__ == 900
diff --git a/src/fastertransformer/utils/cuda_utils.h b/src/fastertransformer/utils/cuda_utils.h
index 331b6f297..b8a02ab5f 100644
--- a/src/fastertransformer/utils/cuda_utils.h
+++ b/src/fastertransformer/utils/cuda_utils.h
@@ -27,6 +27,7 @@
 #include <iostream>
 #include <string>
 #include <vector>
+#include <stdio.h>
 #ifdef SPARSITY_ENABLED
 #include <cusparseLt.h>
 #endif