diff --git a/src/Compiler.jl b/src/Compiler.jl index 781b9b621c..b00a77e8eb 100644 --- a/src/Compiler.jl +++ b/src/Compiler.jl @@ -1031,12 +1031,8 @@ function compile_xla(f, args; client=nothing, optimize=true, no_nan=false, devic end end - # compile MLIR module to XLA executable - exec = XLA.Compile( - client, - mod - ) + exec = XLA.Compile(client, mod) ( exec, linear_args, diff --git a/src/XLA.jl b/src/XLA.jl index 48d85e142e..29c09d11d6 100644 --- a/src/XLA.jl +++ b/src/XLA.jl @@ -561,10 +561,7 @@ end end end -function Compile( - client::Client, - mod::MLIR.IR.Module -) +function Compile(client::Client, mod::MLIR.IR.Module) max_local_id = length(client.global_ordinals) GC.@preserve client mod begin executable = LoadedExecutable(