Applehw (#643)

* 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 <[email protected]> Co-authored-by: Mosè Giordano <[email protected]>
EnzymeAD · Jan 28, 2025 · 27330e0 · 27330e0
1 parent db2aa15
commit 27330e0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 4 deletions.
diff --git 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
@@ -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(