[ReactantExtra] Update XLA and adapt API.cpp

[skip tests]

Author: giordano

deps/ReactantExtra/API.cpp

@@ -837,7 +837,7 @@ xla::CompileOptions GenerateCompileOptions(int64_t device_id, bool is_sharded,
   return options;
 }
 
-extern "C" xla::PjRtLoadedExecutable *
+extern "C" xla::PjRtExecutable *
 ClientCompile(PjRtClient *client, MlirModule cmod, int64_t device_id,
               bool is_sharded, const int64_t *mesh_ids, int64_t num_mesh_ids,
               const char *xla_gpu_cuda_data_dir, bool use_shardy_partitioner) {

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 = "0b0e36a3a78a4dcda5f50ca42b4cd4f2f6534fa1"
+ENZYMEXLA_COMMIT = "e0b13d70cab494c927c0bd9b3251f5ed8dcdc8a5"
 ENZYMEXLA_SHA256 = ""
 
 http_archive(
@@ -129,7 +129,10 @@ http_archive(
     patches = ["@enzyme_ad//:patches/jax.patch"],
 )
 
-load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
+# load("@jax//third_party/xla:workspace.bzl", "XLA_COMMIT", "XLA_SHA256")
+
+XLA_COMMIT = "821715be3f7ded31723b9b4c0d5de8f0d2a30a6b"
+XLA_SHA256 = ""
 
 http_archive(
     name = "xla",