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

7
src/MetagraphOptimization.jl
View File

@@ -78,7 +78,7 @@ export gen_graph
export execute
export parse_dag, parse_process
export gen_process_input
export get_compute_function
export get_compute_function, get_cuda_kernel
export gen_tape, execute_tape
# estimator
@@ -86,7 +86,8 @@ export cost_type, graph_cost, operation_effect
export GlobalMetricEstimator, CDCost
# optimization
export AbstractOptimizer, GreedyOptimizer, ReductionOptimizer, RandomWalkOptimizer
export AbstractOptimizer, GreedyOptimizer, RandomWalkOptimizer
export ReductionOptimizer, SplitOptimizer, FusionOptimizer
export optimize_step!, optimize!
export fixpoint_reached, optimize_to_fixpoint!
@@ -166,6 +167,8 @@ include("optimization/interface.jl")
include("optimization/greedy.jl")
include("optimization/random_walk.jl")
include("optimization/reduce.jl")
include("optimization/fuse.jl")
include("optimization/split.jl")
include("models/interface.jl")
include("models/print.jl")