-
Notifications
You must be signed in to change notification settings - Fork 14.4k
[IR2Vec] Add embeddings mode to llvm-ir2vec tool #147844
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: users/svkeerthy/07-09-ir2vec_tool
Are you sure you want to change the base?
[IR2Vec] Add embeddings mode to llvm-ir2vec tool #147844
Conversation
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Premerge failures here also look relevant.
@@ -0,0 +1,73 @@ | |||
; RUN: llvm-ir2vec --mode=embeddings --ir2vec-vocab-path=%S/../../Analysis/IR2Vec/Inputs/dummy_3D_nonzero_opc_vocab.json %s | FileCheck %s -check-prefix=CHECK-DEFAULT |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Given you access %S../../Analysis/IR2Vec/Inputs
so often, I'm wondering if it makes sense to add a lit substitution. Maybe something like %ir2vec_test_vocab
or something.
7c4d86d
to
5f1f3fe
Compare
bf757c0
to
684d298
Compare
✅ With the latest revision this PR passed the Python code formatter. |
684d298
to
2d88b38
Compare
2d88b38
to
6fd2dca
Compare
Add embedding generation functionality to the llvm-ir2vec tool, complementing the existing triplet generation mode.
This change completes the IR2Vec tool by adding the embedding generation functionality, which was previously mentioned as a TODO item. The tool now supports both triplet generation for vocabulary training and embedding generation using a trained vocabulary.