feat: initial throw impl

Author: avik-pal

src/Ops.jl

@@ -15,6 +15,24 @@ using ..Reactant:
 using ReactantCore: ReactantCore
 using Functors: fmap
 
+using Reactant_jll: Reactant_jll
+
+function __init__()
+    if Reactant_jll.is_available()
+        error_fn_ptr = @cfunction(error, Union{}, (String,))
+        println_fn_ptr = @cfunction(println, Nothing, (String,))
+
+        for (ptr, enzymexla_name) in
+            [(error_fn_ptr, :enzymexla_error), (println_fn_ptr, :enzymexla_println)]
+            @ccall MLIR.API.mlir_c.EnzymeJaXMapSymbol(
+                enzymexla_name::Cstring, ptr::Ptr{Cvoid}
+            )::Cvoid
+        end
+    end
+
+    return nothing
+end
+
 function mlir_type(x::Union{RNumber,RArray})::MLIR.IR.Type
     return MLIR.IR.TensorType(collect(Int, size(x)), MLIR.IR.Type(unwrapped_eltype(x)))
 end
@@ -3114,4 +3132,55 @@ end
     end
 end
 
+@noinline function throw(
+    msg::String; location=mlir_stacktrace("throw", @__FILE__, @__LINE__)
+)
+    mod = MLIR.IR.mmodule()
+
+    sym_name = string(Reactant.TracedUtils.__lookup_unique_name_in_module(mod, "error_msg"))
+    MLIR.IR.inject!(
+        sym_name,
+        "llvm.mlir.global constant @$(sym_name)(\"$(msg)\")";
+        mod,
+        location,
+        verify=true,
+    )
+
+    if MLIR.IR.mlirIsNull(
+        MLIR.API.mlirSymbolTableLookup(
+            MLIR.IR.SymbolTable(MLIR.IR.Operation(mod)), "enzymexla_println"
+        ),
+    )
+        MLIR.IR.inject!(
+            "enzymexla_println",
+            "llvm.func external @enzymexla_println(!llvm.ptr)";
+            mod,
+            location,
+        )
+    end
+
+    error_fn = string(Reactant.TracedUtils.__lookup_unique_name_in_module(mod, "error"))
+    MLIR.IR.inject!(
+        error_fn,
+        """
+        func.func @$(error_fn)() attributes {no_inline} {
+            %err_msg_ptr = llvm.mlir.addressof @$(sym_name) : !llvm.ptr
+            llvm.call @enzymexla_println(%err_msg_ptr) : (!llvm.ptr) -> ()
+            func.return
+        }
+        """;
+        mod,
+        location,
+    )
+
+    MLIR.Dialects.enzymexla.jit_call(
+        MLIR.IR.Value[];
+        fn=MLIR.IR.FlatSymbolRefAttribute(error_fn),
+        result_0=MLIR.IR.Type[],
+        location,
+    )
+
+    return nothing
+end
+
 end # module Ops