Improve reactant error messages

Project.toml

@@ -60,7 +60,7 @@ PythonCall = "0.9"
 Random = "1.10"
 Random123 = "1.7"
 ReactantCore = "0.1.3"
-Reactant_jll = "0.0.27"
+Reactant_jll = "0.0.29"
 Scratch = "1.2"
 Statistics = "1.10"
 YaoBlocks = "0.13"

deps/ReactantExtra/BUILD

@@ -430,6 +430,7 @@ cc_library(
 "-Wl,-exported_symbol,_RegisterCustomCallTarget",
 "-Wl,-exported_symbol,_ConvertLLVMToMLIR",
 "-Wl,-exported_symbol,_EnzymeGPUCustomCall",
+"-Wl,-exported_symbol,_ReactantThrowError",
     ]}),
     deps = [
                 "@enzyme//:EnzymeMLIR",

src/XLA.jl

@@ -91,6 +91,19 @@ const default_device_idx = Ref{Int}(0)
 using Reactant_jll
 using Libdl
 using Scratch, Downloads
+
+struct ReactantInternalError <: Base.Exception
+    msg::String
+end
+
+function Base.showerror(io::IO, ece::ReactantInternalError)
+    return print(io, ece.msg, '\n')
+end
+
+function reactant_err(msg::Cstring)::Cvoid
+    throw(ReactantInternalError(Base.unsafe_string(msg)))
+end
+
 function __init__()
     initLogs = Libdl.dlsym(Reactant_jll.libReactantExtra_handle, "InitializeLogs")
     ccall(initLogs, Cvoid, ())
@@ -136,6 +149,14 @@ function __init__()
         cglobal((:EnzymeGPUCustomCall, MLIR.API.mlir_c))::Ptr{Cvoid},
         "CUDA"::Cstring,
     )::Cvoid
+
+    # This wasn't properly exported on macos, we'll remove the try once macOS JLL
+    # has the fix.
+    try
+        errptr = cglobal((:ReactantThrowError, MLIR.API.mlir_c), Ptr{Ptr{Cvoid}})
+        unsafe_store!(errptr, @cfunction(reactant_err, Cvoid, (Cstring,)))
+    finally
+    end
     return nothing
 end