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][TVM] Bring Your Own Codegen to TVM #4258

Closed
wants to merge 34 commits into from
Closed

[WIP][TVM] Bring Your Own Codegen to TVM #4258

wants to merge 34 commits into from

Commits on Nov 28, 2019

  1. graph partitioning

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    487cee0 View commit details
    Browse the repository at this point in the history

  2. extern op coloring infra

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    e0877e8 View commit details
    Browse the repository at this point in the history

  3. eliminate redundant subgraph annotation

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    1b2041d View commit details
    Browse the repository at this point in the history

  4. A failing example

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    f7e299b View commit details
    Browse the repository at this point in the history

  5. Refine partition algorithm 

    Remaining issues:
- External function body does not refer to the generated arguments when
function body is more than one node.

- Runtime is not working due to the fix of external codegen library.

- Not working when there have more than one subgraphs during partition.
    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    60004d1 View commit details
    Browse the repository at this point in the history

  6. Support multiple subgraphs (runtime not work)

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    fe3c486 View commit details
    Browse the repository at this point in the history

  7. Support multiple function body nodes 

    - All nodes except for Call and Tuple are required to propogate
subgraph.
- Multiple outputs are not supported yet.
- Both unit tests are correct in terms of graph partition.
- Runtime is not working yet. Need to implement external add and
multiply.
    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    8734d0a View commit details
    Browse the repository at this point in the history

  8. Add a hack for multiple subgraphes 

    - Each of these subgraphes only contains one primitive op.

- Subgraphes contains multiple nodes are not handled.
    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    590996e View commit details
    Browse the repository at this point in the history

  9. Add rest node visiting to propogate subgraphs.

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    5a6f955 View commit details
    Browse the repository at this point in the history

  10. cblas template

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    2ffc785 View commit details
    Browse the repository at this point in the history

  11. make Cblas working and refactor contrib codegen

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    c8e1499 View commit details
    Browse the repository at this point in the history

  12. small fix for style and check handle_ before closing it

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    7eef6f1 View commit details
    Browse the repository at this point in the history

  13. refactor the interface for different data types

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    001f9c6 View commit details
    Browse the repository at this point in the history

  14. add MKLDNN support and refine interface

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    7febcdb View commit details
    Browse the repository at this point in the history

  15. Simplify runtime invoke 

    Remaining issues:
- For the op that has multiple outputs like batch norm, we don't know
how many it has. This is also a problem when supporting subgraph with
many nodes.
- Have no idea how to deal with the function that has different data
types of inputs.
    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    47517af View commit details
    Browse the repository at this point in the history

  16. to vm: add an InvokeExternal Instruction

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    66779ec View commit details
    Browse the repository at this point in the history

  17. refactor backend interface and remove cblas

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    a7380a1 View commit details
    Browse the repository at this point in the history

  18. To vm: enalbe multiple function compilation

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    d86a5f7 View commit details
    Browse the repository at this point in the history

  19. enable vm test for subgraph with multiple nodes

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    f4c55a5 View commit details
    Browse the repository at this point in the history

  20. fix lint

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    5f8ecaf View commit details
    Browse the repository at this point in the history

  21. remove get lib path API

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    c7c74c3 View commit details
    Browse the repository at this point in the history

  22. initial commit tutorial

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    5d5c907 View commit details
    Browse the repository at this point in the history

  23. add annotation to tutorial

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    0dcc5d0 View commit details
    Browse the repository at this point in the history

  24. Refine tutorial a bit

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    70d1e33 View commit details
    Browse the repository at this point in the history

  25. rebase to upstream

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    125b28f View commit details
    Browse the repository at this point in the history

  26. Improve: 

    - Separate compilation and runtime
- Create separate build pipeline for external and normal functions
- Serialize InvokeExternal Instruction
    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    a298f9c View commit details
    Browse the repository at this point in the history

  27. change macro to function to reduce .so size

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    7891208 View commit details
    Browse the repository at this point in the history

  28. create attr const string for Function

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    b8f1db0 View commit details
    Browse the repository at this point in the history

  29. rebase to upstream

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    2736619 View commit details
    Browse the repository at this point in the history

  30. return csourcemodule from external codegen

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    d950234 View commit details
    Browse the repository at this point in the history

  31. fix test and clean code

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    6fb0605 View commit details
    Browse the repository at this point in the history

  32. fix ci

    @comaniac @zhiics
    comaniac authored and zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    7beb0e3 View commit details
    Browse the repository at this point in the history

  33. more cleanup

    @zhiics
    zhiics committed Nov 28, 2019
    Configuration menu
    Copy the full SHA
    5ff7fa6 View commit details
    Browse the repository at this point in the history

Commits on Dec 1, 2019

  1. fix typo

    @comaniac
    comaniac committed Dec 1, 2019
    Configuration menu
    Copy the full SHA
    cf7ac3c View commit details
    Browse the repository at this point in the history