Skip to content

[NFC][IR2Vec]Minor comment fixes #141672

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

Merged
merged 1 commit into from
May 27, 2025
Merged
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
19 changes: 6 additions & 13 deletions llvm/include/llvm/Analysis/IR2Vec.h
Original file line number Diff line number Diff line change
Expand Up @@ -111,18 +111,13 @@ class Embedder {
const Vocab &Vocabulary,
unsigned Dimension);

/// Returns a map containing instructions and the corresponding vector
/// representations for a given module corresponding to the IR2Vec
/// algorithm.
/// Returns a map containing instructions and the corresponding embeddings.
const InstEmbeddingsMap &getInstVecMap() const { return InstVecMap; }

/// Returns a map containing basic block and the corresponding vector
/// representations for a given module corresponding to the IR2Vec
/// algorithm.
/// Returns a map containing basic block and the corresponding embeddings.
const BBEmbeddingsMap &getBBVecMap() const { return BBVecMap; }

/// Returns the vector representation for a given function corresponding to
/// the IR2Vec algorithm.
/// Returns the embedding for the current function.
const Embedding &getFunctionVector() const { return FuncVector; }
};

Expand All @@ -131,15 +126,13 @@ class Embedder {
/// representations obtained from the Vocabulary.
class SymbolicEmbedder : public Embedder {
private:
/// Utility function to compute the vector representation for a given basic
/// block.
/// Utility function to compute the embedding for a given basic block.
Embedding computeBB2Vec(const BasicBlock &BB);

/// Utility function to compute the vector representation for a given type.
/// Utility function to compute the embedding for a given type.
Embedding getTypeEmbedding(const Type *Ty) const;

/// Utility function to compute the vector representation for a given
/// operand.
/// Utility function to compute the embedding for a given operand.
Embedding getOperandEmbedding(const Value *Op) const;

public:
Expand Down
Loading