Merge pull request #86 from mcabbott/unsafe

Mark all shifted indices unsafe in gradients

src/macro.jl

@@ -310,6 +310,7 @@ function parse_input(expr, store)
     unique!(store.leftind)
     store.sharedind = unique!(setdiff(store.sharedind, store.notfree))
     store.rightind = unique!(setdiff(store.rightind, store.notfree))
+    union!(store.unsaferight, store.shiftedind)
     any(==(:_), vcat(store.leftind, store.rightind)) && throw("can't use _ as an index name")
 
     unique!(store.outpre) # kill mutiple assertions, and evaluate any f(A) only once

test/cuda.jl

@@ -2,7 +2,7 @@
 using Tullio, Test
 using CUDA, KernelAbstractions
 CUDA.allowscalar(false)
-using Tracker
+using Tracker, ForwardDiff
 @tullio grad=Base
 
 # matmul
@@ -17,14 +17,14 @@ A = rand(3,40); B = rand(40,500);
 @test ΔA ≈ ones(3,500) * B'
 @test cu(ΔA) ≈ Tracker.gradient((A,B) -> sum(mul(A, B)), cu(A), cu(B))[1]
 
-#=
 # shifts
 @tullio D[i,j] := A[i,j+k]  k in 0:10
 @test axes(D) == (1:3, 1:30)
 @tullio cD[i,j] := cu(A)[i,j+k]  k in 0:10
 @test cD isa CuArray
 @test cD ≈ cu(D)
 
+#=
 # ranges
 @tullio E[i,j] := A[i,j+k-1] + (-1:0.5:1)[k]
 @test axes(E) == (1:3, 1:36)