EnzymeAD / EnzymeAD/Enzyme.jl

`LLVMException` on forward mode matmul operations + cuda

Open
#3,442 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

here is a MWE @wsmoses

```
using CUDA, Enzyme
using LinearAlgebra: dot

a = CuArray(randn(Float32, 8)); b = CuArray(randn(Float32, 8))
da = CuArray(randn(Float32, 8)); db = CuArray(randn(Float32, 8))

Enzyme.autodiff(Forward, dot, Duplicated, Duplicated(a, da), Duplicated(b, db))
```

This applies for forward mode: `dot(a,b)`, `mul!(C,A,B,1,0)`, `sum(A*B)`

_Originally posted by @rsenne in https://github.com/EnzymeAD/Enzyme.jl/pull/3148#discussion_r3670199697_

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.