Triton: Fix missing include header

PiperOrigin-RevId: 720575368
openxla · Jan 28, 2025 · c2a9a2d · c2a9a2d
1 parent 2dd08e7
commit c2a9a2d
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/third_party/triton/temporary/header.patch b/third_party/triton/temporary/header.patch
@@ -0,0 +1,12 @@
+
+--- a/include/triton/Dialect/TritonGPU/IR/Dialect.h	2025-01-21 05:40:49.000000000 -0800
++++ b/include/triton/Dialect/TritonGPU/IR/Dialect.h	2025-01-28 05:00:43.000000000 -0800
+@@ -11,6 +11,8 @@
+ #include "triton/Dialect/TritonGPU/IR/Attributes.h"
+ #include "triton/Dialect/TritonGPU/IR/Types.h"
+
++#include <unordered_map>
++
+ // LinearLayoutCache Utils
+ using CacheKey =
+     std::tuple<std::vector<int64_t>, mlir::Attribute, std::optional<int32_t>>;
diff --git a/third_party/triton/temporary/series.bzl b/third_party/triton/temporary/series.bzl
@@ -14,5 +14,9 @@ those to this list.
 """
 
 temporary_patch_list = [
+    # Patch merged into Triton upstream (https://github.com/triton-lang/triton/pull/5721)
+    # but not merged into llvm_head branch nor Google's fork yet.
+    # Without this XLA fails to build on macos crosscompile and other targets.
+    "//third_party/triton:temporary/header.patch",
     # Add new patches just above this line
 ]