testing

docs/src/docs/array_api.md

@@ -151,14 +151,14 @@ be useful to distinguish between different uses of the same function using the
 `tag` keyword argument to `compute` or `fuse`.  For example, we may wish to
 distinguish one spmv from another, as follows:
 
-```jldoctest example2
+```jldoctest example2; setup=:(using Finch)
 julia> A = rand(1000, 1000); B = rand(1000, 1000); C = fsprand(1000, 1000, 0.0001);
 
-julia> fused((A, B, C) -> C .* (A * B), tag=:very_sparse_sddmm);
+julia> fused((A, B, C) -> C .* (A * B), A, B, C, tag=:very_sparse_sddmm);
 
 julia> C = fsprand(1000, 1000, 0.9);
 
-julia> fused((A, B, C) -> C .* (A * B), tag=:very_dense_sddmm);
+julia> fused((A, B, C) -> C .* (A * B), A, B, C, tag=:very_dense_sddmm);
 
 ```
 

src/Finch.jl

@@ -49,7 +49,7 @@ export diagmask, lotrimask, uptrimask, bandmask, chunkmask
 export scale, products, offset, permissive, protocolize, swizzle, toeplitz, window
 export PlusOneVector
 
-export lazy, compute, tensordot, @einsum
+export lazy, compute, fused, tensordot, @einsum
 
 export choose, minby, maxby, overwrite, initwrite, filterop, d
 

src/interface/lazy.jl

@@ -487,7 +487,7 @@ program:
     - `tag=:global`: A tag to distinguish between different classes of inputs for the same program.
 """
 function fused(f, args...; kwargs...)
-    compute(f(map(LazyTensor, args...)), kwargs...)
+    compute(f(map(LazyTensor, args)...); kwargs...)
 end
 
 current_scheduler = Ref{Any}(default_scheduler())