Merge pull request #8 from DJ4Earth/fix_Enzyme_slow_issue

fix Enzyme too slow issue, add ad test back
DJ4Earth · Mar 13, 2024 · d9c630f · d9c630f
2 parents 35e2d78 + f528e71
commit d9c630f
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 6 deletions.
diff --git a/src/core/control.jl b/src/core/control.jl
@@ -1,9 +1,10 @@
 using Enzyme
 using Optimization, OptimizationOptimJL
 
-Enzyme.API.looseTypeAnalysis!(false)
-Enzyme.API.strictAliasing!(false)
-Enzyme.API.typeWarning!(false)
+#Enzyme.API.looseTypeAnalysis!(false)
+#Enzyme.API.strictAliasing!(false)
+#Enzyme.API.typeWarning!(false)
+Enzyme.Compiler.RunAttributor[] = false
 
 function Control_Core(md::model, femmodel::FemModel) #{{{
 	# Compute gradient 

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -17,8 +17,8 @@ end
 	end
 
 	# AD test
-	#@time include("testad.jl")
-	#@time include("testad2.jl")
+	@time include("testad.jl")
+	@time include("testad2.jl")
 
 	# GPU test
 	#@time include("testGPU.jl")

diff --git a/test/testad.jl b/test/testad.jl
@@ -3,6 +3,9 @@ module enzymeDiff
 using dJUICE
 using MAT
 using Test
+using Enzyme
+
+Enzyme.Compiler.RunAttributor[] = false
 
 #Load model from MATLAB file
 #file = matopen(joinpath(@__DIR__, "..", "data","temp12k.mat")) #BIG model

diff --git a/test/testad2.jl b/test/testad2.jl
@@ -1,7 +1,9 @@
-#!/Applications/Julia-1.8.app/Contents/Resources/julia/bin/julia --project
 using dJUICE
 using MAT
 using Test
+using Enzyme
+
+Enzyme.Compiler.RunAttributor[] = false
 
 #Load model from MATLAB file
 #file = matopen(joinpath(@__DIR__, "..", "data","temp12k.mat")) #BIG model