Skip to content

Commit

Permalink
Mark test as unbroken
Browse files Browse the repository at this point in the history
  • Loading branch information
pxl-th committed Jan 21, 2025
1 parent 811ae8c commit fc5dac1
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions test/ext_cuda/cuda.jl
Original file line number Diff line number Diff line change
Expand Up @@ -106,7 +106,7 @@ end
# Trivial functions
@test gradient(x -> sum(abs, gpu(x)), a)[1] isa Matrix
@test gradient(x -> sum(gpu(x)), a)[1] isa Matrix
@test_broken gradient(x -> sum(gpu(x)), a')[1] isa Matrix # sum(::Adjoint{T,CuArray}) makes a Fill
@test gradient(x -> sum(gpu(x)), a')[1] isa Matrix # sum(::Adjoint{T,CuArray}) makes a Fill
@test gradient(x -> sum(abs, cpu(x)), ca)[1] isa CuArray
# This test should really not go through indirections and pull out Fills for efficiency
# but we forcefully materialise. TODO: remove materialising CuArray here
Expand Down Expand Up @@ -207,4 +207,4 @@ end
@test collect(post2) isa Vector{<:NamedTuple{(:x, :y)}} # collect makes no sense, but check eltype?

# @test_throws Exception gpu(((x = Flux.DataLoader(X), y = Y),))
end
end

0 comments on commit fc5dac1

Please sign in to comment.