[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