onnx-mlir
[WIP] High-Level (Coase grain) parallelization with ONNXParallelOp and ONNXForkOp based on OpenMP
#2756
Open
Open
[WIP] High-Level (Coase grain) parallelization with ONNXParallelOp and ONNXForkOp based on OpenMP #2756
imaihal
wants to merge
37 commits into
onnx:main
from
imaihal:parallelop_forkop_with_omp
Added ONNXParallelOp and ONNXForkOp
c9f68fd3
Add lowering pass for ONNXParallelOp and ONNXForkOp
8f87a9cd
Remove SCFToCF Pass
9234f25c
New files for ONNXParallelOp and ONNXForkOp
98aef8ac
Revert: enable deallocation op as ilegal op
796aa80c
Clean up by creating a function for moving alloc op
84f9f0d1
Merge branch 'main' into parallelop_forkop_with_omp
f24dac6f
Update KrnlToAffine pass to support scf::ifOp in body of KrnlIterateOp.
e6591f02
Update comments.
0db91dc2
Merge branch 'main' into parallelop_forkop_with_omp
22cbc08d
Merge branch 'main' into parallelop_forkop_with_omp
f24c9a64
Dynamic dim support v1.
c4f0a3ba
Fix static dim case.
f7fc4cae
Merge branch 'main' into parallelop_forkop_with_omp
38161c11
Add KrnlRegionOp in region of scf::ifOp.
c8f8075c
Merge branch 'main' into parallelop_forkop_with_omp
0fbfa9d6
Update dynamic dim support.
8d10b0bf
Merge branch 'main' into parallelop_forkop_with_omp
ed5fec11
Update condition to apply unstickStickRemoval optimization
9077e7d5
Update comments
91b6f468
Merge branch 'main' into parallelop_forkop_with_omp
0e06aec5
Merge branch 'main' into parallelop_forkop_with_omp
c73d8ad1
Merge branch 'main' into parallelop_forkop_with_omp
d9342733
Added lit test for shape inference.
9a45f0f4
Fix lit test.
dad1f998
Update dynamic dim support
6d111a74
Merge branch 'main' into parallelop_forkop_with_omp
44396c4e
Fix insertion of KrnlRegionOp.
37e12b2a
Add shapeHelpr for ONNXParallelOp and ONNXForkOp.
c5e76633
Add dimAnalysis for ONNXParallelOp and ONNXForkOp.
2f31a162
Merge branch 'main' into parallelop_forkop_with_omp
b70fbcc8
Merge branch 'main' into parallelop_forkop_with_omp
b6ee5091
Merge branch 'main' into parallelop_forkop_with_omp
06cc16cb
Fix inserting regionOp in iterateOp.
2bcabe8a
Add scripts for rewriting a model for operator-level parallelization.
c837809d
Merge branch 'main' into parallelop_forkop_with_omp
53805d9a
black format
17ac56d1
Reviewers
No reviews
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub