Skip to content

Commit f9e45a0

Browse files
author
Sathvik Bhagavan
committed
refactor: parser using Symbolics [skip ci]
1 parent 6194b64 commit f9e45a0

7 files changed

+474
-623
lines changed

Diff for: src/NeuralPDE.jl

+8-3
Original file line numberDiff line numberDiff line change
@@ -19,8 +19,9 @@ using RuntimeGeneratedFunctions
1919
using Statistics
2020
using ArrayInterface
2121
import Optim
22-
using Symbolics: wrap, unwrap, arguments, operation
23-
using SymbolicUtils
22+
using Symbolics: wrap, unwrap, arguments, operation, symtype, @arrayop, Arr
23+
using SymbolicUtils.Code
24+
using SymbolicUtils: Prewalk, Postwalk, Chain
2425
using AdvancedHMC, LogDensityProblems, LinearAlgebra, Functors, MCMCChains
2526
using MonteCarloMeasurements: Particles
2627
using ModelingToolkit: value, nameof, toexpr, build_expr, expand_derivatives, Interval,
@@ -32,7 +33,9 @@ using SciMLBase: @add_kwonly, parameterless_type
3233
using UnPack: @unpack
3334
import ChainRulesCore, Lux, ComponentArrays
3435
using Lux: FromFluxAdaptor, recursive_eltype
35-
using ChainRulesCore: @non_differentiable
36+
using ChainRulesCore: @non_differentiable, @ignore_derivatives
37+
using PDEBase: AbstractVarEqMapping, VariableMap, cardinalize_eqs!, get_depvars,
38+
get_indvars, differential_order
3639

3740
RuntimeGeneratedFunctions.init(@__MODULE__)
3841

@@ -41,13 +44,15 @@ abstract type AbstractPINN end
4144
abstract type AbstractTrainingStrategy end
4245

4346
include("pinn_types.jl")
47+
include("eq_data.jl")
4448
include("symbolic_utilities.jl")
4549
include("training_strategies.jl")
4650
include("adaptive_losses.jl")
4751
include("ode_solve.jl")
4852
# include("rode_solve.jl")
4953
include("dae_solve.jl")
5054
include("transform_inf_integral.jl")
55+
include("loss_function_generation.jl")
5156
include("discretize.jl")
5257
include("neural_adapter.jl")
5358
include("advancedHMC_MCMC.jl")

0 commit comments

Comments
 (0)