[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