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

experiments (#1)
All checks were successful
MetagraphOptimization_CI / docs (push) Successful in 10m41s
Details
MetagraphOptimization_CI / test (push) Successful in 30m40s
Details

Browse Source 
Co-authored-by: Anton Reinhard <anton.reinhard@proton.me>
Reviewed-on: #1
This commit is contained in:
rubydragon
2024-05-08 12:03:27 +02:00
parent 82ed774b7e
commit 87dbaf2c32
155 changed files with 5372 additions and 1029 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
src/operation/find.jl
View File

@ -197,8 +197,7 @@ function generate_operations(graph::DAG)
# launch thread for node reduction insertion
# remove duplicates
nr_task = @task nr_insertion!(graph.possibleOperations, generatedReductions)
schedule(nr_task)
nr_task = @spawn nr_insertion!(graph.possibleOperations, generatedReductions)
# --- find possible node fusions ---
@threads for node in nodeArray
@ -223,8 +222,7 @@ function generate_operations(graph::DAG)
end
# launch thread for node fusion insertion
nf_task = @task nf_insertion!(graph, graph.possibleOperations, generatedFusions)
schedule(nf_task)
nf_task = @spawn nf_insertion!(graph, graph.possibleOperations, generatedFusions)
# find possible node splits
@threads for node in nodeArray
@ -234,8 +232,7 @@ function generate_operations(graph::DAG)
end
# launch thread for node split insertion
ns_task = @task ns_insertion!(graph.possibleOperations, generatedSplits)
schedule(ns_task)
ns_task = @spawn ns_insertion!(graph.possibleOperations, generatedSplits)
empty!(graph.dirtyNodes)