EnzymeAD / EnzymeAD/Enzyme.jl

Hella slow `test_reverse`

Open
#2,887 0 comments 0 reactions 0 assignees View on GitHub
Dominant language
Julia
Stars
586
Forks
108
Avg merge
1d 5h
Merged PRs (30d)
44

Description

MWE:

```julia
using Enzyme, EnzymeTestUtils
using TestExtras
using StableRNGs
using LinearAlgebra

@timedtestset "GEAM AD Rules with eltype $T" for T in (Float64, ComplexF64)
rng = StableRNG(12345)
m = 54
A = randn(rng, T, m, m)
B = randn(rng, T, m, m)
@timedtestset "reverse: RT $RT, TA $TA" for RT in (Duplicated,), TA in (Duplicated,)
@time "$T" test_reverse(+, RT, (A, TA), (B, TA))
end
end
```

Result:

```
Float64: 48.088790 seconds (2.25 G allocations: 137.453 GiB, 8.99% gc time, 33.01% compilation time)
Test Summary: | Pass Total Time
GEAM AD Rules with eltype Float64 | 11 11 48.1s
ComplexF64: 75.736083 seconds (4.39 G allocations: 466.193 GiB, 16.56% gc time, 1.99% compilation time)
Test Summary: | Pass Total Time
GEAM AD Rules with eltype ComplexF64 | 11 11 1m15.7s
```

It seems crazy to me that this takes a whole minute for each test case. I'm guessing it's something to do with the testing frameworks memory mapping?

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.