Skip to content

Commit ae5a0b6

Browse files
committed
Add tests for DaggerForm
1 parent 3e96e40 commit ae5a0b6

File tree

2 files changed

+9
-1
lines changed

2 files changed

+9
-1
lines changed

Project.toml

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,9 +47,10 @@ Unitful = "1.1"
4747
julia = "1.2"
4848

4949
[extras]
50+
Dagger = "d58978e5-989f-55fb-8d15-ea34adc7bf54"
5051
OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
5152
StochasticDiffEq = "789caeaf-c7a9-5a7d-9973-96adeb23e2a0"
5253
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
5354

5455
[targets]
55-
test = ["OrdinaryDiffEq", "Test", "StochasticDiffEq"]
56+
test = ["Dagger", "OrdinaryDiffEq", "Test", "StochasticDiffEq"]

test/bigsystem.jl

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -54,10 +54,14 @@ using Distributed
5454
addprocs(4)
5555
distributedf = eval(ModelingToolkit.build_function(du,u,parallel=ModelingToolkit.DistributedForm())[2])
5656

57+
using Dagger
58+
daggerf = eval(ModelingToolkit.build_function(du,u,parallel=ModelingToolkit.DaggerForm())[2])
59+
5760
jac = sparse(ModelingToolkit.jacobian(vec(du),vec(u),simplify=false))
5861
serialjac = eval(ModelingToolkit.build_function(vec(jac),u)[2])
5962
multithreadedjac = eval(ModelingToolkit.build_function(vec(jac),u,parallel=ModelingToolkit.MultithreadedForm())[2])
6063
distributedjac = eval(ModelingToolkit.build_function(vec(jac),u,parallel=ModelingToolkit.DistributedForm())[2])
64+
daggerjac = eval(ModelingToolkit.build_function(vec(jac),u,parallel=ModelingToolkit.DaggerForm())[2])
6165

6266
MyA = zeros(N,N)
6367
AMx = zeros(N,N)
@@ -66,15 +70,18 @@ DA = zeros(N,N)
6670
f(_du,_u,nothing,0.0)
6771
multithreadedf(_du,_u)
6872
#distributedf(_du,_u)
73+
#daggerf(_du,_u)
6974

7075
#=
7176
using BenchmarkTools
7277
@btime f(_du,_u,nothing,0.0)
7378
@btime multithreadedf(_du,_u)
7479
@btime distributedf(_du,_u)
80+
@btime daggerf(_du,_u)
7581
7682
_jac = similar(jac,Float64)
7783
@btime serialjac(_jac,_u)
7884
@btime multithreadedjac(_jac,_u)
7985
@btime distributedjac(_jac,_u)
86+
@btime daggerjac(_jac,_u)
8087
=#

0 commit comments

Comments
 (0)