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

[Runtime] Parallel-for with threading backend #16133

Merged
merged 1 commit into from
Nov 16, 2023
Merged

[Runtime] Parallel-for with threading backend #16133

merged 1 commit into from
Nov 16, 2023

Commits on Nov 15, 2023

  1. [Runtime] Parallel-for with threading backend 

    This PR introduces the runtime parallel-for helper function
in C++ with the threading backend in TVM.

Right now the existing [parallel-for](https://github.com/apache/tvm/blob/bd67d2e5ebde1aec18bcfa74c087516579bda1ae/include/tvm/support/parallel_for.h#L48-L68)
in TVM is not thread persistent,
in which case we cannot get persistent TLS for each thread.

The introduced parallel-for-with-threading-backend function
leverages the threading backend in TVM and persists threads.
    @MasterJH5574
    MasterJH5574 committed Nov 15, 2023
    Configuration menu
    Copy the full SHA
    05d2240 View commit details
    Browse the repository at this point in the history