diff --git a/src/compiler.jl b/src/compiler.jl index af5b32678d..b04550fe0a 100644 --- a/src/compiler.jl +++ b/src/compiler.jl @@ -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