FoldMulAddChainToConvOp.h

Go to the documentation of this file.

//===--FoldMulAddChainToConvOp.h - Fold Mul Add Chain To AIEVec Conv Op --===//
//
// Copyright (C) 2022 Xilinx, Inc.
// Copyright (C) 2022-2024 Advanced Micro Devices, Inc.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
 
#ifndef FOLDMULADDCHAINTOCONVOP_H
#define FOLDMULADDCHAINTOCONVOP_H
 
#include "mlir/Pass/AnalysisManager.h"
#include "mlir/Transforms/DialectConversion.h"
 
namespace xilinx {
 
namespace aievec {
//===----------------------------------------------------------------------===//
// This is the implementation of the folding pass from mul add chain
// to AIEVec convolution operations, compatible with the AIE2 architecture.
//===----------------------------------------------------------------------===//
 
// Configure the legalizations for aievec conv op transformation
void configureAIEVecConvOpTransformationLegalizations(
    mlir::ConversionTarget &target, mlir::AnalysisManager &am);
 
// Populate the conversion pattern by FoldMulAddChainToConvOpPattern, which
// folds a mul add chain into mul_conv and mac_conv.
void populateAIEVecConvOpTransformationPatterns(
    mlir::RewritePatternSet &patterns, mlir::AnalysisManager &am,
    unsigned shiftParam);
 
} // namespace aievec
} // namespace xilinx
 
#endif // FOLDMULADDCHAINTOCONVOP_H