llvm/mlir/lib/Dialect/Utils/StructuredOpsUtils.cpp

//===- StructuredOpsUtils.cpp - Utilities used by structured ops ----------===//
//
// 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
//
//===----------------------------------------------------------------------===//

#include "mlir/Dialect/Utils/StructuredOpsUtils.h"
#include "mlir/IR/AffineMap.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/BuiltinAttributes.h"
#include "mlir/IR/IRMapping.h"
#include "llvm/ADT/StringSet.h"

#include "mlir/Dialect/Utils/DialectUtilsEnums.cpp.inc"

usingnamespacemlir;

bool mlir::isRowMajorMatmul(ArrayAttr indexingMaps) { … }

bool mlir::isColumnMajorMatmul(ArrayAttr indexingMaps) { … }

bool mlir::isRowMajorBatchMatmul(ArrayAttr indexingMaps) { … }

bool mlir::isVecmat(ArrayAttr indexingMaps) { … }

bool mlir::isBatchVecmat(ArrayAttr indexingMaps) { … }

bool mlir::isMatvec(ArrayAttr indexingMaps) { … }

bool mlir::isBatchMatvec(ArrayAttr indexingMaps) { … }

Operation *mlir::clone(OpBuilder &b, Operation *op, TypeRange newResultTypes,
                       ValueRange newOperands) { … }

Operation *mlir::cloneWithoutRegions(OpBuilder &b, Operation *op,
                                     TypeRange newResultTypes,
                                     ValueRange newOperands) { … }

SmallVector<NamedAttribute>
mlir::getPrunedAttributeList(Operation *op, ArrayRef<StringRef> elidedAttrs) { … }