rubydragon/metagraphoptimization.jl
1
0
Fork 0
Code Issues 6 Pull Requests Actions Packages Projects Releases Wiki Activity

Add device info to nodes during scheduling

Browse Source
This commit is contained in:
Anton Reinhard
2023-10-12 00:29:48 +02:00
parent 3267daadfd
commit 9b28601f18
4 changed files with 33 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
test/unit_tests_execution.jl
View File

@@ -30,6 +30,9 @@ include("../examples/profiling_utilities.jl")
graph = parse_dag(joinpath(@__DIR__, "..", "input", "AB->AB.txt"), ABCModel())
@test isapprox(execute(graph, process_2_2, machine, particles_2_2), expected_result; rtol = 0.001)
# graph should be fully scheduled after being executed
@test is_scheduled(graph)
func = get_compute_function(graph, process_2_2, machine)
@test isapprox(func(particles_2_2), expected_result; rtol = 0.001)
end
@@ -39,9 +42,13 @@ include("../examples/profiling_utilities.jl")
for i in 1:1000
graph = parse_dag(joinpath(@__DIR__, "..", "input", "AB->AB.txt"), ABCModel())
random_walk!(graph, 50)
@test is_valid(graph)
@test isapprox(execute(graph, process_2_2, machine, particles_2_2), expected_result; rtol = 0.001)
# graph should be fully scheduled after being executed
@test is_scheduled(graph)
end
end