Michael Abbott
                                            Michael Abbott
                                        
                                    Similarly `Int[1 2 3]` and `Float32[1 2; 3 4]`. Ideally these would share as much code as possible with the existing cat rules, perhaps just changing a few lines. Maybe...
Yes this really ought to work. It can see the `1` so it should be smart enough to make a `OneTo` for you. (In fact I thought there was already...
OK there is code to make OneTo, but it's too restrictive: https://github.com/mcabbott/Tullio.jl/blob/5d3105552e87efe6de981fb4d78e23e7df587fbb/src/macro.jl#L633-L637 ```julia julia> r = :(1:N) :(1:N) julia> r.args[1] == :(:) && length(r.args) == 3 true julia> r.args[2] ==...
One way around this would be to make an SMatrix: ```julia julia> using StaticArrays julia> @tullio _[n]:=inv(SA[x[n] -1.0;-1.0 x[n]-e[k]]) 10-element Vector{SMatrix{2, 2, Float64, 4}}: [-59.268603211659475 49.268603211659475; 49.268603211659475 -49.268603211659475] [-105.02504355561057 71.68614498769712;...
I can reproduce this, seems to be a bug. Thanks for the report! Slightly shorter version below: ```julia julia> using Tullio, Zygote, CUDA, KernelAbstractions julia> batchmul(A, B) = @tullio C[i,j,k]...
Do you mean the order of loop nesting, or the order of traversal? The latter is (for large enough arrays) done in parallel.
In this case `i` is innermost. This info block has the order (which it's just taking from the order supplied); `verbose=2` prints the actual loops: ```julia julia> @tullio A[_] :=...
> would the nesting of loop also change? Yes. When there are arrays, you can't really change it, there's no option to specify this. (Sometimes re-ordering an expression `a[...] *...
Sorry I thought I replied to this, but must not have pasted what I wrote. Maybe this indicates how active this project is -- I aim to keep it working...
I think the issue is the `::Float64` annotation with LoopVectorization: ```julia julia> using Tullio, LoopVectorization julia> t=collect(range(start=-3,step=0.01,stop=3)); julia> nb(e)::Float64 = inv(expm1(e/1e-4)); julia> @tullio test1[n] := nb(t[n]); ┌ Warning: #= /Users/me/.julia/packages/Tullio/NGyNM/src/macro.jl:1093...