revert broadcast

SciML · Sep 22, 2021 · 7d728dd · 7d728dd
1 parent 087c775
commit 7d728dd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/docs/src/pinn/neural_adapter.md b/docs/src/pinn/neural_adapter.md
@@ -170,6 +170,7 @@ for i in 1:count_decomp
     phi_in(cord) = phis[i-1](cord,reses[i-1].minimizer)
     phi_bound(x,y) = phi_in(vcat(x,y))
     @register phi_bound(x,y)
+    Base.Broadcast.broadcasted(::typeof(phi_bound), x,y) = phi_bound(x,y)
     bcs_ = create_bcs(domains_[1].domain, phi_bound)
     @named pde_system_ = PDESystem(eq, bcs_, domains_, [x, y], [u(x, y)])
     push!(pde_system_map,pde_system_)

diff --git a/test/neural_adapter_tests.jl b/test/neural_adapter_tests.jl
@@ -196,6 +196,7 @@ for i in 1:count_decomp
     # phi_bound(x,y) = if (x isa Matrix)  phi_in(vcat(x, fill(y,size(x)))) else  phi_in(vcat(fill(x,size(y)),y)) end
     phi_bound(x,y) = phi_in(vcat(x,y))
     @register phi_bound(x,y)
+    Base.Broadcast.broadcasted(::typeof(phi_bound), x,y) = phi_bound(x,y)
     bcs_ = create_bcs(domains_[1].domain, phi_bound)
     @named pde_system_ = PDESystem(eq, bcs_, domains_, [x, y], [u(x,y)])
     push!(pde_system_map,pde_system_)