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

Convert AOT to TECompiler #8697

Merged
merged 3 commits into from
Aug 12, 2021
Merged

Convert AOT to TECompiler #8697

merged 3 commits into from
Aug 12, 2021

Commits on Aug 11, 2021

  1. Convert AOT to TECompiler 

    This removes the dependency on "compile_engine.h" from aot_executor_codegen.cc. This required a few changes to how AOT was operating:
* AOT run_model is now based on the post lowering main_module
* AOTOnDemandAllocator is ran twice to ensure SIDs are updated post-lowering
* Moved to using tec::UpdateFunctionMetadata

Tests are passing, but would appreciate other validation 😸
    @Mousius
    Mousius committed Aug 11, 2021
    Configuration menu
    Copy the full SHA
    3f3750c View commit details
    Browse the repository at this point in the history

  2. Clarify reasoning behind replanning memory later

    @Mousius
    Mousius committed Aug 11, 2021
    Configuration menu
    Copy the full SHA
    c9c965d View commit details
    Browse the repository at this point in the history

  3. Use main_func_info rather than bespoke logic in AOT 

    This moves from using the bespoke AOT UpdateMainWorkspaceSize to the
LoweredModule main_func_info property to unify with Graph executor
codegen.
    @Mousius
    Mousius committed Aug 11, 2021
    Configuration menu
    Copy the full SHA
    b39e8cc View commit details
    Browse the repository at this point in the history