Add symv and bump jll

EnzymeAD · Sep 28, 2024 · e05be35 · e05be35
1 parent 7c5b018
commit e05be35
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -36,7 +36,7 @@ BFloat16s = "0.2, 0.3, 0.4, 0.5"
 CEnum = "0.4, 0.5"
 ChainRulesCore = "1"
 EnzymeCore = "0.8.3"
-Enzyme_jll = "0.0.150, 0.0.151"
+Enzyme_jll = "0.0.151"
 GPUCompiler = "0.21, 0.22, 0.23, 0.24, 0.25, 0.26, 0.27"
 LLVM = "6.1, 7, 8, 9"
 LogExpFunctions = "0.3"

diff --git a/src/compiler.jl b/src/compiler.jl
@@ -7106,7 +7106,7 @@ function GPUCompiler.codegen(
     disableFallback = String[]
 
     ForwardModeDerivatives =
-        ("nrm2", "dot", "gemm", "gemv", "axpy", "copy", "scal", "symm", "syrk", "potrf")
+        ("nrm2", "dot", "gemm", "gemv", "axpy", "copy", "scal", "symv", "symm", "syrk", "potrf")
     ReverseModeDerivatives = (
         "nrm2",
         "dot",
@@ -7115,6 +7115,7 @@ function GPUCompiler.codegen(
         "axpy",
         "copy",
         "scal",
+        "symv",
         "symm",
         "trmv",
         "syrk",