From 6b15a13055f193f657e5c8bddc8c03f8ba2b24ef Mon Sep 17 00:00:00 2001
From: Paula Respondek <paula.respondek@uni-rostock.de>
Date: Thu, 6 Jun 2024 10:49:33 +0200
Subject: [PATCH 1/2] Change assembly of BilForm

LinearMaps are no longer added but a LinearCombination is made, because the summation of LinearMaps results in a Tuple, which for large DirectProductSpaces causes extremely long compile times.
---
 src/solvers/solver.jl      | 15 +++++++++++++--
 test/test_directproduct.jl |  5 +++++
 2 files changed, 18 insertions(+), 2 deletions(-)

diff --git a/src/solvers/solver.jl b/src/solvers/solver.jl
index aa9bb0e4..8d7715ce 100644
--- a/src/solvers/solver.jl
+++ b/src/solvers/solver.jl
@@ -219,14 +219,17 @@ lift(a::ConvolutionOperators.AbstractConvOp ,I,J,U,V) =
 function assemble(bf::BilForm, X::DirectProductSpace, Y::DirectProductSpace;
     materialize=BEAST.assemble)
 
+    T = Int32
     @assert !isempty(bf.terms)
 
     spaceTimeBasis = isa(X.factors[1], BEAST.SpaceTimeBasis)
 
     if spaceTimeBasis
         p = [numstages(temporalbasis(ch)) for ch in X.factors]
+        lincombv = ConvolutionOperators.LiftedConvOp[]
     else
         p = 1
+        lincombv = LinearMap[]
     end
 
     M = numfunctions.(spatialbasis(X).factors) .* p
@@ -237,7 +240,7 @@ function assemble(bf::BilForm, X::DirectProductSpace, Y::DirectProductSpace;
     U = BlockArrays.blockedrange(M)
     V = BlockArrays.blockedrange(N)
 
-    sum(bf.terms) do term
+    for term in bf.terms
 
         x = X.factors[term.test_id]
         for op in reverse(term.test_ops)
@@ -251,7 +254,15 @@ function assemble(bf::BilForm, X::DirectProductSpace, Y::DirectProductSpace;
         
         a = term.coeff * term.kernel
         z = materialize(a, x, y)
-        lift(z, Block(term.test_id), Block(term.trial_id), U, V)
+
+        Smap = lift(z, Block(term.test_id), Block(term.trial_id), U, V)
+        T = promote_type(T, eltype(Smap))
+        push!(lincombv, Smap)
+    end
+    if spaceTimeBasis
+        return sum(lincombv)
+    else
+        return LinearMaps.LinearCombination{T}(lincombv)
     end
 end
 
diff --git a/test/test_directproduct.jl b/test/test_directproduct.jl
index f8619f0f..b79f6501 100644
--- a/test/test_directproduct.jl
+++ b/test/test_directproduct.jl
@@ -25,4 +25,9 @@ for U in [Float32, Float64]
     t = assemble(T, X, X)
 
     @test size(t) == (nt,nt)
+
+    bilterms = [BEAST.Variational.BilTerm(1,1,Any[],Any[],1,T)]
+
+    BilForm = BEAST.Variational.BilForm(:i, :j, bilterms)
+    @test typeof(assemble(BilForm, X, X)) == LinearMaps.LinearCombination{U, Vector{LinearMap}}
 end
\ No newline at end of file

From 6c6d8feeff54625933bb986cc80f1be3823418be Mon Sep 17 00:00:00 2001
From: Kristof Cools <krcools@gmail.com>
Date: Fri, 28 Jun 2024 14:26:16 +0200
Subject: [PATCH 2/2] update with upstream

---
 test/Manifest.toml         | 18 +++++++++++++++++-
 test/Project.toml          |  1 +
 test/test_directproduct.jl |  3 ++-
 3 files changed, 20 insertions(+), 2 deletions(-)

diff --git a/test/Manifest.toml b/test/Manifest.toml
index 476c9617..8479a59f 100644
--- a/test/Manifest.toml
+++ b/test/Manifest.toml
@@ -2,7 +2,7 @@
 
 julia_version = "1.10.2"
 manifest_format = "2.0"
-project_hash = "571829245fb8bd97876fa91a4e7d398091f62379"
+project_hash = "1b78102b70e82631772273acf6687b12f1d8fa74"
 
 [[deps.ArgTools]]
 uuid = "0dad84c5-d112-42e6-8d28-ef12dabb789f"
@@ -299,6 +299,22 @@ git-tree-sha1 = "71e8ee0f9fe0e86a8f8c7f28361e5118eab2f93f"
 uuid = "18c40d15-f7cd-5a6d-bc92-87468d86c5db"
 version = "5.0.0+0"
 
+[[deps.LinearMaps]]
+deps = ["LinearAlgebra"]
+git-tree-sha1 = "ee79c3208e55786de58f8dcccca098ced79f743f"
+uuid = "7a12625a-238d-50fd-b39a-03d52299707e"
+version = "3.11.3"
+
+    [deps.LinearMaps.extensions]
+    LinearMapsChainRulesCoreExt = "ChainRulesCore"
+    LinearMapsSparseArraysExt = "SparseArrays"
+    LinearMapsStatisticsExt = "Statistics"
+
+    [deps.LinearMaps.weakdeps]
+    ChainRulesCore = "d360d2e6-b24c-11e9-a2a3-2a2ae2dbcce4"
+    SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
+    Statistics = "10745b16-79ce-11e8-11f9-7d13ad32a3b2"
+
 [[deps.LogExpFunctions]]
 deps = ["DocStringExtensions", "IrrationalConstants", "LinearAlgebra"]
 git-tree-sha1 = "c3ce8e7420b3a6e071e0fe4745f5d4300e37b13f"
diff --git a/test/Project.toml b/test/Project.toml
index 4676299c..c9ab94c0 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -5,6 +5,7 @@ CompScienceMeshes = "3e66a162-7b8c-5da0-b8f8-124ecd2c3ae1"
 DelimitedFiles = "8bb1440f-4735-579b-a4ab-409b98df4dab"
 Distributed = "8ba89e20-285c-5b6f-9357-94700520ee1b"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
+LinearMaps = "7a12625a-238d-50fd-b39a-03d52299707e"
 Pkg = "44cfe95a-1eb2-52ea-b672-e2afdf69b78f"
 SauterSchwabQuadrature = "535c7bfe-2023-5c1d-b712-654ef9d93a38"
 SparseArrays = "2f01184e-e22b-5df5-ae63-d93ebab69eaf"
diff --git a/test/test_directproduct.jl b/test/test_directproduct.jl
index b79f6501..fcede8c8 100644
--- a/test/test_directproduct.jl
+++ b/test/test_directproduct.jl
@@ -2,6 +2,7 @@ using CompScienceMeshes
 using BEAST
 
 using Test
+import LinearMaps
 
 for U in [Float32, Float64]
     m1 = meshrectangle(U(1.0), U(1.0), U(0.5))
@@ -29,5 +30,5 @@ for U in [Float32, Float64]
     bilterms = [BEAST.Variational.BilTerm(1,1,Any[],Any[],1,T)]
 
     BilForm = BEAST.Variational.BilForm(:i, :j, bilterms)
-    @test typeof(assemble(BilForm, X, X)) == LinearMaps.LinearCombination{U, Vector{LinearMap}}
+    @test typeof(assemble(BilForm, X, X)) == LinearMaps.LinearCombination{U, Vector{LinearMaps.LinearMap}}
 end
\ No newline at end of file