Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[WIP] High-Level (Coase grain) parallelization with ONNXParallelOp and ONNXForkOp based on OpenMP #2756

Draft
wants to merge 37 commits into
base: main
Choose a base branch
from
Draft

[WIP] High-Level (Coase grain) parallelization with ONNXParallelOp and ONNXForkOp based on OpenMP #2756

wants to merge 37 commits into from

Commits on Mar 15, 2024

  1. Added ONNXParallelOp and ONNXForkOp 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    c9f68fd View commit details
    Browse the repository at this point in the history

  2. Add lowering pass for ONNXParallelOp and ONNXForkOp 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    8f87a9c View commit details
    Browse the repository at this point in the history

  3. Remove SCFToCF Pass 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    9234f25 View commit details
    Browse the repository at this point in the history

  4. New files for ONNXParallelOp and ONNXForkOp 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 15, 2024
    Configuration menu
    Copy the full SHA
    98aef8a View commit details
    Browse the repository at this point in the history

Commits on Mar 18, 2024

  1. Revert: enable deallocation op as ilegal op 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 18, 2024
    Configuration menu
    Copy the full SHA
    796aa80 View commit details
    Browse the repository at this point in the history

Commits on Mar 21, 2024

  1. Clean up by creating a function for moving alloc op 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 21, 2024
    Configuration menu
    Copy the full SHA
    84f9f0d View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Mar 21, 2024
    Configuration menu
    Copy the full SHA
    f24dac6 View commit details
    Browse the repository at this point in the history

Commits on Mar 26, 2024

  1. Update KrnlToAffine pass to support scf::ifOp in body of KrnlIterateOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    e6591f0 View commit details
    Browse the repository at this point in the history

  2. Update comments. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    0db91dc View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Mar 26, 2024
    Configuration menu
    Copy the full SHA
    22cbc08 View commit details
    Browse the repository at this point in the history

Commits on Mar 27, 2024

  1. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Mar 27, 2024
    Configuration menu
    Copy the full SHA
    f24c9a6 View commit details
    Browse the repository at this point in the history

Commits on Mar 29, 2024

  1. Dynamic dim support v1. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Mar 29, 2024
    Configuration menu
    Copy the full SHA
    c4f0a3b View commit details
    Browse the repository at this point in the history

Commits on Apr 1, 2024

  1. Fix static dim case. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 1, 2024
    Configuration menu
    Copy the full SHA
    f7fc4ca View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 1, 2024
    Configuration menu
    Copy the full SHA
    38161c1 View commit details
    Browse the repository at this point in the history

Commits on Apr 2, 2024

  1. Add KrnlRegionOp in region of scf::ifOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 2, 2024
    Configuration menu
    Copy the full SHA
    c8f8075 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 2, 2024
    Configuration menu
    Copy the full SHA
    0fbfa9d View commit details
    Browse the repository at this point in the history

Commits on Apr 3, 2024

  1. Update dynamic dim support. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 3, 2024
    Configuration menu
    Copy the full SHA
    8d10b0b View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 3, 2024
    Configuration menu
    Copy the full SHA
    ed5fec1 View commit details
    Browse the repository at this point in the history

Commits on Apr 4, 2024

  1. Update condition to apply unstickStickRemoval optimization 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 4, 2024
    Configuration menu
    Copy the full SHA
    9077e7d View commit details
    Browse the repository at this point in the history

  2. Update comments 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 4, 2024
    Configuration menu
    Copy the full SHA
    91b6f46 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 4, 2024
    Configuration menu
    Copy the full SHA
    0e06aec View commit details
    Browse the repository at this point in the history

Commits on Apr 5, 2024

  1. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 5, 2024
    Configuration menu
    Copy the full SHA
    c73d8ad View commit details
    Browse the repository at this point in the history

Commits on Apr 12, 2024

  1. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 12, 2024
    Configuration menu
    Copy the full SHA
    d934273 View commit details
    Browse the repository at this point in the history

  2. Added lit test for shape inference. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 12, 2024
    Configuration menu
    Copy the full SHA
    9a45f0f View commit details
    Browse the repository at this point in the history

Commits on Apr 13, 2024

  1. Fix lit test. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 13, 2024
    Configuration menu
    Copy the full SHA
    dad1f99 View commit details
    Browse the repository at this point in the history

Commits on Apr 17, 2024

  1. Update dynamic dim support 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 17, 2024
    Configuration menu
    Copy the full SHA
    6d111a7 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 17, 2024
    Configuration menu
    Copy the full SHA
    44396c4 View commit details
    Browse the repository at this point in the history

Commits on Apr 22, 2024

  1. Fix insertion of KrnlRegionOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 22, 2024
    Configuration menu
    Copy the full SHA
    37e12b2 View commit details
    Browse the repository at this point in the history

Commits on Apr 25, 2024

  1. Add shapeHelpr for ONNXParallelOp and ONNXForkOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    c5e7663 View commit details
    Browse the repository at this point in the history

  2. Add dimAnalysis for ONNXParallelOp and ONNXForkOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    2f31a16 View commit details
    Browse the repository at this point in the history

  3. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Apr 25, 2024
    Configuration menu
    Copy the full SHA
    b70fbcc View commit details
    Browse the repository at this point in the history

Commits on May 1, 2024

  1. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed May 1, 2024
    Configuration menu
    Copy the full SHA
    b6ee509 View commit details
    Browse the repository at this point in the history

Commits on May 27, 2024

  1. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed May 27, 2024
    Configuration menu
    Copy the full SHA
    06cc16c View commit details
    Browse the repository at this point in the history

Commits on May 29, 2024

  1. Fix inserting regionOp in iterateOp. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed May 29, 2024
    Configuration menu
    Copy the full SHA
    2bcabe8 View commit details
    Browse the repository at this point in the history

Commits on Jun 6, 2024

  1. Add scripts for rewriting a model for operator-level parallelization. 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
Co-authored-by: Yasushi Negishi <negishi@jp.ibm.com>
    @imaihal @negiyas
    imaihal and negiyas committed Jun 6, 2024
    Configuration menu
    Copy the full SHA
    c837809 View commit details
    Browse the repository at this point in the history

  2. Merge branch 'main' into parallelop_forkop_with_omp

    @imaihal
    imaihal committed Jun 6, 2024
    Configuration menu
    Copy the full SHA
    53805d9 View commit details
    Browse the repository at this point in the history

  3. black format 

    Signed-off-by: Haruki Imai <imaihal@jp.ibm.com>
    @imaihal
    imaihal committed Jun 6, 2024
    Configuration menu
    Copy the full SHA
    17ac56d View commit details
    Browse the repository at this point in the history