re-enable one cuda test

mcabbott · Mar 13, 2021 · 1d7d150 · 1d7d150
1 parent 8ec26b5
commit 1d7d150
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/cuda.jl b/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)