#sdy bump version due to JAX MacOS breakage round 2.

PiperOrigin-RevId: 700665963
47-studio-org · Nov 27, 2024 · 8cf9702 · 8cf9702
1 parent 79586ba
commit 8cf9702
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 30 deletions.
diff --git a/third_party/shardy/temporary.patch b/third_party/shardy/temporary.patch
@@ -1,13 +0,0 @@
-diff --git a/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc b/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-index e1ff855..361b1a6 100644
---- a/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-+++ b/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-@@ -144,7 +144,7 @@ void saveShardingOriginsOnModule(ModuleOp moduleOp,
-                            DictionaryAttr::get(context, entries));
-     }
-     TypeSwitch<Operation*, void>(owningOp)
--        .Case<func::FuncOp>([&](func::FuncOp funcOp) {
-+        .Case<func::FuncOp>([&, value = value](func::FuncOp funcOp) {
-           funcOp.setArgAttr(cast<BlockArgument>(value).getArgNumber(),
-                             kOriginShardingAttr,
-                             DictionaryAttr::get(context, entries));

diff --git a/third_party/shardy/workspace.bzl b/third_party/shardy/workspace.bzl
@@ -3,8 +3,8 @@
 load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 
 def repo():
-    SHARDY_COMMIT = "0c02ab1406aec7b0416832c01dd7847ea830b8fe"
-    SHARDY_SHA256 = "d164c0062d9ab746dfd417423cb9c154ebdbde2906a11816fdce3d857508bba3"
+    SHARDY_COMMIT = "0560e29b05a82d69055d0408a844402ccd188714"
+    SHARDY_SHA256 = "dac84cf89ae9d61a121886d8c77a8ddb66cf471919d8a44bf576464535749de7"
 
     tf_http_archive(
         name = "shardy",

diff --git a/third_party/xla/third_party/shardy/temporary.patch b/third_party/xla/third_party/shardy/temporary.patch
@@ -1,13 +0,0 @@
-diff --git a/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc b/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-index e1ff855..361b1a6 100644
---- a/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-+++ b/shardy/dialect/sdy/transforms/propagation/debugging/source_sharding.cc
-@@ -144,7 +144,7 @@ void saveShardingOriginsOnModule(ModuleOp moduleOp,
-                            DictionaryAttr::get(context, entries));
-     }
-     TypeSwitch<Operation*, void>(owningOp)
--        .Case<func::FuncOp>([&](func::FuncOp funcOp) {
-+        .Case<func::FuncOp>([&, value = value](func::FuncOp funcOp) {
-           funcOp.setArgAttr(cast<BlockArgument>(value).getArgNumber(),
-                             kOriginShardingAttr,
-                             DictionaryAttr::get(context, entries));

diff --git a/third_party/xla/third_party/shardy/workspace.bzl b/third_party/xla/third_party/shardy/workspace.bzl
@@ -3,8 +3,8 @@
 load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
 
 def repo():
-    SHARDY_COMMIT = "0c02ab1406aec7b0416832c01dd7847ea830b8fe"
-    SHARDY_SHA256 = "d164c0062d9ab746dfd417423cb9c154ebdbde2906a11816fdce3d857508bba3"
+    SHARDY_COMMIT = "0560e29b05a82d69055d0408a844402ccd188714"
+    SHARDY_SHA256 = "dac84cf89ae9d61a121886d8c77a8ddb66cf471919d8a44bf576464535749de7"
 
     tf_http_archive(
         name = "shardy",