Skip to content

Commit

Permalink
Update compiler.jl
Browse files Browse the repository at this point in the history
  • Loading branch information
wsmoses authored Aug 12, 2024
1 parent 410a866 commit 074d7f4
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/compiler.jl
Original file line number Diff line number Diff line change
Expand Up @@ -5538,8 +5538,8 @@ function GPUCompiler.codegen(output::Symbol, job::CompilerJob{<:EnzymeTarget};

disableFallback = String[]

ForwardModeDerivatives = ("nrm2","dot","gemm","gemv","axpy","copy","scal", "syrk", "potrf")
ReverseModeDerivatives = ("nrm2","dot","gemm","gemv","axpy","copy","scal", "trmv", "syrk", "trmm", "trsm", "potrf")
ForwardModeDerivatives = ("nrm2","dot","gemm","gemv","axpy","copy","scal", "symm", "syrk", "potrf")
ReverseModeDerivatives = ("nrm2","dot","gemm","gemv","axpy","copy","scal", "symm", "trmv", "syrk", "trmm", "trsm", "potrf")
ForwardModeTypes = ("s", "d", "c", "z")
ReverseModeTypes = ("s", "d")
# Tablegen BLAS does not support forward mode yet
Expand Down

0 comments on commit 074d7f4

Please sign in to comment.