[llvm] [IR2Vec] Add support for flow-aware embeddings (PR #152613)
Aiden Grossman via llvm-commits
llvm-commits at lists.llvm.org
Tue Aug 26 15:32:19 PDT 2025
================
@@ -552,8 +601,11 @@ PreservedAnalyses IR2VecPrinterPass::run(Module &M,
assert(Vocabulary.isValid() && "IR2Vec Vocabulary is invalid");
for (Function &F : M) {
- std::unique_ptr<Embedder> Emb =
- Embedder::create(IR2VecKind::Symbolic, F, Vocabulary);
+ std::unique_ptr<Embedder> Emb;
+ if (IR2VecEmbeddingKind == IR2VecKind::Symbolic)
----------------
boomanaiden154 wrote:
Do we want a switch here to ensure that this handles all `IR2VecKind` values?
https://github.com/llvm/llvm-project/pull/152613
More information about the llvm-commits
mailing list