From 5377b28a3b17ed4940b31f2c1fc0a79f3758a42f Mon Sep 17 00:00:00 2001
From: William Moses <gh@wsmoses.com>
Date: Wed, 15 May 2024 23:50:16 -0700
Subject: [PATCH] Update XLA.jl

---
 src/XLA.jl | 14 +++++++++++---
 1 file changed, 11 insertions(+), 3 deletions(-)

diff --git a/src/XLA.jl b/src/XLA.jl
index e66cef010..d473cb9b1 100644
--- a/src/XLA.jl
+++ b/src/XLA.jl
@@ -61,7 +61,11 @@ function GPUClient(node_id=0, num_nodes=1, platform="gpu")
     #allowed_devices = [-1]
     # GC.@preserve allowed_devices begin
         f = Libdl.dlsym(Reactant_jll.libReactantExtra_handle, "MakeGPUClient")
-        client = ccall(f, Ptr{Cvoid}, (Cint, Cint, Ptr{Cvoid}, Cint, Cstring), node_id, num_nodes, C_NULL, 0, platform)
+        refstr = Ref{Cstring}()
+        client = ccall(f, Ptr{Cvoid}, (Cint, Cint, Ptr{Cvoid}, Cint, Cstring, Ptr{Cstring}), node_id, num_nodes, C_NULL, 0, platform, refstr)
+        if client == C_NULL
+           throw(AssertionError(refstr[]))
+    end
     # end
     return Client(client)
 end
@@ -77,8 +81,12 @@ function __init__()
     cpu = CPUClient()
     backends["cpu"] = cpu
     @static if !Sys.isapple()
-        gpu = GPUClient()
-        backends["gpu"] = gpu
+        try
+          gpu = GPUClient()
+          backends["gpu"] = gpu
+        catch e
+          println(stdout, e)
+        end
     end
     default_backend[] = cpu
 end