AIEXDialect.h

Go to the documentation of this file.

//===- AIEDialect.h ---------------------------------------------*- C++ -*-===//
//
// This file is licensed 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
//
// (c) Copyright 2019 Xilinx Inc.
//
//===----------------------------------------------------------------------===//
 
#ifndef MLIR_AIEX_DIALECT_H
#define MLIR_AIEX_DIALECT_H
 
#include "aie/Dialect/AIE/IR/AIEDialect.h"
 
#include "mlir/IR/BuiltinAttributes.h"
 
// Include dialect declarations such as parseAttributes, parseType
#include "aie/Dialect/AIEX/IR/AIEXDialect.h.inc"
#include "mlir/IR/Operation.h"
 
// include TableGen generated Op definitions
#define GET_OP_CLASSES
#include "aie/Dialect/AIEX/IR/AIEX.h.inc"
 
#define GET_TYPEDEF_CLASSES
#include "aie/Dialect/AIEX/IR/AIEXTypes.h.inc"
 
#include "llvm/ADT/StringRef.h"
 
#include <optional>
 
namespace xilinx {
namespace AIEX {
 
void getHardwareStridesWraps(const AIE::AIETargetModel &targetModel,
                             mlir::Operation *op,
                             mlir::BaseMemRefType referencedBufType,
                             llvm::SmallVector<int64_t, 4> inputSizes,
                             llvm::SmallVector<int64_t, 4> inputStrides,
                             llvm::SmallVector<int64_t, 4> &sizes,
                             llvm::SmallVector<int64_t, 4> &strides);
mlir::LogicalResult
verifyStridesWraps(mlir::Operation *forOp,
                   mlir::BaseMemRefType referencedBufType, int tileCol,
                   int tileRow, llvm::SmallVector<int64_t, 4> inputSizes,
                   llvm::SmallVector<int64_t, 4> inputStrides,
                   llvm::SmallVector<int64_t, 4> hardwareSizes,
                   llvm::SmallVector<int64_t, 4> hardwareStrides,
                   bool skipTransformationChecks = false);
bool isLinearTransfer(llvm::ArrayRef<int64_t> sizes,
                      llvm::ArrayRef<int64_t> strides);
 
} // namespace AIEX
} // namespace xilinx
 
#endif