Fix compile error on ubuntu22.04_g++11.3.0 (#5434)

b1ec112e · Rhett Ying · GitHub · 1313067e · b1ec112e
Unverified Commit b1ec112e authored Mar 08, 2023 by Rhett Ying Committed by GitHub Mar 08, 2023
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/array/cuda/atomic.cuh src/array/cuda/atomic.cuh +2 -0

No files found.
--- a/src/array/cuda/atomic.cuh
+++ b/src/array/cuda/atomic.cuh
@@ -9,6 +9,8 @@
 #include <cuda_runtime.h>

 #include <cassert>
+#include <cstdint>
+#include <cstdio>

 #include "bf16.cuh"
 #include "fp16.cuh"