llvm-project
c878baf1 - [mlgo][inliner] Fix potential concurrency issue in local ThinLTO + IR2Vec cases (#156120)

Commit
10 days ago
[mlgo][inliner] Fix potential concurrency issue in local ThinLTO + IR2Vec cases (#156120) The inliner's `FeatureMap` used to be immutable, but in IR2Vec cases we don't know the shapes of the embedding vectors until later, so we need to initialize it at the time we construct the advisor. In non-distributed ThinLTO cases, for example, this means we'd mutate shared state. The feature set is also needed when constructing the underlying model runner. The alternative here is to postpone the creation of the model runner to the time we construct the advisor, and also make the feature map a member of the advisor object. (issue identified by @efriedma-quic in PR #154541)
Author
Parents
Loading