diff --git a/taichi/rhi/impl_support.h b/taichi/rhi/impl_support.h
index 25e7ef68cfcf2..90b92d5efc796 100644
--- a/taichi/rhi/impl_support.h
+++ b/taichi/rhi/impl_support.h
@@ -3,6 +3,7 @@
 #include "taichi/rhi/device.h"
 #include <assert.h>
 #include <forward_list>
+#include <unordered_set>
 #include <mutex>
 
 namespace taichi::lang {