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