Passes.h

Go to the documentation of this file.

//===- Passes.h - Pass Entrypoints ------------------------------*- C++ -*-===//
//
// Part of the MLIR 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 header file defines prototypes that expose pass constructors in the loop
// transformation library.
//
//===----------------------------------------------------------------------===//

#ifndef MLIR_TRANSFORMS_PASSES_H
#define MLIR_TRANSFORMS_PASSES_H

#include "mlir/Support/LLVM.h"
#include <functional>
#include <limits>

namespace mlir {

class AffineForOp;
class FuncOp;
class ModuleOp;
class Pass;
template <typename T> class OpPassBase;

std::unique_ptr<Pass> createCanonicalizerPass();

std::unique_ptr<Pass> createCSEPass();

std::unique_ptr<OpPassBase<FuncOp>>
createVectorizePass(ArrayRef<int64_t> virtualVectorSize);

std::unique_ptr<OpPassBase<FuncOp>> createVectorizerTestPass();

std::unique_ptr<OpPassBase<FuncOp>>
createMaterializeVectorsPass(ArrayRef<int64_t> vectorSize);

std::unique_ptr<OpPassBase<FuncOp>> createLoopUnrollPass(
    int unrollFactor = -1, int unrollFull = -1,
    const std::function<unsigned(AffineForOp)> &getUnrollFactor = nullptr);

std::unique_ptr<OpPassBase<FuncOp>>
createLoopUnrollAndJamPass(int unrollJamFactor = -1);

std::unique_ptr<OpPassBase<FuncOp>> createSimplifyAffineStructuresPass();

std::unique_ptr<OpPassBase<FuncOp>>
createLoopFusionPass(unsigned fastMemorySpace = 0,
                     uint64_t localBufSizeThreshold = 0,
                     bool maximalFusion = false);

std::unique_ptr<Pass> createLoopInvariantCodeMotionPass();

std::unique_ptr<OpPassBase<FuncOp>> createAffineLoopInvariantCodeMotionPass();

std::unique_ptr<OpPassBase<FuncOp>> createPipelineDataTransferPass();

std::unique_ptr<OpPassBase<FuncOp>> createLowerAffinePass();

std::unique_ptr<OpPassBase<FuncOp>>
createLoopTilingPass(uint64_t cacheSizeBytes);

std::unique_ptr<OpPassBase<FuncOp>>
createSimpleParametricTilingPass(ArrayRef<int64_t> outerLoopSizes);

std::unique_ptr<OpPassBase<FuncOp>> createLoopCoalescingPass();

std::unique_ptr<OpPassBase<FuncOp>> createAffineDataCopyGenerationPass(
    unsigned slowMemorySpace, unsigned fastMemorySpace,
    unsigned tagMemorySpace = 0, int minDmaTransferSize = 1024,
    uint64_t fastMemCapacityBytes = std::numeric_limits<uint64_t>::max());

std::unique_ptr<OpPassBase<FuncOp>> createMemRefDataFlowOptPass();

std::unique_ptr<OpPassBase<FuncOp>> createStripDebugInfoPass();

std::unique_ptr<OpPassBase<FuncOp>> createTestLoopFusionPass();

std::unique_ptr<Pass> createInlinerPass();
} // end namespace mlir

#endif // MLIR_TRANSFORMS_PASSES_H