Class for storing and accessing the IR2Vec vocabulary. More...

#include "llvm/Analysis/IR2Vec.h"

Public Types
enum class	CanonicalTypeID : unsigned { FloatTy , VoidTy , LabelTy , MetadataTy , VectorTy , TokenTy , IntegerTy , FunctionTy , PointerTy , StructTy , ArrayTy , UnknownTy , MaxCanonicalType }
	Canonical type IDs supported by IR2Vec Vocabulary. More...
enum class	OperandKind : unsigned { FunctionID , PointerID , ConstantID , VariableID , MaxOperandKind }
	Operand kinds supported by IR2Vec Vocabulary. More...
using	const_iterator = VocabStorage::const_iterator
	Const Iterator type aliases.

Public Member Functions
	Vocabulary ()=default
LLVM_ABI	Vocabulary (VocabStorage &&Storage)
	Vocabulary (const Vocabulary &)=delete
Vocabulary &	operator= (const Vocabulary &)=delete
	Vocabulary (Vocabulary &&)=default
Vocabulary &	operator= (Vocabulary &&Other)=delete
LLVM_ABI bool	isValid () const
LLVM_ABI unsigned	getDimension () const
LLVM_ABI const ir2vec::Embedding &	operator[] (unsigned Opcode) const
	Accessors to get the embedding for a given entity.
LLVM_ABI const ir2vec::Embedding &	operator[] (Type::TypeID TypeID) const
LLVM_ABI const ir2vec::Embedding &	operator[] (const Value &Arg) const
LLVM_ABI const ir2vec::Embedding &	operator[] (CmpInst::Predicate P) const
const_iterator	begin () const
const_iterator	cbegin () const
const_iterator	end () const
const_iterator	cend () const
LLVM_ABI bool	invalidate (Module &M, const PreservedAnalyses &PA, ModuleAnalysisManager::Invalidator &Inv) const

Static Public Member Functions
static LLVM_ABI Expected< Vocabulary >	fromFile (StringRef VocabFilePath, float OpcWeight=1.0, float TypeWeight=0.5, float ArgWeight=0.2)
	Create a Vocabulary by loading embeddings from a JSON file.
static constexpr size_t	getCanonicalSize ()
	Total number of entries (opcodes + canonicalized types + operand kinds + predicates)
static LLVM_ABI StringRef	getVocabKeyForOpcode (unsigned Opcode)
	Function to get vocabulary key for a given Opcode.
static LLVM_ABI StringRef	getVocabKeyForTypeID (Type::TypeID TypeID)
	Function to get vocabulary key for a given TypeID.
static LLVM_ABI StringRef	getVocabKeyForOperandKind (OperandKind Kind)
	Function to get vocabulary key for a given OperandKind.
static LLVM_ABI OperandKind	getOperandKind (const Value *Op)
	Function to classify an operand into OperandKind.
static LLVM_ABI StringRef	getVocabKeyForPredicate (CmpInst::Predicate P)
	Function to get vocabulary key for a given predicate.
static LLVM_ABI unsigned	getIndex (unsigned Opcode)
	Functions to return flat index.
static LLVM_ABI unsigned	getIndex (Type::TypeID TypeID)
static LLVM_ABI unsigned	getIndex (const Value &Op)
static LLVM_ABI unsigned	getIndex (CmpInst::Predicate P)
static LLVM_ABI StringRef	getStringKey (unsigned Pos)
	Returns the string key for a given index position in the vocabulary.
static LLVM_ABI VocabStorage	createDummyVocabForTest (unsigned Dim=1)
	Create a dummy vocabulary for testing purposes.

Static Public Attributes
static constexpr unsigned	MaxTypeIDs = Type::TypeID::TargetExtTyID + 1
static constexpr unsigned	MaxCanonicalTypeIDs
static constexpr unsigned	MaxOperandKinds
static constexpr unsigned	MaxPredicateKinds

Friends
class	llvm::IR2VecVocabAnalysis

Detailed Description

Class for storing and accessing the IR2Vec vocabulary.

The Vocabulary class manages seed embeddings for LLVM IR entities. The seed embeddings are the initial learned representations of the entities of LLVM IR. The IR2Vec representation for a given IR is derived from these seed embeddings.

The vocabulary contains the seed embeddings for three types of entities: instruction opcodes, types, and operands. Types are grouped/canonicalized for better learning (e.g., all float variants map to FloatTy). The vocabulary abstracts away the canonicalization effectively, the exposed APIs handle all the known LLVM IR opcodes, types and operands.

This class helps populate the seed embeddings in an internal vector-based ADT. It provides logic to map every IR entity to a specific slot index or position in this vector, enabling O(1) embedding lookup while avoiding unnecessary computations involving string based lookups while generating the embeddings.

Definition at line 248 of file IR2Vec.h.

Member Typedef Documentation

◆ const_iterator

using llvm::ir2vec::Vocabulary::const_iterator = VocabStorage::const_iterator

Const Iterator type aliases.

Definition at line 425 of file IR2Vec.h.

Member Enumeration Documentation

◆ CanonicalTypeID

enum class llvm::ir2vec::Vocabulary::CanonicalTypeID : unsigned

strong

Canonical type IDs supported by IR2Vec Vocabulary.

Enumerator
FloatTy
VoidTy
LabelTy
MetadataTy
VectorTy
TokenTy
IntegerTy
FunctionTy
PointerTy
StructTy
ArrayTy
UnknownTy
MaxCanonicalType

Definition at line 287 of file IR2Vec.h.

◆ OperandKind

enum class llvm::ir2vec::Vocabulary::OperandKind : unsigned

strong

Operand kinds supported by IR2Vec Vocabulary.

Enumerator
FunctionID
PointerID
ConstantID
VariableID
MaxOperandKind

Definition at line 304 of file IR2Vec.h.

Constructor & Destructor Documentation

◆ Vocabulary() [1/4]

llvm::ir2vec::Vocabulary::Vocabulary ( )

default

Referenced by fromFile(), operator=(), operator=(), Vocabulary(), and Vocabulary().

◆ Vocabulary() [2/4]

LLVM_ABI llvm::ir2vec::Vocabulary::Vocabulary ( VocabStorage && Storage )

inline

Definition at line 328 of file IR2Vec.h.

References LLVM_ABI, and llvm::move().

◆ Vocabulary() [3/4]

llvm::ir2vec::Vocabulary::Vocabulary ( const Vocabulary & )

delete

References Vocabulary().

◆ Vocabulary() [4/4]

llvm::ir2vec::Vocabulary::Vocabulary ( Vocabulary && )

default

References Vocabulary().

Member Function Documentation

◆ begin()

const_iterator llvm::ir2vec::Vocabulary::begin ( ) const

inline

Definition at line 427 of file IR2Vec.h.

References assert(), and isValid().

Referenced by cbegin().

◆ cbegin()

const_iterator llvm::ir2vec::Vocabulary::cbegin ( ) const

inline

Definition at line 432 of file IR2Vec.h.

References begin().

◆ cend()

const_iterator llvm::ir2vec::Vocabulary::cend ( ) const

inline

Definition at line 439 of file IR2Vec.h.

References end().

◆ createDummyVocabForTest()

VocabStorage Vocabulary::createDummyVocabForTest ( unsigned Dim = 1 )

static

Create a dummy vocabulary for testing purposes.

Definition at line 432 of file IR2Vec.cpp.

References I, MaxCanonicalTypeIDs, MaxOperandKinds, MaxPredicateKinds, and llvm::ir2vec::VocabStorage::VocabStorage().

◆ end()

const_iterator llvm::ir2vec::Vocabulary::end ( ) const

inline

Definition at line 434 of file IR2Vec.h.

References assert(), and isValid().

Referenced by cend().

◆ fromFile()

Expected< Vocabulary > Vocabulary::fromFile	(	StringRef	VocabFilePath,
		float	OpcWeight = 1.0,
		float	TypeWeight = 0.5,
		float	ArgWeight = 0.2 )

static

Create a Vocabulary by loading embeddings from a JSON file.

This is the primary entry point for programmatic vocabulary creation, suitable for use in Python bindings or other contexts where command-line options are not available. Weights are applied to scale the embeddings for opcodes, types, and arguments respectively.

Definition at line 609 of file IR2Vec.cpp.

References llvm::ir2vec::ArgWeight, llvm::ir2vec::OpcWeight, llvm::ir2vec::TypeWeight, and Vocabulary().

Referenced by llvm::IR2VecVocabAnalysis::run().

◆ getCanonicalSize()

constexpr size_t llvm::ir2vec::Vocabulary::getCanonicalSize ( )

inlinestaticconstexpr

Total number of entries (opcodes + canonicalized types + operand kinds + predicates)

Definition at line 357 of file IR2Vec.h.

◆ getDimension()

LLVM_ABI unsigned llvm::ir2vec::Vocabulary::getDimension ( ) const

inline

Definition at line 350 of file IR2Vec.h.

References assert(), isValid(), and LLVM_ABI.

Referenced by llvm::FunctionPropertiesInfo::getFunctionPropertiesInfo().

◆ getIndex() [1/4]

LLVM_ABI unsigned llvm::ir2vec::Vocabulary::getIndex ( CmpInst::Predicate P )

inlinestatic

Definition at line 397 of file IR2Vec.h.

References LLVM_ABI, and P.

◆ getIndex() [2/4]

LLVM_ABI unsigned llvm::ir2vec::Vocabulary::getIndex ( const Value & Op )

inlinestatic

Definition at line 391 of file IR2Vec.h.

References assert(), getOperandKind(), LLVM_ABI, and MaxOperandKinds.

◆ getIndex() [3/4]

LLVM_ABI unsigned llvm::ir2vec::Vocabulary::getIndex ( Type::TypeID TypeID )

inlinestatic

Definition at line 386 of file IR2Vec.h.

References assert(), LLVM_ABI, and MaxTypeIDs.

◆ getIndex() [4/4]

LLVM_ABI unsigned llvm::ir2vec::Vocabulary::getIndex ( unsigned Opcode )

inlinestatic

Functions to return flat index.

Definition at line 381 of file IR2Vec.h.

References assert(), and LLVM_ABI.

◆ getOperandKind()

Vocabulary::OperandKind Vocabulary::getOperandKind ( const Value * Op )

static

Function to classify an operand into OperandKind.

Definition at line 369 of file IR2Vec.cpp.

References ConstantID, FunctionID, llvm::isa(), PointerID, and VariableID.

Referenced by getIndex(), and operator[]().

◆ getStringKey()

StringRef Vocabulary::getStringKey ( unsigned Pos )

static

Returns the string key for a given index position in the vocabulary.

This is useful for debugging or printing the vocabulary. Do not use this for embedding generation as string based lookups are inefficient.

Definition at line 408 of file IR2Vec.cpp.

References assert(), getVocabKeyForOpcode(), getVocabKeyForOperandKind(), and getVocabKeyForPredicate().

◆ getVocabKeyForOpcode()

StringRef Vocabulary::getVocabKeyForOpcode ( unsigned Opcode )

static

Function to get vocabulary key for a given Opcode.

Definition at line 357 of file IR2Vec.cpp.

References assert().

Referenced by getStringKey().

◆ getVocabKeyForOperandKind()

LLVM_ABI StringRef llvm::ir2vec::Vocabulary::getVocabKeyForOperandKind ( OperandKind Kind )

inlinestatic

Function to get vocabulary key for a given OperandKind.

Definition at line 368 of file IR2Vec.h.

References assert(), LLVM_ABI, and MaxOperandKinds.

Referenced by getStringKey().

◆ getVocabKeyForPredicate()

StringRef Vocabulary::getVocabKeyForPredicate ( CmpInst::Predicate P )

static

Function to get vocabulary key for a given predicate.

Definition at line 398 of file IR2Vec.cpp.

References llvm::CmpInst::FIRST_ICMP_PREDICATE, and llvm::CmpInst::getPredicateName().

Referenced by getStringKey().

◆ getVocabKeyForTypeID()

LLVM_ABI StringRef llvm::ir2vec::Vocabulary::getVocabKeyForTypeID ( Type::TypeID TypeID )

inlinestatic

Function to get vocabulary key for a given TypeID.

Definition at line 363 of file IR2Vec.h.

References LLVM_ABI.

◆ invalidate()

bool Vocabulary::invalidate	(	Module &	M,
		const PreservedAnalyses &	PA,
		ModuleAnalysisManager::Invalidator &	Inv ) const

Definition at line 426 of file IR2Vec.cpp.

References llvm::PreservedAnalyses::getChecker(), and llvm::IR2VecVocabAnalysis.

◆ isValid()

LLVM_ABI bool llvm::ir2vec::Vocabulary::isValid ( ) const

inline

Definition at line 346 of file IR2Vec.h.

References LLVM_ABI.

Referenced by begin(), end(), getDimension(), llvm::FunctionPropertiesInfo::getFunctionPropertiesInfo(), and llvm::IR2VecPrinterPass::run().

◆ operator=() [1/2]

Vocabulary & llvm::ir2vec::Vocabulary::operator= ( const Vocabulary & )

delete

References Vocabulary().

◆ operator=() [2/2]

Vocabulary & llvm::ir2vec::Vocabulary::operator= ( Vocabulary && Other )

delete

References llvm::ir2vec::ArgWeight, LLVM_ABI, llvm::ir2vec::OpcWeight, llvm::Other, llvm::ir2vec::TypeWeight, and Vocabulary().

◆ operator[]() [1/4]

LLVM_ABI const ir2vec::Embedding & llvm::ir2vec::Vocabulary::operator[] ( CmpInst::Predicate P ) const

inline

Definition at line 419 of file IR2Vec.h.

References LLVM_ABI, and P.

◆ operator[]() [2/4]

LLVM_ABI const ir2vec::Embedding & llvm::ir2vec::Vocabulary::operator[] ( const Value & Arg ) const

inline

Definition at line 413 of file IR2Vec.h.

References assert(), getOperandKind(), LLVM_ABI, and MaxOperandKinds.

◆ operator[]() [3/4]

LLVM_ABI const ir2vec::Embedding & llvm::ir2vec::Vocabulary::operator[] ( Type::TypeID TypeID ) const

inline

Definition at line 407 of file IR2Vec.h.

References assert(), LLVM_ABI, and MaxTypeIDs.

◆ operator[]() [4/4]

LLVM_ABI const ir2vec::Embedding & llvm::ir2vec::Vocabulary::operator[] ( unsigned Opcode ) const

inline

Accessors to get the embedding for a given entity.

Definition at line 402 of file IR2Vec.h.

References assert(), and LLVM_ABI.

◆ llvm::IR2VecVocabAnalysis

friend class llvm::IR2VecVocabAnalysis

friend

Definition at line 249 of file IR2Vec.h.

Referenced by invalidate().

Member Data Documentation

◆ MaxCanonicalTypeIDs

unsigned llvm::ir2vec::Vocabulary::MaxCanonicalTypeIDs

staticconstexpr

Initial value:

=

static_cast<unsigned>(CanonicalTypeID::MaxCanonicalType)

llvm::ir2vec::Vocabulary::CanonicalTypeID::MaxCanonicalType

@ MaxCanonicalType

Definition IR2Vec.h:300

Definition at line 318 of file IR2Vec.h.

Referenced by createDummyVocabForTest().

◆ MaxOperandKinds

unsigned llvm::ir2vec::Vocabulary::MaxOperandKinds

staticconstexpr

Initial value:

=

static_cast<unsigned>(OperandKind::MaxOperandKind)

llvm::ir2vec::Vocabulary::OperandKind::MaxOperandKind

@ MaxOperandKind

Definition IR2Vec.h:309

Definition at line 320 of file IR2Vec.h.

Referenced by createDummyVocabForTest(), getIndex(), getVocabKeyForOperandKind(), and operator[]().

◆ MaxPredicateKinds

unsigned llvm::ir2vec::Vocabulary::MaxPredicateKinds

staticconstexpr

Initial value:

=

NumICmpPredicates + NumFCmpPredicates

Definition at line 324 of file IR2Vec.h.

Referenced by createDummyVocabForTest().

◆ MaxTypeIDs

unsigned llvm::ir2vec::Vocabulary::MaxTypeIDs = Type::TypeID::TargetExtTyID + 1

staticconstexpr

Definition at line 317 of file IR2Vec.h.

Referenced by getIndex(), and operator[]().

The documentation for this class was generated from the following files:

include/llvm/Analysis/IR2Vec.h
lib/Analysis/IR2Vec.cpp

Public Types

Public Member Functions

Static Public Member Functions

Static Public Attributes

Friends

Detailed Description

Member Typedef Documentation

◆ const_iterator

Member Enumeration Documentation

◆ CanonicalTypeID

◆ OperandKind

Constructor & Destructor Documentation

◆ Vocabulary() [1/4]

◆ Vocabulary() [2/4]

◆ Vocabulary() [3/4]

◆ Vocabulary() [4/4]

Member Function Documentation

◆ begin()

◆ cbegin()

◆ cend()

◆ createDummyVocabForTest()

◆ end()

◆ fromFile()

◆ getCanonicalSize()

◆ getDimension()

◆ getIndex() [1/4]

◆ getIndex() [2/4]

◆ getIndex() [3/4]

◆ getIndex() [4/4]

◆ getOperandKind()

◆ getStringKey()

◆ getVocabKeyForOpcode()

◆ getVocabKeyForOperandKind()

◆ getVocabKeyForPredicate()

◆ getVocabKeyForTypeID()

◆ invalidate()

◆ isValid()

◆ operator=() [1/2]

◆ operator=() [2/2]

◆ operator[]() [1/4]

◆ operator[]() [2/4]

◆ operator[]() [3/4]

◆ operator[]() [4/4]

Friends And Related Symbol Documentation

◆ llvm::IR2VecVocabAnalysis

Member Data Documentation

◆ MaxCanonicalTypeIDs

◆ MaxOperandKinds

◆ MaxPredicateKinds

◆ MaxTypeIDs