Skip to content

Commit 6f395db

Browse files
committed
skip, etc
1 parent 5611aa8 commit 6f395db

3 files changed

Lines changed: 19 additions & 4 deletions

File tree

test/gradients.jl

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,9 @@ end
6565
@test abs2_grad _gradient(v -> (@tullio s := abs2(1 + v[i]^2)), va)[1]
6666

6767
end
68+
69+
println("... 69")
70+
6871
@testset "zero-arrays" begin
6972

7073
# Using zero-dim arrays fails on ReverseDiff & Tracker
@@ -103,6 +106,9 @@ end
103106
# [1.0, 4.0, 9.0, 16.0, 25.0, 36.0, 0.0, 64.0, 81.0, 100.0, 121.0] ≈ [1.0, 4.0, 9.0, 16.0, 25.0, 36.0, 49.0, 64.0, 81.0, 100.0, 121.0]
104107

105108
end
109+
110+
println("... 110")
111+
106112
@testset "shifts, etc" begin
107113

108114
c1(N,K) = @tullio M[x,y,c] := N[x+i-1, y+j-1,c] * K[i,j]
@@ -155,6 +161,9 @@ end
155161
end
156162

157163
end
164+
165+
println("... 165")
166+
158167
@testset "from TensorTrace" begin
159168
# These can all be handled using TensorOperations
160169

@@ -221,6 +230,8 @@ end
221230

222231
end
223232

233+
println("... 233")
234+
224235
if Tullio._GRAD[] != :Dual
225236
#=
226237
@testset "products" begin
@@ -316,6 +327,9 @@ if Tullio._GRAD[] != :Dual
316327
# I suspect that @avx is re-ordering loops, which makes onlyone() incorrect.
317328

318329
end
330+
331+
println("... 331")
332+
319333
@testset "finalisers" begin
320334

321335
norm2(m) = @tullio n[i] := m[i,j]^2 |> sqrt
@@ -355,3 +369,5 @@ if GRAD == :Zygote
355369

356370
end
357371
end
372+
373+
println("... 373!")

test/parsing.jl

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -79,14 +79,13 @@ using Tullio, Test, LinearAlgebra
7979
@test A2 == 2 .* A
8080

8181
# broadcasting
82-
@tullio S[i] := sqrt.(M[:,i]) # avx & grad now disabled by try/catch
83-
# @tullio T[i] := A[i] .+ A[j] # dot does nothing, fails with LoopVectorization loaded
82+
@tullio S[i] := sqrt.(M[:,i])
8483

8584
# scope
8685
f(x,k) = @tullio y[i] := x[i] + i + $k
8786
@test f(ones(3),j) == 1 .+ (1:3) .+ j
8887

89-
g(x) = @tullio y := sqrt(x[i])
88+
g(x) = @tullio y := sqrt(x[i]) avx=false
9089
@test g(fill(4,5)) == 10
9190

9291
# ranges

test/runtests.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ if test_group in ["all", "1"]
2121
include("group-1.jl")
2222
end
2323

24-
if test_group in ["all", "2"] && VERSION <= v"1.6" # KA testing time-out https://github.com/JuliaGPU/KernelAbstractions.jl/issues/155
24+
if test_group in ["all", "2"]
2525
include("group-2.jl")
2626
end
2727

0 commit comments

Comments
 (0)