fix #22

src/blas.jl

@@ -1,22 +1,45 @@
 # generic implementations
 
+_name(x::Symbol) = x
+_name(x::Expr) = (@assert x.head == :(::); x.args[1])
 macro reroute(f, g)
+    fname = f.args[1]
+    fargs = f.args[2:end]
     quote
+        @inline function Cassette.overdub(ctx::SparsityContext,
+                                          f::typeof($(esc(fname))),
+                                          $(fargs...))
+            Cassette.recurse(
+                ctx,
+                invoke,
+                f,
+                $(esc(:(Tuple{$(g.args[2:end]...)}))),
+                $(map(_name, fargs)...))
+        end
+
         @inline function Cassette.overdub(ctx::HessianSparsityContext,
-                                          f::typeof($(esc(f))),
+                                          f::typeof($(esc(fname))),
                                           args...)
-            println("rerouted")
-            Cassette.overdub(
+            Cassette.recurse(
                 ctx,
                 invoke,
                 $(esc(g.args[1])),
                 $(esc(:(Tuple{$(g.args[2:end]...)}))),
-                args...)
+                $(map(_name, fargs)...))
         end
     end
 end
 
-@reroute LinearAlgebra.BLAS.dot LinearAlgebra.dot(Any, Any)
-@reroute LinearAlgebra.BLAS.axpy! LinearAlgebra.axpy!(Any,
+@reroute LinearAlgebra.BLAS.dot(x,y) LinearAlgebra.dot(Any, Any)
+@reroute LinearAlgebra.BLAS.axpy!(x, y) LinearAlgebra.axpy!(Any,
                                                       AbstractArray,
                                                       AbstractArray)
+@reroute LinearAlgebra.mul!(y::AbstractVector,
+                            A::AbstractVecOrMat,
+                            x::AbstractVector,
+                            α::Number,
+                            β::Number) LinearAlgebra.mul!(AbstractVector,
+                                                          AbstractVecOrMat,
+                                                          AbstractVector,
+                                                          Number,
+                                                          Number)

test/basics.jl

@@ -32,3 +32,15 @@ let
     @test sparsity!(testsparse!, out, x) == sparse([1,2,1,2,3,2,3,4,3,4],
                                                    [1,1,2,2,2,3,3,3,4,4], true)
 end
+
+@testset "BLAS" begin
+    function f(out,in)
+        A = rand(length(in), length(in))
+        out .= A * in
+        return nothing
+    end
+
+    x = [1.0:10;]
+    out = similar(x)
+    @test all(sparsity!(f, out, x) .== 1)
+end