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

[compiler] strip execution context affinities in const eval #18663

Open
wants to merge 4 commits into
base: main
Choose a base branch
from
Open

[compiler] strip execution context affinities in const eval #18663

wants to merge 4 commits into from
+98 −0

Commits on Oct 2, 2024

  1. [compiler] strip execution context affinities in const eval 

    During compile-time constant evaluation in pass
iree-consteval-jit-globals it does not make sense to assign
device/queue affinities. We will be compiling and executing it on the
compilation host.

The JITed IR is first stripped of all attributes that assign affinities
and all flow.tensor.transfer ops.

Signed-off-by: Boian Petkantchin <[email protected]>
    @sogartar
    sogartar committed Oct 2, 2024
    Configuration menu
    Copy the full SHA
    865a8bd View commit details
    Browse the repository at this point in the history

  2. Add missing dependency 

    Signed-off-by: Boian Petkantchin <[email protected]>
    @sogartar
    sogartar committed Oct 2, 2024
    Configuration menu
    Copy the full SHA
    5a294e4 View commit details
    Browse the repository at this point in the history

  3. Move call to stripExecutionContextAffinities in ProgramBuilder::impor… 

    …tInitializer

Signed-off-by: Boian Petkantchin <[email protected]>
    @sogartar
    sogartar committed Oct 2, 2024
    Configuration menu
    Copy the full SHA
    5b0910f View commit details
    Browse the repository at this point in the history

Commits on Oct 3, 2024

  1. Remove comment about required affinites and rename stuff

    @sogartar
    sogartar committed Oct 3, 2024
    Configuration menu
    Copy the full SHA
    f032251 View commit details
    Browse the repository at this point in the history