Improved external function support by allowing multiple modules, e.g. a.b.c(x,y,z).

Author: chriselrod

docs/src/examples/filtering.md

@@ -2,6 +2,8 @@
 
 Here, we convolve a small matrix `kern` with a larger matrix `A`, storing the results in `out`, using Julia's generic [Cartesian Indexing](https://julialang.org/blog/2016/02/iteration/):
 ```julia
+using LoopVectorization, OffsetArrays, Images
+kern = Images.Kernel.gaussian((1, 1), (3, 3))
 function filter2davx!(out::AbstractMatrix, A::AbstractMatrix, kern)
     @avx for J in CartesianIndices(out)
         tmp = zero(eltype(out))

src/add_compute.jl

@@ -187,7 +187,8 @@ function add_compute!(
     mpref::Union{Nothing,ArrayReferenceMetaPosition} = nothing
 )
     @assert ex.head === :call
-    instr = instruction(first(ex.args))::Symbol
+    # instr = instruction(first(ex.args))::Symbol
+    instr = instruction!(ls, first(ex.args))::Symbol
     args = @view(ex.args[2:end])
     (instr === :(^) && length(args) == 2 && (args[2] isa Number)) && return add_pow!(ls, var, args[1], args[2], elementbytes, position)
     vparents = Operation[]

src/graphs.jl

@@ -450,6 +450,15 @@ end
 function instruction(x)
     x isa Symbol ? x : last(x.args).value
 end
+function instruction!(ls::LoopSet, x)
+    x isa Symbol && return x
+    instr = last(x.args).value
+    if Instruction(instr) ∉ keys(COST)
+        instr = gensym(:f)
+        pushpreamble!(ls, Expr(:(=), instr, x))
+    end
+    instr
+end
 function add_operation!(
     ls::LoopSet, LHS::Symbol, RHS::Expr, elementbytes::Int, position::Int
 )