From 65770c98bee7ef7da57bfd135e6f8f91cee9823c Mon Sep 17 00:00:00 2001 From: William Moses Date: Mon, 12 Aug 2024 07:45:10 -0700 Subject: [PATCH] Update compiler.jl (#1729) --- src/compiler.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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