From 913adff0be007e7282ef0983bd59e210f1c3860d Mon Sep 17 00:00:00 2001 From: Guray Ozen Date: Thu, 5 Oct 2023 09:14:02 +0200 Subject: [PATCH] [mlir] Change the name class name of the `GenerateWarpgroupDescriptorOp` (nfc) --- mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td | 2 +- mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp | 6 +++--- mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp | 4 ++-- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td b/mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td index 337174d6367157a..7093eeea689486b 100644 --- a/mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td +++ b/mlir/include/mlir/Dialect/NVGPU/IR/NVGPU.td @@ -667,7 +667,7 @@ def NVGPU_TmaCreateDescriptorOp : NVGPU_Op<"tma.create.descriptor", []> { let hasVerifier = 1; } -def NVGPU_GenerateWarpgroupDescriptorOp : NVGPU_Op<"warpgroup.generate.descriptor", []> { +def NVGPU_WarpgroupGenerateDescriptorOp : NVGPU_Op<"warpgroup.generate.descriptor", []> { let summary = "Generate a warpgroup matrix descriptor"; let description = [{ This Op builds a `nvgpu.warpgroup.descriptor` that is used by diff --git a/mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp b/mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp index a35c3d9c069e709..b0df2feae16b49f 100644 --- a/mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp +++ b/mlir/lib/Conversion/NVGPUToNVVM/NVGPUToNVVM.cpp @@ -968,12 +968,12 @@ struct NVGPUTmaAsyncLoadOpLowering } }; struct NVGPUGenerateWarpgroupDescriptorLowering - : public ConvertOpToLLVMPattern { + : public ConvertOpToLLVMPattern { using ConvertOpToLLVMPattern< - nvgpu::GenerateWarpgroupDescriptorOp>::ConvertOpToLLVMPattern; + nvgpu::WarpgroupGenerateDescriptorOp>::ConvertOpToLLVMPattern; LogicalResult - matchAndRewrite(nvgpu::GenerateWarpgroupDescriptorOp op, OpAdaptor adaptor, + matchAndRewrite(nvgpu::WarpgroupGenerateDescriptorOp op, OpAdaptor adaptor, ConversionPatternRewriter &rewriter) const override { ImplicitLocOpBuilder b(op->getLoc(), rewriter); diff --git a/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp b/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp index eb8fc4b65bc89ad..0ee0f70eebcf603 100644 --- a/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp +++ b/mlir/lib/Dialect/NVGPU/IR/NVGPUDialect.cpp @@ -367,10 +367,10 @@ LogicalResult TmaCreateDescriptorOp::verify() { } //===----------------------------------------------------------------------===// -// NVGPU_GenerateWarpgroupDescriptorOp +// NVGPU_WarpgroupGenerateDescriptorOp //===----------------------------------------------------------------------===// -LogicalResult GenerateWarpgroupDescriptorOp::verify() { +LogicalResult WarpgroupGenerateDescriptorOp::verify() { MemRefType memrefType = getTensor().getType(); MemRefType tensorMapType = getTensorMap().getType().getTensor();