We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a612088 commit 5da85b1Copy full SHA for 5da85b1
third_party/nvidia/lib/TritonNVIDIAGPUToLLVM/DotOpToLLVM/MMAv2.cpp
@@ -516,6 +516,9 @@ LogicalResult convertDot(const LLVMTypeConverter *typeConverter,
516
517
const auto &mmaInstructions =
518
isTuring ? mmaInstrPtxTuring : mmaInstrPtxAmpere;
519
+ if (mmaInstructions.find(mmaType) == mmaInstructions.end()) {
520
+ return emitError(loc, "Unsupported MMA instruction for the given mma type");
521
+ }
522
auto rank = dTensorTy.getRank();
523
auto elemsPerThread = triton::gpu::getElemsPerThread(dTensorTy);
524
auto batchOffset =
0 commit comments