Integrate Triton up to 25b4212a9

PiperOrigin-RevId: 634797641
tensorflow · May 17, 2024 · 4d73219 · 4d73219
1 parent 42ac0dd
commit 4d73219
Show file tree

Hide file tree

Showing 8 changed files with 6 additions and 186 deletions.
diff --git a/third_party/triton/temporary/fix_register_constraints.patch b/third_party/triton/temporary/fix_register_constraints.patch
diff --git a/third_party/triton/temporary/fp8_splat.patch b/third_party/triton/temporary/fp8_splat.patch
diff --git a/third_party/triton/temporary/series.bzl b/third_party/triton/temporary/series.bzl
@@ -5,7 +5,4 @@ These are created temporarily and should be moved to the first copybara workflow
 internal patch during the next triton integration process.
 """
 
-temporary_patch_list = [
-    "//third_party/triton/temporary:fp8_splat.patch",
-    "//third_party/triton/temporary:fix_register_constraints.patch",
-]
+temporary_patch_list = []
diff --git a/third_party/triton/workspace.bzl b/third_party/triton/workspace.bzl
@@ -8,8 +8,8 @@ load("//third_party/triton/xla_extensions:series.bzl", "extensions_files_patch_l
 def repo():
     """Imports Triton."""
 
-    TRITON_COMMIT = "cl633653394"
-    TRITON_SHA256 = "4bf1b25ce6f7ce6ab6fb4e2403b24c8c0bf71f6a9f5c08f0125fdb44bcb371a8"
+    TRITON_COMMIT = "cl634675237"
+    TRITON_SHA256 = "7151d057ee8443c2f45cbe18a7435a42f37e18f562e5d238b844b6e09fc560e6"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,

diff --git a/third_party/xla/third_party/triton/temporary/fix_register_constraints.patch b/third_party/xla/third_party/triton/temporary/fix_register_constraints.patch
diff --git a/third_party/xla/third_party/triton/temporary/fp8_splat.patch b/third_party/xla/third_party/triton/temporary/fp8_splat.patch
diff --git a/third_party/xla/third_party/triton/temporary/series.bzl b/third_party/xla/third_party/triton/temporary/series.bzl
@@ -5,7 +5,4 @@ These are created temporarily and should be moved to the first copybara workflow
 internal patch during the next triton integration process.
 """
 
-temporary_patch_list = [
-    "//third_party/triton/temporary:fp8_splat.patch",
-    "//third_party/triton/temporary:fix_register_constraints.patch",
-]
+temporary_patch_list = []
diff --git a/third_party/xla/third_party/triton/workspace.bzl b/third_party/xla/third_party/triton/workspace.bzl
@@ -8,8 +8,8 @@ load("//third_party/triton/xla_extensions:series.bzl", "extensions_files_patch_l
 def repo():
     """Imports Triton."""
 
-    TRITON_COMMIT = "cl633653394"
-    TRITON_SHA256 = "4bf1b25ce6f7ce6ab6fb4e2403b24c8c0bf71f6a9f5c08f0125fdb44bcb371a8"
+    TRITON_COMMIT = "cl634675237"
+    TRITON_SHA256 = "7151d057ee8443c2f45cbe18a7435a42f37e18f562e5d238b844b6e09fc560e6"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,