From 8cf97026b76a48510df0f1277d1a4a0f4b2fd5e5 Mon Sep 17 00:00:00 2001 From: Bart Chrzaszcz Date: Wed, 27 Nov 2024 05:32:17 -0800 Subject: [PATCH] #sdy bump version due to JAX MacOS breakage round 2. PiperOrigin-RevId: 700665963 --- third_party/shardy/temporary.patch | 13 ------------- third_party/shardy/workspace.bzl | 4 ++-- third_party/xla/third_party/shardy/temporary.patch | 13 ------------- third_party/xla/third_party/shardy/workspace.bzl | 4 ++-- 4 files changed, 4 insertions(+), 30 deletions(-) diff --git a/third_party/shardy/temporary.patch b/third_party/shardy/temporary.patch index 45648c050f5b6f..e69de29bb2d1d6 100644 --- 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(owningOp) -- .Case([&](func::FuncOp funcOp) { -+ .Case([&, value = value](func::FuncOp funcOp) { - funcOp.setArgAttr(cast(value).getArgNumber(), - kOriginShardingAttr, - DictionaryAttr::get(context, entries)); diff --git a/third_party/shardy/workspace.bzl b/third_party/shardy/workspace.bzl index 446ac940e5acd3..9cf8345dde71a3 100644 --- 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 index 45648c050f5b6f..e69de29bb2d1d6 100644 --- 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(owningOp) -- .Case([&](func::FuncOp funcOp) { -+ .Case([&, value = value](func::FuncOp funcOp) { - funcOp.setArgAttr(cast(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 index 446ac940e5acd3..9cf8345dde71a3 100644 --- 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",