EnzymeAD / EnzymeAD/Enzyme.jl

Atomic fadd of vector

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

Description

```julia
using LoopVectorization, Enzyme

function simple_lv_loop(x)
y = similar(x)
@turbo for i ∈ eachindex(x)
y[i] = abs2(sin(x[i]))
end
return sum(y)
end

x = rand(10)

simple_lv_loop(x)

Enzyme.gradient(Reverse, simple_lv_loop, x)
```

Logs: https://gist.github.com/avik-pal/d1f2ab140db39bac9b8e146647618bb0

```julia
julia> versioninfo()
Julia Version 1.10.4
Commit 48d4fd48430 (2024-06-04 10:41 UTC)
Build Info:
Official https://julialang.org/ release
Platform Info:
OS: Linux (x86_64-linux-gnu)
CPU: 12 × AMD Ryzen 5 4600H with Radeon Graphics
WORD_SIZE: 64
LIBM: libopenlibm
LLVM: libLLVM-15.0.7 (ORCJIT, znver2)
Threads: 12 default, 0 interactive, 6 GC (on 12 virtual cores)
Environment:
JULIA_EDITOR = vim
```

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.