Improve precompilation for extensions

sintefmath · Jun 6, 2023 · 5f7aba7 · 5f7aba7 · moyner · Jun 6, 2023
1 parent 7612063
commit 5f7aba7
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "JutulDarcy"
 uuid = "82210473-ab04-4dce-b31b-11573c4f8e0a"
 authors = ["Olav Møyner <olav.moyner@gmail.com>"]
-version = "0.2.4"
+version = "0.2.5"
 
 [deps]
 AlgebraicMultigrid = "2169fc97-5a83-5252-b627-83903c6c433c"

diff --git a/ext/JutulDarcyHYPREExt/JutulDarcyHYPREExt.jl b/ext/JutulDarcyHYPREExt/JutulDarcyHYPREExt.jl
@@ -3,5 +3,16 @@ module JutulDarcyHYPREExt
     using HYPRE
     using Jutul
     using SparseArrays
+    using PrecompileTools
     include("cpr.jl")
+
+    @compile_workload begin
+        targets = [(true, :csc), (true, :csr)]
+        # MPI, trivial partition
+        JutulDarcy.precompile_darcy_multimodels(targets,
+            dims = (4, 1, 1),
+            precond = :cpr,
+            amg_type = :hypre
+        )
+    end
 end
diff --git a/ext/JutulDarcyPartitionedArraysExt/JutulDarcyPartitionedArraysExt.jl b/ext/JutulDarcyPartitionedArraysExt/JutulDarcyPartitionedArraysExt.jl
@@ -129,7 +129,11 @@ module JutulDarcyPartitionedArraysExt
         # MPI, trivial partition
         JutulDarcy.precompile_darcy_multimodels(targets,
             dims = (4, 1, 1),
-            setuparg = (mode = :mpi, ),
+            default_linsolve = false,
+            setuparg = (
+                mode = :mpi,
+                precond = :ilu0
+                ),
             split_wells = true
         )
         # Native PArray, non-trivial partition