fix: update distributed APIs to new jll (#1454)

Author: avik-pal

Project.toml

@@ -1,7 +1,7 @@
 name = "Reactant"
 uuid = "3c362404-f566-11ee-1572-e11a4b42c853"
 authors = ["William Moses <[email protected]>", "Valentin Churavy <[email protected]>", "Sergio Sánchez Ramírez <[email protected]>", "Paul Berg <[email protected]>", "Avik Pal <[email protected]>", "Mosè Giordano <[email protected]>"]
-version = "0.2.144"
+version = "0.2.145"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -90,7 +90,7 @@ PythonCall = "0.9.25"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.15"
-Reactant_jll = "0.0.216"
+Reactant_jll = "0.0.217"
 ScopedValues = "1.3.0"
 Scratch = "1.2"
 Sockets = "1.10"

src/xla/Distributed.jl

@@ -14,7 +14,6 @@ function DistributedRuntimeClient(
     rpc_timeout_in_seconds::Integer=120,
     shutdown_timeout_in_minutes::Integer=5,
     heartbeat_interval_in_seconds::Integer=10,
-    max_missing_heartbeats::Integer=10,
     use_compression::Bool=true,
 )
     GC.@preserve coordinator_bind_address begin
@@ -24,7 +23,6 @@ function DistributedRuntimeClient(
             Int32(rpc_timeout_in_seconds)::Int32,
             Int32(shutdown_timeout_in_minutes)::Int32,
             Int32(heartbeat_interval_in_seconds)::Int32,
-            Cint(max_missing_heartbeats)::Cint,
             use_compression::Bool,
         )::Ptr{Cvoid}
     end
@@ -69,7 +67,6 @@ function DistributedRuntimeService(
     coordinator_bind_address::String,
     num_nodes::Integer;
     heartbeat_interval_in_seconds::Integer=10,
-    max_missing_heartbeats::Integer=10,
     cluster_register_timeout_in_minutes::Integer=60,
     shutdown_timeout_in_minutes::Integer=5,
 )
@@ -78,7 +75,6 @@ function DistributedRuntimeService(
             coordinator_bind_address::Cstring,
             Cint(num_nodes)::Cint,
             Int32(heartbeat_interval_in_seconds)::Int32,
-            Cint(max_missing_heartbeats)::Cint,
             Int32(cluster_register_timeout_in_minutes)::Int32,
             Int32(shutdown_timeout_in_minutes)::Int32,
         )::Ptr{Cvoid}
@@ -135,7 +131,6 @@ function update!(
     rpc_timeout_in_seconds::Integer=120,
     shutdown_timeout_in_minutes::Integer=5,
     heartbeat_interval_in_seconds::Integer=10,
-    max_missing_heartbeats::Integer=10,
     use_compression::Bool=true,
 )
     @assert 0 ≤ process_id < num_processes
@@ -166,7 +161,6 @@ function update!(
             coordinator_bind_address,
             num_processes;
             heartbeat_interval_in_seconds,
-            max_missing_heartbeats,
             cluster_register_timeout_in_minutes,
             shutdown_timeout_in_minutes,
         )
@@ -189,7 +183,6 @@ function update!(
         rpc_timeout_in_seconds,
         shutdown_timeout_in_minutes,
         heartbeat_interval_in_seconds,
-        max_missing_heartbeats,
         use_compression,
     )
     @debug "[PID $(process_id)] Connecting to Reactant distributed service on \