#include "AArch64.h"
#include "AArch64ExpandImm.h"
#include "MCTargetDesc/AArch64AddressingModes.h"

Include dependency graph for AArch64ExpandImm.cpp:

Functions
static uint64_t	getChunk (uint64_t Imm, unsigned ChunkIdx)
	Helper function which extracts the specified 16-bit chunk from a 64-bit value. More...

static bool	canUseOrr (uint64_t Chunk, uint64_t &Encoding)
	Check whether the given 16-bit chunk replicated to full 64-bit width can be materialized with an ORR instruction. More...

static bool	tryToreplicateChunks (uint64_t UImm, SmallVectorImpl< ImmInsnModel > &Insn)
	Check for identical 16-bit chunks within the constant and if so materialize them with a single ORR instruction. More...

static bool	isStartChunk (uint64_t Chunk)
	Check whether this chunk matches the pattern '1...0...'. More...

static bool	isEndChunk (uint64_t Chunk)
	Check whether this chunk matches the pattern '0...1...' This pattern ends a contiguous sequence of ones if we look at the bits from the LSB towards the MSB. More...

static uint64_t	updateImm (uint64_t Imm, unsigned Idx, bool Clear)
	Clear or set all bits in the chunk at the given index. More...

static bool	trySequenceOfOnes (uint64_t UImm, SmallVectorImpl< ImmInsnModel > &Insn)
	Check whether the constant contains a sequence of contiguous ones, which might be interrupted by one or two chunks. More...

static void	expandMOVImmSimple (uint64_t Imm, unsigned BitSize, unsigned OneChunks, unsigned ZeroChunks, SmallVectorImpl< ImmInsnModel > &Insn)
	Expand a MOVi32imm or MOVi64imm pseudo instruction to a MOVZ or MOVN of width BitSize followed by up to 3 MOVK instructions. More...

Function Documentation

◆ canUseOrr()

static bool canUseOrr	(	uint64_t	Chunk,
		uint64_t &	Encoding
	)

static

Check whether the given 16-bit chunk replicated to full 64-bit width can be materialized with an ORR instruction.

Definition at line 30 of file AArch64ExpandImm.cpp.

References llvm::AArch64_AM::processLogicalImmediate().

Referenced by tryToreplicateChunks().

◆ expandMOVImmSimple()

static void expandMOVImmSimple	(	uint64_t	Imm,
		unsigned	BitSize,
		unsigned	OneChunks,
		unsigned	ZeroChunks,
		SmallVectorImpl< ImmInsnModel > &	Insn
	)

inlinestatic

Expand a MOVi32imm or MOVi64imm pseudo instruction to a MOVZ or MOVN of width BitSize followed by up to 3 MOVK instructions.

Definition at line 244 of file AArch64ExpandImm.cpp.

References llvm::countLeadingZeros(), llvm::countTrailingZeros(), llvm::AArch64_AM::getShifterImm(), llvm::RISCVMatInt::Imm, llvm::M68kBeads::Imm16, Insn, llvm::AArch64_AM::LSL, llvm::BitmaskEnumDetail::Mask(), and Shift.

Referenced by llvm::AArch64_IMM::expandMOVImm().

◆ getChunk()

static uint64_t getChunk	(	uint64_t	Imm,
		unsigned	ChunkIdx
	)

static

Helper function which extracts the specified 16-bit chunk from a 64-bit value.

Definition at line 22 of file AArch64ExpandImm.cpp.

References assert(), and llvm::RISCVMatInt::Imm.

Referenced by llvm::AArch64_IMM::expandMOVImm(), trySequenceOfOnes(), and tryToreplicateChunks().

◆ isEndChunk()

static bool isEndChunk ( uint64_t Chunk )

static

Check whether this chunk matches the pattern '0...1...' This pattern ends a contiguous sequence of ones if we look at the bits from the LSB towards the MSB.

Definition at line 116 of file AArch64ExpandImm.cpp.

References llvm::isMask_64(), and llvm::max().

Referenced by trySequenceOfOnes().

◆ isStartChunk()

static bool isStartChunk ( uint64_t Chunk )

static

Check whether this chunk matches the pattern '1...0...'.

This pattern starts a contiguous sequence of ones if we look at the bits from the LSB towards the MSB.

Definition at line 106 of file AArch64ExpandImm.cpp.

References llvm::isMask_64(), and llvm::max().

Referenced by trySequenceOfOnes().

◆ trySequenceOfOnes()

static bool trySequenceOfOnes	(	uint64_t	UImm,
		SmallVectorImpl< ImmInsnModel > &	Insn
	)

static

Check whether the constant contains a sequence of contiguous ones, which might be interrupted by one or two chunks.

If so, materialize the sequence of contiguous ones with an ORR instruction. Materialize the chunks which are either interrupting the sequence or outside of the sequence with a MOVK instruction.

Assuming S is a chunk which starts the sequence (1...0...), E is a chunk which ends the sequence (0...1...). Then we are looking for constants which contain at least one S and E chunk. E.g. |E|A|B|S|, |A|E|B|S| or |A|B|E|S|.

We are also looking for constants like |S|A|B|E| where the contiguous sequence of ones wraps around the MSB into the LSB.

Definition at line 150 of file AArch64ExpandImm.cpp.

References assert(), getChunk(), llvm::AArch64_AM::getShifterImm(), Insn, isEndChunk(), isStartChunk(), llvm::AArch64_AM::LSL, llvm::BitmaskEnumDetail::Mask(), llvm::AArch64_AM::processLogicalImmediate(), std::swap(), and updateImm().

Referenced by llvm::AArch64_IMM::expandMOVImm().

◆ tryToreplicateChunks()

static bool tryToreplicateChunks	(	uint64_t	UImm,
		SmallVectorImpl< ImmInsnModel > &	Insn
	)

static

Check for identical 16-bit chunks within the constant and if so materialize them with a single ORR instruction.

The remaining one or two 16-bit chunks will be materialized with MOVK instructions.

This allows us to materialize constants like |A|B|A|A| or |A|B|C|A| (order of the chunks doesn't matter), assuming |A|A|A|A| can be materialized with an ORR instruction.

Definition at line 43 of file AArch64ExpandImm.cpp.

References canUseOrr(), getChunk(), llvm::AArch64_AM::getShifterImm(), llvm::M68kBeads::Imm16, Insn, and llvm::AArch64_AM::LSL.

Referenced by llvm::AArch64_IMM::expandMOVImm().