[llvm] [llvm-ir2vec] adding function-embedding map API to ir2vec python bindings (PR #177166)

Mircea Trofin via llvm-commits llvm-commits at lists.llvm.org
Mon Feb 2 11:54:28 PST 2026


================
@@ -42,12 +42,18 @@ class PyIR2VecTool {
   std::unique_ptr<LLVMContext> Ctx;
   std::unique_ptr<Module> M;
   std::unique_ptr<IR2VecTool> Tool;
+  IR2VecKind EmbKind;
 
 public:
   PyIR2VecTool(const std::string &Filename, const std::string &Mode,
                const std::string &VocabPath) {
-    if (Mode != "sym" && Mode != "fa")
+    EmbKind = [](const std::string &Mode) -> IR2VecKind {
----------------
mtrofin wrote:

nit: s/EmbKind/GetEmbeddingKind

https://github.com/llvm/llvm-project/pull/177166


More information about the llvm-commits mailing list