From 81e1eeaa1ea3ec9cf49fe03b74ee6806e9300398 Mon Sep 17 00:00:00 2001 From: "William S. Moses" Date: Mon, 27 Jan 2025 17:44:56 -0600 Subject: [PATCH 1/5] Fix hw int --- deps/ReactantExtra/WORKSPACE | 3 +++ 1 file changed, 3 insertions(+) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 541a411520..38f8531d5c 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -54,6 +54,9 @@ XLA_PATCHES = XLA_PATCHES + [ sed -i.bak0 "s/__cpp_lib_hardware_interference_size/HW_INTERFERENCE_SIZE/g" xla/backends/cpu/runtime/thunk_executor.h """, """ +sed -i.bak0 "s/__cpp_lib_hardware_interference_size/HW_INTERFERENCE_SIZE/g" xla/backends/cpu/runtime/work_queue.h +""", +""" sed -i.bak0 "s/__cpp_lib_hardware_interference_size/HW_INTERFERENCE_SIZE/g" xla/tsl/concurrency/async_value_ref.h """, """ From c73d72f18c09f01d71c81f42e8eb0bf1f995c364 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= <765740+giordano@users.noreply.github.com> Date: Tue, 28 Jan 2025 17:05:21 +0100 Subject: [PATCH 2/5] Update XLA commit --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 38f8531d5c..01e9cb7014 100644 --- a/deps/ReactantExtra/WORKSPACE +++ b/deps/ReactantExtra/WORKSPACE @@ -139,7 +139,7 @@ http_archive( ) # load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256") -XLA_COMMIT = "7ad64b5df46daf1f84d8a80ec87ae5ddbb3f82b7" +XLA_COMMIT = "c2a9a2dfe9494e52f5134b53989e9ca0de307dfe" XLA_SHA256 = "" http_archive( From 2832c78187891abb47a7e92098dd816e801ec043 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= <765740+giordano@users.noreply.github.com> Date: Tue, 28 Jan 2025 17:15:24 +0100 Subject: [PATCH 3/5] Update also Enzyme-JAX commit --- deps/ReactantExtra/WORKSPACE | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 01e9cb7014..cce06d553f 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 = "33bd450e8e7132a9081d2b97419a20c515b15076" +ENZYMEXLA_COMMIT = "c38ca3f187ef11de6b2292f3cc55c5eb60530d15" ENZYMEXLA_SHA256 = "" http_archive( From d06220fcfd920341f35257da75db07579ed126f7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Tue, 28 Jan 2025 16:58:10 +0000 Subject: [PATCH 4/5] Add `xla/stream_executor/cuda:cuda_compute_capability_proto` --- deps/ReactantExtra/BUILD | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD index 72808252a4..9f06d074f6 100644 --- a/deps/ReactantExtra/BUILD +++ b/deps/ReactantExtra/BUILD @@ -556,7 +556,8 @@ cc_library( "@jax//jaxlib/triton:triton_dialect_capi_objects", ] + select({ "@xla//xla/tsl:is_cuda_enabled_and_oss":[ - "@xla//xla/stream_executor/cuda:all_runtime", + "@xla//xla/stream_executor/cuda:all_runtime", + "@xla//xla/stream_executor/cuda:cuda_compute_capability_proto", "@xla//xla/service/gpu/model:hlo_op_profiles", "@xla//xla/service/gpu/model:hlo_op_profile_proto_cc_impl", "@xla//xla/service/gpu:nvptx_compiler", From 123bb2a453b891ca883af3410ba2dcd2256bbc79 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Mos=C3=A8=20Giordano?= Date: Tue, 28 Jan 2025 17:55:06 +0000 Subject: [PATCH 5/5] Add more targets --- deps/ReactantExtra/BUILD | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD index 9f06d074f6..f305766798 100644 --- a/deps/ReactantExtra/BUILD +++ b/deps/ReactantExtra/BUILD @@ -365,7 +365,7 @@ cc_library( # "@com_google_protobuf//:src/google/protobuf/io/coded_stream.cc", # "@xla//xla:xla.pb.cc", "@xla//xla:xla_data.pb.cc", - "@xla//xla/stream_executor:device_description.pb.cc", + # "@xla//xla/stream_executor:device_description.pb.cc", "@xla//xla/service:hlo.pb.cc", # # "@tsl//tsl/protobuf:dnn.pb.cc", #"@tsl//tsl/protobuf:histogram.pb.cc", @@ -497,6 +497,8 @@ cc_library( "@xla//xla:xla_proto_cc", "@xla//xla:xla_proto_cc_impl", + "@xla//xla/stream_executor:device_description_proto_cc_impl", + "@xla//xla/stream_executor/cuda:cuda_compute_capability_proto_cc_impl", "@xla//xla/service:metrics_proto_cc", "@xla//xla/service:metrics_proto_cc_impl",