Adding mutex.h for TE pytorch extension compilation (#26)

std::unique_lock requires <mutex>
ROCm · May 23, 2024 · 3f89e3a · 3f89e3a
1 parent 6035fb4
commit 3f89e3a
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/v2src/triton_kernel.cc b/v2src/triton_kernel.cc
@@ -5,6 +5,7 @@
 #include <aotriton/runtime.h>
 #include <incbin.h>
 #include <iostream>
+#include <mutex>
 #if AOTRITON_USE_ZSTD
 #include <zstd.h>
 #endif