From 27330e055625b32d71732a01a37b39f47308aa61 Mon Sep 17 00:00:00 2001 From: William Moses Date: Tue, 28 Jan 2025 19:17:01 +0100 Subject: [PATCH] Applehw (#643) MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit * Fix hw int * Update XLA commit * Update also Enzyme-JAX commit * Add `xla/stream_executor/cuda:cuda_compute_capability_proto` * Add more targets --------- Co-authored-by: Mosè Giordano <765740+giordano@users.noreply.github.com> Co-authored-by: Mosè Giordano --- deps/ReactantExtra/BUILD | 7 +++++-- deps/ReactantExtra/WORKSPACE | 7 +++++-- 2 files changed, 10 insertions(+), 4 deletions(-) diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD index 72808252a..f30576679 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", @@ -556,7 +558,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", diff --git a/deps/ReactantExtra/WORKSPACE b/deps/ReactantExtra/WORKSPACE index 541a41152..cce06d553 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( @@ -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 """, """ @@ -136,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(