github/ROCm
1
1
Fork 0
mirror of https://github.com/ROCm/ROCm.git synced 2026-04-05 03:01:17 -04:00
Code Issues Packages Projects Releases Wiki Activity

[BACKEND] Add error reporting to report non-kernel-argument (#2552)

Browse Source 
Co-authored-by: dongdongl <dongdongl@nvidia.com>
This commit is contained in:
Dongdong Li
2023-11-02 08:22:10 +08:00
committed by GitHub
parent 702cde0d6f
commit d0098da7b1
2 changed files with 13 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
lib/Conversion/TritonGPUToLLVM/LoadStoreOpToLLVM.cpp
View File

@@ -877,6 +877,11 @@ private:
return -1 -
op.getValue().dyn_cast<IntegerAttr>().getValue().getZExtValue();
}
if (!isa<BlockArgument>(v) &&
!isa<mlir::UnrealizedConversionCastOp, arith::ExtSIOp>(
v.getDefiningOp()))
llvm::report_fatal_error(
"Operand of `MakeTensorPtrOp` is not the function's argument");
if (v.getDefiningOp() &&
isa<mlir::UnrealizedConversionCastOp>(v.getDefiningOp())) {
return getArgIdx(v.getDefiningOp()->getOperand(0));
@@ -1771,6 +1776,11 @@ private:
return -1 -
op.getValue().dyn_cast<IntegerAttr>().getValue().getZExtValue();
}
if (!isa<BlockArgument>(v) &&
!isa<mlir::UnrealizedConversionCastOp, arith::ExtSIOp>(
v.getDefiningOp()))
llvm::report_fatal_error(
"Operand of `MakeTensorPtrOp` is not the function's argument");
if (v.getDefiningOp() &&
isa<mlir::UnrealizedConversionCastOp>(v.getDefiningOp())) {
return getArgIdx(v.getDefiningOp()->getOperand(0));