`LinearAlgebra.exp()` on Matrix does not work

[Moelf]

using Reactant
@code_hlo exp(Reactant.to_rarray(rand(2,2)))

using LinearAlgebra, Reactant

reactant_data = Reactant.to_rarray(rand(2,2))
exp_compiled = @compile exp(reactant_data)
@time exp_compiled(reactant_data)

akako@Jerrys-Mac-mini /t/ninebynine> julia --project b.jl
ERROR: LoadError: MethodError: no method matching exp!(::Reactant.TracedRArray{Float64, 2})
The function `exp!` exists, but no method is defined for this combination of argument types.

Closest candidates are:
  exp!(::StridedMatrix{T}) where T<:Union{Float32, Float64, ComplexF64, ComplexF32}
   @ LinearAlgebra ~/.julia/juliaup/julia-1.11.6+0.aarch64.apple.darwin14/share/julia/stdlib/v1.11/LinearAlgebra/src/dense.jl:677

Stacktrace:
  [1] macro expansion
    @ ~/.julia/packages/Reactant/hB4Fs/src/utils.jl:0 [inlined]
  [2] call_with_reactant(::typeof(LinearAlgebra.exp!), ::Reactant.TracedRArray{Float64, 2})
    @ Reactant ~/.julia/packages/Reactant/hB4Fs/src/utils.jl:875
  [3] exp
    @ ~/.julia/juliaup/julia-1.11.6+0.aarch64.apple.darwin14/share/julia/stdlib/v1.11/LinearAlgebra/src/dense.jl:622 [inlined]
  [4] (::Nothing)(none::typeof(exp), none::Reactant.TracedRArray{Float64, 2})
    @ Reactant ./<missing>:0
  [5] exp
    @ ~/.julia/juliaup/julia-1.11.6+0.aarch64.apple.darwin14/share/julia/stdlib/v1.11/LinearAlgebra/src/dense.jl:622 [inlined]
  [6] call_with_reactant(::typeof(exp), ::Reactant.TracedRArray{Float64, 2})
    @ Reactant ~/.julia/packages/Reactant/hB4Fs/src/utils.jl:0
  [7] f
    @ /private/tmp/ninebynine/b.jl:4 [inlined]
  [8] (::Nothing)(none::typeof(f), none::Reactant.TracedRArray{Float64, 2})
    @ Reactant ./<missing>:0
  [9] f
    @ /private/tmp/ninebynine/b.jl:4 [inlined]
 [10] call_with_reactant(::typeof(f), ::Reactant.TracedRArray{Float64, 2})
    @ Reactant ~/.julia/packages/Reactant/hB4Fs/src/utils.jl:0
 [11] make_mlir_fn(f::typeof(f), args::Tuple{ConcretePJRTArray{Float64, 2, 1, Reactant.Sharding.ShardInfo{Reactant.Sharding.NoSharding, Nothing}}}, kwargs::@NamedTuple{}, name::String, concretein::Bool; toscalar::Bool, return_dialect::Symbol, args_in_result::Symbol, construct_function_without_args::Bool, do_transpose::Bool, input_shardings::Nothing, output_shardings::Nothing, runtime::Val{:PJRT}, verify_arg_names::Nothing, argprefix::Symbol, resprefix::Symbol, resargprefix::Symbol, num_replicas::Int64, optimize_then_pad::Bool)
    @ Reactant.TracedUtils ~/.julia/packages/Reactant/hB4Fs/src/TracedUtils.jl:330
 [12] compile_mlir!(mod::Reactant.MLIR.IR.Module, f::Function, args::Tuple{ConcretePJRTArray{Float64, 2, 1, Reactant.Sharding.ShardInfo{Reactant.Sharding.NoSharding, Nothing}}}, compile_options::CompileOptions, callcache::Dict{Vector, @NamedTuple{f_name::String, mlir_result_types::Vector{Reactant.MLIR.IR.Type}, traced_result, mutated_args::Vector{Int64}, linear_results::Vector{Union{ReactantCore.MissingTracedValue, Reactant.TracedRArray, Reactant.TracedRNumber}}, fnwrapped::Bool, argprefix::Symbol, resprefix::Symbol, resargprefix::Symbol}}, sdycache::Dict{Tuple{AbstractVector{Int64}, NTuple{var"#s1626", Symbol} where var"#s1626", NTuple{N, Int64} where N}, @NamedTuple{sym_name::Reactant.MLIR.IR.Attribute, mesh_attr::Reactant.MLIR.IR.Attribute, mesh_op::Reactant.MLIR.IR.Operation, mesh::Reactant.Sharding.Mesh}}; fn_kwargs::@NamedTuple{}, backend::String, runtime::Val{:PJRT}, legalize_stablehlo_to_mhlo::Bool, kwargs::@Kwargs{})
    @ Reactant.Compiler ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:1544
 [13] compile_mlir! (repeats 2 times)
    @ ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:1511 [inlined]
 [14] compile_xla(f::Function, args::Tuple{ConcretePJRTArray{Float64, 2, 1, Reactant.Sharding.ShardInfo{Reactant.Sharding.NoSharding, Nothing}}}; before_xla_optimizations::Bool, client::Nothing, serializable::Bool, kwargs::@Kwargs{compile_options::CompileOptions, fn_kwargs::@NamedTuple{}})
    @ Reactant.Compiler ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:3420
 [15] compile_xla
    @ ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:3393 [inlined]
 [16] compile(f::Function, args::Tuple{ConcretePJRTArray{Float64, 2, 1, Reactant.Sharding.ShardInfo{Reactant.Sharding.NoSharding, Nothing}}}; kwargs::@Kwargs{fn_kwargs::@NamedTuple{}, client::Nothing, reshape_propagate::Symbol, raise_first::Bool, assert_nonallocating::Bool, serializable::Bool, legalize_chlo_to_stablehlo::Bool, transpose_propagate::Symbol, donated_args::Symbol, optimize_then_pad::Bool, cudnn_hlo_optimize::Bool, compile_options::Missing, sync::Bool, no_nan::Bool, raise::Bool, shardy_passes::Symbol, optimize::Bool, optimize_communications::Bool})
    @ Reactant.Compiler ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:3492
 [17] top-level scope
    @ ~/.julia/packages/Reactant/hB4Fs/src/Compiler.jl:2573
in expression starting at /private/tmp/ninebynine/b.jl:8

[mofeing]

This was discussed some time ago on discourse and it can be a lil bit tricky...

We can either do the eigendecomposition and exponentiate the eigenvalues (for which i'm working on adding the eigen op) or do the same as in LinearAlgebra.jl https://github.com/JuliaLang/LinearAlgebra.jl/blob/37c86b840cb82c67c877cdeb59ad473708f58341/src/dense.jl#L766-L900