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 8e1f698 commit 98246b0Copy full SHA for 98246b0
mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp
@@ -1195,7 +1195,7 @@ struct NVGPUWarpgroupMmaOpLowering
1195
auto scaleIn = NVVM::WGMMAScaleInAttr::get(ctx, NVVM::WGMMAScaleIn::one);
1196
auto layoutA = NVVM::MMALayoutAttr::get(ctx, NVVM::MMALayout::row);
1197
auto layoutB = NVVM::MMALayoutAttr::get(ctx, NVVM::MMALayout::col);
1198
- // todo input type
+ // todo: handle other input and output types
1199
auto itype = NVVM::WGMMATypesAttr::get(ctx, NVVM::WGMMATypes::f16);
1200
auto overflow =
1201
NVVM::MMAIntOverflowAttr::get(ctx, NVVM::MMAIntOverflow::wrapped);
0 commit comments