fix

EnzymeAD · Jun 18, 2024 · b9a5ed3 · b9a5ed3
1 parent 2b6a9e8
commit b9a5ed3
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 11 deletions.
diff --git a/Project.toml b/Project.toml
@@ -11,11 +11,22 @@ GPUCompiler = "61eb1bfa-7361-4325-ad38-22787b887f55"
 LLVM = "929cbde3-209d-540e-8aea-75f648917ca0"
 Libdl = "8f399da3-3557-5675-b5ff-fb832c97cbdb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+OMEinsum = "ebe7aa44-baf0-506c-a96f-8464559b3922"
 ObjectFile = "d8793406-e978-5875-9003-1fc021f44a92"
 Preferences = "21216c6a-2e73-6563-6e65-726566657250"
 Printf = "de0858da-6303-5e67-8744-51eddeeeb8d7"
 Random = "9a3f8284-a2c9-5f02-9a11-845980a1fd5c"
 
+[weakdeps]
+ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
+StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
+
+[extensions]
+EnzymeChainRulesCoreExt = "ChainRulesCore"
+EnzymeSpecialFunctionsExt = "SpecialFunctions"
+EnzymeStaticArraysExt = "StaticArrays"
+
 [compat]
 CEnum = "0.4, 0.5"
 ChainRulesCore = "1"
@@ -29,17 +40,7 @@ SpecialFunctions = "1, 2"
 StaticArrays = "1"
 julia = "1.6"
 
-[extensions]
-EnzymeChainRulesCoreExt = "ChainRulesCore"
-EnzymeSpecialFunctionsExt = "SpecialFunctions"
-EnzymeStaticArraysExt = "StaticArrays"
-
 [extras]
 ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
 SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
 StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
-
-[weakdeps]
-ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
-SpecialFunctions = "276daf66-3868-5448-9aa4-cd146d93841b"
-StaticArrays = "90137ffa-7385-5640-81b9-e52037218182"
diff --git a/src/compiler.jl b/src/compiler.jl
@@ -5160,7 +5160,7 @@ function GPUCompiler.codegen(output::Symbol, job::CompilerJob{<:EnzymeTarget};
     # Tablegen BLAS does not support forward mode yet
     if !(mode == API.DEM_ForwardMode && Enzyme.API.runtimeActivity())
         for ty in ("s", "d")
-            for func in mode == API.DEM_ForwardMode ? ForwardModeDerivatives : ReverseModeDerivatives
+            for func in (mode == API.DEM_ForwardMode ? ForwardModeDerivatives : ReverseModeDerivatives)
                 for prefix in ("", "cblas_")
                     for ending in ("", "_", "64_", "_64_")
                         push!(disableFallback, prefix*ty*func*ending)