X86LowerAMXType.cpp File Reference

#include "X86.h"
#include "llvm/ADT/PostOrderIterator.h"
#include "llvm/ADT/SetVector.h"
#include "llvm/ADT/SmallSet.h"
#include "llvm/Analysis/OptimizationRemarkEmitter.h"
#include "llvm/Analysis/TargetLibraryInfo.h"
#include "llvm/Analysis/TargetTransformInfo.h"
#include "llvm/CodeGen/Passes.h"
#include "llvm/CodeGen/TargetPassConfig.h"
#include "llvm/CodeGen/ValueTypes.h"
#include "llvm/IR/DataLayout.h"
#include "llvm/IR/Function.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/Instructions.h"
#include "llvm/IR/IntrinsicInst.h"
#include "llvm/IR/IntrinsicsX86.h"
#include "llvm/IR/PatternMatch.h"
#include "llvm/InitializePasses.h"
#include "llvm/Pass.h"
#include "llvm/Target/TargetMachine.h"
#include "llvm/Transforms/Utils/AssumeBundleBuilder.h"
#include "llvm/Transforms/Utils/Local.h"
#include <map>

Include dependency graph for X86LowerAMXType.cpp:

Go to the source code of this file.

Macros
#define	DEBUG_TYPE   "lower-amx-type"

Functions
static bool	isAMXCast (Instruction *II)
static bool	isAMXIntrinsic (Value *I)
static AllocaInst *	createAllocaInstAtEntry (IRBuilder<> &Builder, BasicBlock *BB, Type *Ty)
static Instruction *	getFirstNonAllocaInTheEntryBlock (Function &F)
static std::pair< Value *, Value * >	getShape (IntrinsicInst *II, unsigned OpNo)
static std::pair< Value *, Value * >	getShape (PHINode *Phi)
static Value *	getAllocaPos (BasicBlock *BB)
static Instruction *	createTileStore (Instruction *TileDef, Value *Ptr)
static void	replaceWithTileLoad (Use &U, Value *Ptr, bool IsPHI=false)
static bool	isIncomingOfPHI (Instruction *I)
	INITIALIZE_PASS_BEGIN (X86LowerAMXTypeLegacyPass, DEBUG_TYPE, PassName, false, false) INITIALIZE_PASS_END(X86LowerAMXTypeLegacyPass

Variables
static const char	PassName [] = "Lower AMX type for load/store"
	DEBUG_TYPE
	false

Macro Definition Documentation

DEBUG_TYPE

#define DEBUG_TYPE   "lower-amx-type"

Definition at line 69 of file X86LowerAMXType.cpp.

Function Documentation

createAllocaInstAtEntry()

static AllocaInst* createAllocaInstAtEntry ( IRBuilder<> &  Builder, BasicBlock *  BB, Type *  Ty  )

static

Definition at line 95 of file X86LowerAMXType.cpp.

References BB, Builder, DL, F, llvm::Type::getX86_AMXTy(), M, and llvm::AllocaInst::setAlignment().

createTileStore()

static Instruction* createTileStore ( Instruction *  TileDef, Value *  Ptr  )

static

Definition at line 461 of file X86LowerAMXType.cpp.

References llvm::AMDGPU::HSAMD::Kernel::Key::Args, assert(), BB, Builder, llvm::ilist_node_impl< OptionsT >::getIterator(), llvm::Instruction::getParent(), llvm::Value::getType(), llvm::Type::isX86_AMXTy(), and llvm::None.

getAllocaPos()

static Value* getAllocaPos ( BasicBlock *  BB )

static

Definition at line 445 of file X86LowerAMXType.cpp.

References BB, Builder, DL, F, llvm::VectorType::get(), llvm::ilist_node_impl< OptionsT >::getIterator(), and M.

getFirstNonAllocaInTheEntryBlock()

static Instruction* getFirstNonAllocaInTheEntryBlock ( Function &  F )

static

Definition at line 110 of file X86LowerAMXType.cpp.

References F, I, and llvm_unreachable.

Referenced by getShape().

getShape() [1/2]

static std::pair<Value *, Value *> getShape ( IntrinsicInst *  II, unsigned  OpNo  )

static

Definition at line 117 of file X86LowerAMXType.cpp.

References Builder, llvm::IRBuilderBase::CreateUDiv(), llvm::CallBase::getArgOperand(), getFirstNonAllocaInTheEntryBlock(), llvm::Instruction::getFunction(), llvm::IRBuilderBase::getInt16(), llvm::IntrinsicInst::getIntrinsicID(), llvm::User::getOperand(), and llvm_unreachable.

Referenced by getShape().

getShape() [2/2]

static std::pair<Value *, Value *> getShape ( PHINode *  Phi )

static

Definition at line 183 of file X86LowerAMXType.cpp.

References llvm::Use::getOperandNo(), getShape(), llvm::Use::getUser(), isAMXCast(), isAMXIntrinsic(), llvm::Value::use_begin(), and llvm::Value::use_empty().

INITIALIZE_PASS_BEGIN()

INITIALIZE_PASS_BEGIN	(	X86LowerAMXTypeLegacyPass	,
		DEBUG_TYPE	,
		PassName	,
		false	,
		false
	)

isAMXCast()

static bool isAMXCast ( Instruction *  II )

static

Definition at line 71 of file X86LowerAMXType.cpp.

References llvm::PatternMatch::m_Value(), and llvm::PatternMatch::match().

Referenced by getShape(), and isAMXIntrinsic().

isAMXIntrinsic()

static bool isAMXIntrinsic ( Value *  I )

static

Definition at line 77 of file X86LowerAMXType.cpp.

References I, and isAMXCast().

Referenced by getShape().

isIncomingOfPHI()

static bool isIncomingOfPHI ( Instruction *  I )

static

Definition at line 504 of file X86LowerAMXType.cpp.

References llvm::Use::getUser(), and I.

replaceWithTileLoad()

static void replaceWithTileLoad ( Use &  U, Value *  Ptr, bool  IsPHI = false  )

static

Definition at line 479 of file X86LowerAMXType.cpp.

References llvm::AMDGPU::HSAMD::Kernel::Key::Args, assert(), Builder, llvm::Use::get(), llvm::User::getOperand(), llvm::Value::getType(), llvm::Use::getUser(), llvm::Type::isX86_AMXTy(), llvm::None, and llvm::User::replaceUsesOfWith().

Variable Documentation

DEBUG_TYPE

DEBUG_TYPE

Definition at line 1242 of file X86LowerAMXType.cpp.

false

false

Definition at line 1242 of file X86LowerAMXType.cpp.

PassName

PassName = "Lower AMX type for load/store"

static

Definition at line 1236 of file X86LowerAMXType.cpp.