Skip to content

Commit

Permalink
Update test/cuda/layers.jl
Browse files Browse the repository at this point in the history
Co-authored-by: Dhairya Gandhi <dhairya@juliacomputing.com>
  • Loading branch information
CarloLucibello and DhairyaLGandhi authored Jul 13, 2021
1 parent dfb390d commit ac73906
Showing 1 changed file with 1 addition and 2 deletions.
3 changes: 1 addition & 2 deletions test/cuda/layers.jl
Original file line number Diff line number Diff line change
Expand Up @@ -279,7 +279,7 @@ end
gs_g = gradient(() -> sum(tanh.(m_g(x_g))), params(m_g))
@test collect(gs_g[m_g.weight]) gs[m.weight]

@testset "repeated indexes" begin
@testset "repeated indices" begin
vocab_size, embed_size = 5, 2
m = Flux.Embedding(vocab_size, embed_size)

Expand All @@ -294,4 +294,3 @@ end
@test collect(gs_g[m_g.weight]) gs[m.weight]
end
end

0 comments on commit ac73906

Please sign in to comment.