Integrate Triton up to [c7a37a9d](https://github.com/triton-lang/trit…

…on/commits/c7a37a9d6b3f67c817ca4f2cdc2d3b75205bceef) PiperOrigin-RevId: 646915295
tensorflow · Jun 26, 2024 · 5881722 · 5881722
1 parent b572b20
commit 5881722
Show file tree

Hide file tree

Showing 10 changed files with 8 additions and 342 deletions.
diff --git a/third_party/triton/llvm_integration/cl644752548.patch b/third_party/triton/llvm_integration/cl644752548.patch
diff --git a/third_party/triton/llvm_integration/series.bzl b/third_party/triton/llvm_integration/series.bzl
@@ -4,6 +4,4 @@ Provides a temporary list of patches created during llvm-integration.
 These should be upstreamed to openai/triton as part of the next triton integration process.
 """
 
-llvm_patch_list = [
-    "//third_party/triton/llvm_integration:cl644752548.patch",
-]
+llvm_patch_list = []
diff --git a/third_party/triton/temporary/linear_layout_rank_fix.patch b/third_party/triton/temporary/linear_layout_rank_fix.patch
diff --git a/third_party/triton/temporary/series.bzl b/third_party/triton/temporary/series.bzl
@@ -5,6 +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:linear_layout_rank_fix.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 = "cl644297380"
-    TRITON_SHA256 = "a7fb8651973deaba556ad85347a04577fd8228abc77b7c61d67bb90bf89e11b0"
+    TRITON_COMMIT = "cl646915295"
+    TRITON_SHA256 = "c0dc5f59da07cd615e1d19367a8d45b314a3b61523db30384ef340d794cbc2a8"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,

diff --git a/third_party/xla/third_party/triton/llvm_integration/cl644752548.patch b/third_party/xla/third_party/triton/llvm_integration/cl644752548.patch
diff --git a/third_party/xla/third_party/triton/llvm_integration/series.bzl b/third_party/xla/third_party/triton/llvm_integration/series.bzl
@@ -4,6 +4,4 @@ Provides a temporary list of patches created during llvm-integration.
 These should be upstreamed to openai/triton as part of the next triton integration process.
 """
 
-llvm_patch_list = [
-    "//third_party/triton/llvm_integration:cl644752548.patch",
-]
+llvm_patch_list = []
diff --git a/third_party/xla/third_party/triton/temporary/linear_layout_rank_fix.patch b/third_party/xla/third_party/triton/temporary/linear_layout_rank_fix.patch
diff --git a/third_party/xla/third_party/triton/temporary/series.bzl b/third_party/xla/third_party/triton/temporary/series.bzl
@@ -5,6 +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:linear_layout_rank_fix.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 = "cl644297380"
-    TRITON_SHA256 = "a7fb8651973deaba556ad85347a04577fd8228abc77b7c61d67bb90bf89e11b0"
+    TRITON_COMMIT = "cl646915295"
+    TRITON_SHA256 = "c0dc5f59da07cd615e1d19367a8d45b314a3b61523db30384ef340d794cbc2a8"
     tf_http_archive(
         name = "triton",
         sha256 = TRITON_SHA256,