From 074d7f49b14e4d4b93d0f4679fd9ea72fb2679f0 Mon Sep 17 00:00:00 2001 From: William Moses Date: Mon, 12 Aug 2024 10:02:20 -0400 Subject: [PATCH] Update compiler.jl --- 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