Skip to content

Commit 7eb2b99

Browse files
authored
[mlir] Change the class name of the GenerateWarpgroupDescriptor (#68286)
1 parent c263639 commit 7eb2b99

File tree

3 files changed

+6
-6
lines changed

3 files changed

+6
-6
lines changed

mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -667,7 +667,7 @@ def NVGPU_TmaCreateDescriptorOp : NVGPU_Op<"tma.create.descriptor", []> {
667667
let hasVerifier = 1;
668668
}
669669

670-
def NVGPU_GenerateWarpgroupDescriptorOp : NVGPU_Op<"warpgroup.generate.descriptor", []> {
670+
def NVGPU_WarpgroupGenerateDescriptorOp : NVGPU_Op<"warpgroup.generate.descriptor", []> {
671671
let summary = "Generate a warpgroup matrix descriptor";
672672
let description = [{
673673
This Op builds a `nvgpu.warpgroup.descriptor` that is used by

mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -968,12 +968,12 @@ struct NVGPUTmaAsyncLoadOpLowering
968968
}
969969
};
970970
struct NVGPUGenerateWarpgroupDescriptorLowering
971-
: public ConvertOpToLLVMPattern<nvgpu::GenerateWarpgroupDescriptorOp> {
971+
: public ConvertOpToLLVMPattern<nvgpu::WarpgroupGenerateDescriptorOp> {
972972
using ConvertOpToLLVMPattern<
973-
nvgpu::GenerateWarpgroupDescriptorOp>::ConvertOpToLLVMPattern;
973+
nvgpu::WarpgroupGenerateDescriptorOp>::ConvertOpToLLVMPattern;
974974

975975
LogicalResult
976-
matchAndRewrite(nvgpu::GenerateWarpgroupDescriptorOp op, OpAdaptor adaptor,
976+
matchAndRewrite(nvgpu::WarpgroupGenerateDescriptorOp op, OpAdaptor adaptor,
977977
ConversionPatternRewriter &rewriter) const override {
978978

979979
ImplicitLocOpBuilder b(op->getLoc(), rewriter);

mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -367,10 +367,10 @@ LogicalResult TmaCreateDescriptorOp::verify() {
367367
}
368368

369369
//===----------------------------------------------------------------------===//
370-
// NVGPU_GenerateWarpgroupDescriptorOp
370+
// NVGPU_WarpgroupGenerateDescriptorOp
371371
//===----------------------------------------------------------------------===//
372372

373-
LogicalResult GenerateWarpgroupDescriptorOp::verify() {
373+
LogicalResult WarpgroupGenerateDescriptorOp::verify() {
374374
MemRefType memrefType = getTensor().getType();
375375
MemRefType tensorMapType = getTensorMap().getType().getTensor();
376376

0 commit comments

Comments
 (0)