maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[X86][AMX] Verify illegal types or instructions for x86_amx. 

This patch is related to https://reviews.llvm.org/D100032 which define
some illegal types or operations for x86_amx. There are no arguments,
arrays, pointers, vectors or constants of x86_amx.

Reviewed By: pengfei

Differential Revision: https://reviews.llvm.org/D100472
This commit is contained in:
Luo, Yuanke 2021-04-20 15:52:29 +08:00
parent 5e71b9fa93
commit bcdaccfe34
13 changed files with 90 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
llvm/lib/IR/Type.cpp
View File

@ -95,15 +95,8 @@ bool Type::canLosslesslyBitCastTo(Type *Ty) const {
// else is not lossless. Conservatively assume we can't losslessly convert
// between pointers with different address spaces.
if (auto *PTy = dyn_cast<PointerType>(this)) {
if (auto *OtherPTy = dyn_cast<PointerType>(Ty)) {
// Don't bitcast "load <256 x i32>, <256 x i32>*" to
// "load x86_amx, x86_amx*", because we don't have a corresponding
// instruction to load x86_amx. Doing the transform causes trouble
// to lower "load x86_amx" instruction in backend.
if (OtherPTy->getElementType()->isX86_AMXTy())
return false;
if (auto *OtherPTy = dyn_cast<PointerType>(Ty))
return PTy->getAddressSpace() == OtherPTy->getAddressSpace();
}
return false;
}
return false; // Other types have no identity values
@ -617,7 +610,8 @@ ArrayType *ArrayType::get(Type *ElementType, uint64_t NumElements) {
bool ArrayType::isValidElementType(Type *ElemTy) {
return !ElemTy->isVoidTy() && !ElemTy->isLabelTy() &&
!ElemTy->isMetadataTy() && !ElemTy->isFunctionTy() &&
!ElemTy->isTokenTy() && !isa<ScalableVectorType>(ElemTy);
!ElemTy->isTokenTy() && !ElemTy->isX86_AMXTy() &&
!isa<ScalableVectorType>(ElemTy);
}
//===----------------------------------------------------------------------===//
@ -718,7 +712,8 @@ PointerType *Type::getPointerTo(unsigned addrs) const {
bool PointerType::isValidElementType(Type *ElemTy) {
return !ElemTy->isVoidTy() && !ElemTy->isLabelTy() &&
!ElemTy->isMetadataTy() && !ElemTy->isTokenTy();
!ElemTy->isMetadataTy() && !ElemTy->isTokenTy() &&
!ElemTy->isX86_AMXTy();
}
bool PointerType::isLoadableOrStorableType(Type *ElemTy) {

20
llvm/lib/IR/Verifier.cpp
View File

@ -2493,6 +2493,8 @@ void Verifier::visitFunction(const Function &F) {
"Function takes metadata but isn't an intrinsic", &Arg, &F);
Assert(!Arg.getType()->isTokenTy(),
"Function takes token but isn't an intrinsic", &Arg, &F);
Assert(!Arg.getType()->isX86_AMXTy(),
"Function takes x86_amx but isn't an intrinsic", &Arg, &F);
}
// Check that swifterror argument is only used by loads and stores.
@ -2502,9 +2504,12 @@ void Verifier::visitFunction(const Function &F) {
++i;
}
if (!isLLVMdotName)
if (!isLLVMdotName) {
Assert(!F.getReturnType()->isTokenTy(),
"Functions returns a token but isn't an intrinsic", &F);
"Function returns a token but isn't an intrinsic", &F);
Assert(!F.getReturnType()->isX86_AMXTy(),
"Function returns a x86_amx but isn't an intrinsic", &F);
}
// Get the function metadata attachments.
SmallVector<std::pair<unsigned, MDNode *>, 4> MDs;
@ -3265,9 +3270,12 @@ void Verifier::visitCallBase(CallBase &Call) {
}
// Verify that indirect calls don't return tokens.
if (!Call.getCalledFunction())
if (!Call.getCalledFunction()) {
Assert(!FTy->getReturnType()->isTokenTy(),
"Return type cannot be token for indirect call!");
Assert(!FTy->getReturnType()->isX86_AMXTy(),
"Return type cannot be x86_amx for indirect call!");
}
if (Function *F = Call.getCalledFunction())
if (Intrinsic::ID ID = (Intrinsic::ID)F->getIntrinsicID())
@ -4636,9 +4644,13 @@ void Verifier::visitIntrinsicCall(Intrinsic::ID ID, CallBase &Call) {
// If the intrinsic takes MDNode arguments, verify that they are either global
// or are local to *this* function.
for (Value *V : Call.args())
for (Value *V : Call.args()) {
if (auto *MD = dyn_cast<MetadataAsValue>(V))
visitMetadataAsValue(*MD, Call.getCaller());
if (auto *Const = dyn_cast<Constant>(V))
Assert(!Const->getType()->isX86_AMXTy(),
"const x86_amx is not allowed in argument!");
}
switch (ID) {
default:

9
llvm/lib/Transforms/InstCombine/InstCombineCasts.cpp
View File

@ -2409,10 +2409,11 @@ Instruction *InstCombinerImpl::optimizeBitCastFromPhi(CastInst &CI,
Value *Addr = LI->getOperand(0);
if (Addr == &CI || isa<LoadInst>(Addr))
return nullptr;
// If there is any loss for the pointer bitcast, abandon.
auto *DestPtrTy = DestTy->getPointerTo(LI->getPointerAddressSpace());
auto *SrcPtrTy = Addr->getType();
if (!SrcPtrTy->canLosslesslyBitCastTo(DestPtrTy))
// Don't tranform "load <256 x i32>, <256 x i32>*" to
// "load x86_amx, x86_amx*", because x86_amx* is invalid.
// TODO: Remove this check when bitcast between vector and x86_amx
// is replaced with a specific intrinsic.
if (DestTy->isX86_AMXTy())
return nullptr;
if (LI->hasOneUse() && LI->isSimple())
continue;

2
llvm/test/Verifier/token6.ll
View File

@ -4,4 +4,4 @@ define token @f() {
entry:
ret token undef
}
; CHECK: Functions returns a token but isn't an intrinsic
; CHECK: Function returns a token but isn't an intrinsic

4
llvm/test/Verifier/x86_amx1.ll Normal file
View File

@ -0,0 +1,4 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
@GV = dso_local global [10 x x86_amx] zeroinitializer, align 16
; CHECK: invalid array element type

9
llvm/test/Verifier/x86_amx2.ll Normal file
View File

@ -0,0 +1,9 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
define dso_local void @f() {
entry:
%a.addr = alloca <2 x x86_amx>, align 4
ret void
}
; CHECK: invalid vector element type

8
llvm/test/Verifier/x86_amx3.ll Normal file
View File

@ -0,0 +1,8 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
define void @f(x86_amx %A, x86_amx %B) {
entry:
alloca x86_amx
; CHECK: invalid type for alloca
ret void
}

4
llvm/test/Verifier/x86_amx4.ll Normal file
View File

@ -0,0 +1,4 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
@GV = external global x86_amx
; CHECK: invalid type for global variable

7
llvm/test/Verifier/x86_amx5.ll Normal file
View File

@ -0,0 +1,7 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
define void @f(x86_amx %A) {
entry:
ret void
}
; CHECK: Function takes x86_amx but isn't an intrinsic

7
llvm/test/Verifier/x86_amx6.ll Normal file
View File

@ -0,0 +1,7 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
define x86_amx @f() {
entry:
ret x86_amx undef
}
; CHECK: Function returns a x86_amx but isn't an intrinsic

8
llvm/test/Verifier/x86_amx7.ll Normal file
View File

@ -0,0 +1,8 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
define void @f() {
entry:
call x86_amx () undef ()
ret void
}
; CHECK: Return type cannot be x86_amx for indirect call!

4
llvm/test/Verifier/x86_amx8.ll Normal file
View File

@ -0,0 +1,4 @@
; RUN: not llvm-as %s -o /dev/null 2>&1 | FileCheck %s
@GV = dso_local global x86_amx* null
; CHECK: pointer to this type is invalid

12
llvm/test/Verifier/x86_amx9.ll Normal file
View File

@ -0,0 +1,12 @@
; RUN: not llc %s -o /dev/null 2>&1 | FileCheck %s
@buf = dso_local global [1024 x i8] zeroinitializer, align 16
define dso_local void @test_tile_init(i16 signext %row, i16 signext %col) {
entry:
tail call void @llvm.x86.tilestored64.internal(i16 %row, i16 %col, i8* getelementptr inbounds ([1024 x i8], [1024 x i8]* @buf, i64 0, i64 0), i64 64, x86_amx bitcast (<256 x i32> <i32 1, i32 2, i32 3, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0, i32 0> to x86_amx))
ret void
}
; CHECK: const x86_amx is not allowed in argument!
declare void @llvm.x86.tilestored64.internal(i16, i16, i8*, i64, x86_amx)