Differential D95331 Diff 345358 mlir/lib/Conversion/GPUToNVVM/WmmaOpsToNvvmLowering.h

Changeset View

Standalone View

mlir/lib/Conversion/GPUToNVVM/WmmaOpsToNvvmLowering.h

This file was added.

				//===---- WmmaOpsToNVVMLowering.h - WMMA ops to NVVM lowering -- C++-----===//
				//
				// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
				// See https://llvm.org/LICENSE.txt for license information.
				// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
				//
				//===----------------------------------------------------------------------===//
				//
				// This file contains declarations of patterns to lower GPU Subgroup MMA ops to
				// NVVM Dialect.
				//
				//===----------------------------------------------------------------------===//

				#ifndef MLIR_LIB_CONVERSION_GPUTONVVM_WMMAOPSTONVVMLOWERING_H
				Lint: Pre-merge checks Inline Actions clang-tidy: warning: header guard does not follow preferred style [llvm-header-guard] not useful Lint: Pre-merge checks: clang-tidy: warning: header guard does not follow preferred style [llvm-header-guard] [[https…
				#define MLIR_LIB_CONVERSION_GPUTONVVM_WMMAOPSTONVVMLOWERING_H

				#include "CommonTypes.h"
				#include "mlir/Conversion/StandardToLLVM/ConvertStandardToLLVM.h"
				#include "mlir/Dialect/GPU/GPUDialect.h"

				namespace mlir {

				/// This class implements the conversion of GPU MMA loadOp to wmma.load op
				/// in the NVVM dialect. The conversion not only emits the NVVM op but also
				/// emits code that is necessary to store the data in the destination memref
				/// after it has been loaded.
				struct WmmaLoadOpToNVVMLowering
				: public ConvertOpToLLVMPattern<gpu::SubgroupMmaLoadMatrixOp>,
				private CommonLLVMAndBuiltInMLIRTypes {
				public:
				explicit WmmaLoadOpToNVVMLowering(LLVMTypeConverter &typeConverter);

				LogicalResult
				matchAndRewrite(gpu::SubgroupMmaLoadMatrixOp subgroupMmaLoadMatrixOp,
				ArrayRef<Value> operands,
				ConversionPatternRewriter &rewriter) const override;
				};

				/// This class implements the conversion of GPU MMA storeOp to wmma.store op
				/// in the NVVM dialect. The conversion not only emits the NVVM op but also
				/// emits code that is necessary to unpack the data in the source and convert
				/// the data in the format that is needed by the NVVM op.
				struct WmmaStoreOpToNVVMLowering
				: public ConvertOpToLLVMPattern<gpu::SubgroupMmaStoreMatrixOp>,
				private CommonLLVMAndBuiltInMLIRTypes {
				public:
				explicit WmmaStoreOpToNVVMLowering(LLVMTypeConverter &typeConverter);

				LogicalResult
				matchAndRewrite(gpu::SubgroupMmaStoreMatrixOp subgroupMmaStoreMatrixOp,
				ArrayRef<Value> operands,
				ConversionPatternRewriter &rewriter) const override;
				};

				/// This class implements the conversion of GPU MMA computeOp to wmma.mma op
				/// in the NVVM dialect.
				struct WmmaMmaOpToNVVMLowering
				: public ConvertOpToLLVMPattern<gpu::SubgroupMmaComputeOp>,
				private CommonLLVMAndBuiltInMLIRTypes {
				explicit WmmaMmaOpToNVVMLowering(LLVMTypeConverter &typeConverter);

				LogicalResult
				matchAndRewrite(gpu::SubgroupMmaComputeOp subgroupMmaComputeOp,
				ArrayRef<Value> operands,
				ConversionPatternRewriter &rewriter) const override;
				};

				} // namespace mlir
				#endif