From 6ee5b8172c1761061021298ee7cd6d681e2148ff Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 18:51:27 +0000 Subject: [PATCH 1/6] [ReactantExtra] Bump XLA version --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 333d6d3fe9..62832f4bdd 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -136,7 +136,7 @@ http_archive( ) # load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256") -XLA_COMMIT = "4b9f0e2130d665ef9731154832af9a91155686d2" +XLA_COMMIT = "5928fd05598201a844aaccfd3d12767f190b1ccf" XLA_SHA256 = "" http_archive( From 7c7a51f70d3b8f39f8901d4bd754e5fa1d09ed6e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 19:44:52 +0000 Subject: [PATCH 2/6] [ReactantExtra] Remove `@xla//xla/python:jax_cuda_pip_rpaths` --- deps/ReactantExtra/.bazelrc | 1 - 1 file changed, 1 deletion(-) diff --git a/deps/ReactantExtra/.bazelrc b/deps/ReactantExtra/.bazelrc index e1dba3f200..4f13bbd7f1 100644 --- a/deps/ReactantExtra/.bazelrc +++ b/deps/ReactantExtra/.bazelrc @@ -25,7 +25,6 @@ build:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.4.0" build:cuda --repo_env HERMETIC_CUDA_COMPUTE_CAPABILITIES="sm_50,sm_60,sm_70,sm_80,compute_90" build:cuda --crosstool_top="@local_config_cuda//crosstool:toolchain" build:cuda --@local_config_cuda//:enable_cuda -build:cuda --@xla//xla/python:jax_cuda_pip_rpaths=true # Default hermetic CUDA and CUDNN versions. build:cuda --@local_config_cuda//cuda:include_cuda_libs=true build:cuda --@local_config_cuda//:cuda_compiler=nvcc From 728794fce74df88f0b247cc340975dfc743e6e36 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 19:45:53 +0000 Subject: [PATCH 3/6] [ReactantExtra] Bump Enzyme-JAX --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 62832f4bdd..2dafa65828 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -9,7 +9,7 @@ http_archive( urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)], ) -ENZYMEXLA_COMMIT = "fd5517f2223adcf579c165a0387231ef4931f55b" +ENZYMEXLA_COMMIT = "6103feeaabf401ae6cb06d66b9ebc12b67b45fe5" ENZYMEXLA_SHA256 = "" http_archive( From 034be36668577fc3ff83e230dab8c65f92e2fd08 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 20:01:49 +0000 Subject: [PATCH 4/6] [ReactantExtra] Re-bump Enzyme-JAX --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 2dafa65828..5c82be8005 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -9,7 +9,7 @@ http_archive( urls = ["https://github.com/wsmoses/nsync/archive/{commit}.tar.gz".format(commit = NSYNC_COMMIT)], ) -ENZYMEXLA_COMMIT = "6103feeaabf401ae6cb06d66b9ebc12b67b45fe5" +ENZYMEXLA_COMMIT = "3b33e3373d20ca2e1005a30d2aced83024230977" ENZYMEXLA_SHA256 = "" http_archive( From 7b681495c108953328b04ffcf47e8c3db4eb3b5d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 20:02:03 +0000 Subject: [PATCH 5/6] [ReactantExtra] Fix `API.cpp` --- deps/ReactantExtra/API.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/API.cpp b/deps/ReactantExtra/API.cpp index 5017c042a3..3187806345 100644 --- a/deps/ReactantExtra/API.cpp +++ b/deps/ReactantExtra/API.cpp @@ -697,7 +697,7 @@ extern "C" void InitializeRegistryAndPasses(MlirDialectRegistry creg) { prepareRegistry(registry); mlir::registerenzymePasses(); - registerenzymexlaPasses(); + enzyme::registerenzymexlaPasses(); // Register the standard passes we want. mlir::registerCSEPass(); From 7934b759a4094eb3781f5cd15fa8f4f419a64eed Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Mon, 27 Jan 2025 20:17:36 +0000 Subject: [PATCH 6/6] [ReactantExtra] Update XLA to latest commit on `main` instead of PR --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 5c82be8005..b06621a45b 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -136,7 +136,7 @@ http_archive( ) # load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256") -XLA_COMMIT = "5928fd05598201a844aaccfd3d12767f190b1ccf" +XLA_COMMIT = "7ad64b5df46daf1f84d8a80ec87ae5ddbb3f82b7" XLA_SHA256 = "" http_archive(