Skip to content

Commit c2a9a2d

Browse files
wsmosesGoogle-ML-Automation
authored andcommitted
Triton: Fix missing include header
PiperOrigin-RevId: 720575368
1 parent 2dd08e7 commit c2a9a2d

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed
+12
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
2+
--- a/include/triton/Dialect/TritonGPU/IR/Dialect.h 2025-01-21 05:40:49.000000000 -0800
3+
+++ b/include/triton/Dialect/TritonGPU/IR/Dialect.h 2025-01-28 05:00:43.000000000 -0800
4+
@@ -11,6 +11,8 @@
5+
#include "triton/Dialect/TritonGPU/IR/Attributes.h"
6+
#include "triton/Dialect/TritonGPU/IR/Types.h"
7+
8+
+#include <unordered_map>
9+
+
10+
// LinearLayoutCache Utils
11+
using CacheKey =
12+
std::tuple<std::vector<int64_t>, mlir::Attribute, std::optional<int32_t>>;

third_party/triton/temporary/series.bzl

+4
Original file line numberDiff line numberDiff line change
@@ -14,5 +14,9 @@ those to this list.
1414
"""
1515

1616
temporary_patch_list = [
17+
# Patch merged into Triton upstream (https://github.com/triton-lang/triton/pull/5721)
18+
# but not merged into llvm_head branch nor Google's fork yet.
19+
# Without this XLA fails to build on macos crosscompile and other targets.
20+
"//third_party/triton:temporary/header.patch",
1721
# Add new patches just above this line
1822
]

0 commit comments

Comments
 (0)