Integrate Triton up to [e902d3b6](https://github.com/openai/triton/co…

…mmits/e902d3b617b100f38caac5c4f74e462edab5f18a) PiperOrigin-RevId: 621779585
tensorflow · Apr 4, 2024 · 9ea400f · 9ea400f
1 parent 974045f
commit 9ea400f
Show file tree

Hide file tree

Showing 8 changed files with 4 additions and 290 deletions.
diff --git a/third_party/triton/cl617812302.patch b/third_party/triton/cl617812302.patch
diff --git a/third_party/triton/cl619146327.patch b/third_party/triton/cl619146327.patch
diff --git a/third_party/triton/cl619443019.patch b/third_party/triton/cl619443019.patch
diff --git a/third_party/triton/workspace.bzl b/third_party/triton/workspace.bzl
@@ -5,8 +5,8 @@ load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 def repo():
     """Imports Triton."""
 
-    TRITON_COMMIT = "cl617459344"
-    TRITON_SHA256 = "f23e65175a67b1091ab6782720b0bcb969f33c19cae8168bf93eea523dab8a3f"
+    TRITON_COMMIT = "cl619179472"
+    TRITON_SHA256 = "aa0b0b338bf16aa7eea778312fa549a421278b24d1a4bc04f5d6ced706f693fe"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,
@@ -15,8 +15,5 @@ def repo():
         # For temporary changes which haven't landed upstream yet.
         patch_file = [
             "//third_party/triton:cl607293980.patch",  # long standing :(
-            "//third_party/triton:cl617812302.patch",
-            "//third_party/triton:cl619146327.patch",
-            "//third_party/triton:cl619443019.patch",
         ],
     )
diff --git a/third_party/xla/third_party/triton/cl617812302.patch b/third_party/xla/third_party/triton/cl617812302.patch
diff --git a/third_party/xla/third_party/triton/cl619146327.patch b/third_party/xla/third_party/triton/cl619146327.patch
diff --git a/third_party/xla/third_party/triton/cl619443019.patch b/third_party/xla/third_party/triton/cl619443019.patch
diff --git a/third_party/xla/third_party/triton/workspace.bzl b/third_party/xla/third_party/triton/workspace.bzl
@@ -5,8 +5,8 @@ load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 def repo():
     """Imports Triton."""
 
-    TRITON_COMMIT = "cl617459344"
-    TRITON_SHA256 = "f23e65175a67b1091ab6782720b0bcb969f33c19cae8168bf93eea523dab8a3f"
+    TRITON_COMMIT = "cl619179472"
+    TRITON_SHA256 = "aa0b0b338bf16aa7eea778312fa549a421278b24d1a4bc04f5d6ced706f693fe"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,
@@ -15,8 +15,5 @@ def repo():
         # For temporary changes which haven't landed upstream yet.
         patch_file = [
             "//third_party/triton:cl607293980.patch",  # long standing :(
-            "//third_party/triton:cl617812302.patch",
-            "//third_party/triton:cl619146327.patch",
-            "//third_party/triton:cl619443019.patch",
         ],
     )