Skip to content

Commit 5da85b1

Browse files
authored
[BACKEND] Throw a more intuitive error message when a given mma type is not supported (#5367)
1 parent a612088 commit 5da85b1

File tree

1 file changed

+3
-0
lines changed
  • third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/DotOpToLLVM

1 file changed

+3
-0
lines changed

third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/DotOpToLLVM/MMAv2.cpp

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -516,6 +516,9 @@ LogicalResult convertDot(const LLVMTypeConverter *typeConverter,
516516

517517
const auto &mmaInstructions =
518518
isTuring ? mmaInstrPtxTuring : mmaInstrPtxAmpere;
519+
if (mmaInstructions.find(mmaType) == mmaInstructions.end()) {
520+
return emitError(loc, "Unsupported MMA instruction for the given mma type");
521+
}
519522
auto rank = dTensorTy.getRank();
520523
auto elemsPerThread = triton::gpu::getElemsPerThread(dTensorTy);
521524
auto batchOffset =

0 commit comments

Comments
 (0)