访问Julia中的泛型函数的AST

Question

如何在Julia中访问泛型函数的抽象语法树？

Answer 1

回顾一下：看起来Simon正在寻找与泛型函数相关的特定方法的AST。我们可以得到一个LambdaStaticData对象，其中包含AST，具体方法如下：

julia> f(x,y)=x+y

julia> f0 = methods(f, (Any, Any))[1]
((Any,Any),(),AST(:($(expr(:lambda, {x, y}, {{}, {{x, Any, 0}, {y, Any, 0}}, {}}, quote  # none, line 1:
        return +(x,y)
    end)))),())

julia> f0[3]
AST(:($(expr(:lambda, {x, y}, {{}, {{x, Any, 0}, {y, Any, 0}}, {}}, quote  # none, line 1:
        return +(x,y)
    end))))

julia> typeof(ans)
LambdaStaticData

显然，这个AST既可以是Expr对象，也可以是压缩的AST对象，表示为字节序列：

julia> typeof(f0[3].ast)
Array{Uint8,1}

来自show()的{​​{1}}的{​​{1}}方法说明了遇到此问题时如何解压缩：

LambdaStaticData

Answer 2

Julia有四个函数和四个类似于这些函数的宏，用于检查很多关于泛型函数的方法：

julia> f(x, y) = x + y                                                                                                    
f (generic function with 1 method)

julia> methods(f)                                                                                                         
# 1 method for generic function "f":                                                                                      
f(x,y) at none:1 

降低代码：

julia> code_lowered(f, (Int, Int))                                                                                        
1-element Array{Any,1}:
 :($(Expr(:lambda, {:x,:y}, {{},{{:x,:Any,0},{:y,:Any,0}},{}}, :(begin  # none, line 1:
        return x + y
    end))))

julia> @code_lowered f(1, 1)    # Both `Int`s 
...same output.

输入的代码：

julia> code_typed(f, (Int, Int))    

1-element Array{Any,1}:
 :($(Expr(:lambda, {:x,:y}, {{},{{:x,Int64,0},{:y,Int64,0}},{}}, :(begin  # none, line 1:                                 
        return (top(box))(Int64,(top(add_int))(x::Int64,y::Int64))::Int64                                                 
    end::Int64))))      

julia> @code_lowered f(1, 1)    # Both `Int`s  
...same output.

LLVM代码：

julia> code_llvm(f, (Int, Int))                                                                                           

define i64 @julia_f_24771(i64, i64) {                                                                                     
top:                                                                                                                      
  %2 = add i64 %1, %0, !dbg !1014                                                                                         
  ret i64 %2, !dbg !1014                                                                                                  
}                           

julia> @code_llvm f(1, 1)    # Both `Int`s   
...same output.

原生代码：

julia> code_native(f, (Int, Int))  

      .text                                                                                                               
Filename: none                                                                                                            
Source line: 1                                                                                                            
        push    RBP                                                                                                       
        mov     RBP, RSP                                                                                                  
Source line: 1                                                                                                            
        add     RDI, RSI                                                                                                  
        mov     RAX, RDI                                                                                                  
        pop     RBP                                                                                                       
        ret    

julia> @code_llvm f(1, 1)    # Both `Int`s
...same output.

类型不稳定警告（v0.4 +）：

julia> @code_warntype f(1, 1)
Variables:
  x::Int64
  y::Int64

Body:
  begin  # In[17], line 1:
      return (top(box))(Int64,(top(add_int))(x::Int64,y::Int64))
  end::Int64

  

Reflection and introspection

Answer 3

由于多次调度，我不确定是否存在与通用函数关联的AST。如果您正在编写函数定义fbody，那么您应该能够通过执行dump(quote(fbody))来获取AST。