Fix the broken build for ee42234

llvm · Jul 12, 2024 · d601ca3 · d601ca3
1 parent d5c89cc
commit d601ca3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/mlir/include/mlir/Support/CyclicReplacerCache.h b/mlir/include/mlir/Support/CyclicReplacerCache.h
@@ -17,6 +17,8 @@
 
 #include "llvm/ADT/DenseMap.h"
 #include "llvm/ADT/DenseSet.h"
+#include "llvm/ADT/SmallVector.h"
+#include <functional>
 #include <optional>
 #include <set>
 

diff --git a/mlir/unittests/Support/CyclicReplacerCacheTest.cpp b/mlir/unittests/Support/CyclicReplacerCacheTest.cpp
@@ -7,6 +7,7 @@
 //===----------------------------------------------------------------------===//
 
 #include "mlir/Support/CyclicReplacerCache.h"
+#include "mlir/Support/LLVM.h"
 #include "llvm/ADT/SetVector.h"
 #include "gmock/gmock.h"
 #include <map>