d3588d0814
Summary: Functional.h contains many different methods that have a direct, and more efficient, equivalent in LLVM. This revision replaces all usages with the LLVM equivalent, and removes the header. This is part of larger cleanup, pr45513, merging MLIR support facilities into LLVM. Differential Revision: https://reviews.llvm.org/D78053
42 lines
1.6 KiB
C++
42 lines
1.6 KiB
C++
//===- Builders.cpp - MLIR Declarative Linalg Builders --------------------===//
|
|
//
|
|
// 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/Vector/EDSC/Builders.h"
|
|
#include "mlir/Dialect/Vector/EDSC/Intrinsics.h"
|
|
#include "mlir/Dialect/Vector/VectorOps.h"
|
|
#include "mlir/EDSC/Builders.h"
|
|
#include "mlir/EDSC/Intrinsics.h"
|
|
#include "mlir/IR/AffineExpr.h"
|
|
#include "mlir/IR/Builders.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::edsc;
|
|
using namespace mlir::edsc::intrinsics;
|
|
using namespace mlir::edsc::ops;
|
|
|
|
Value mlir::edsc::ops::vector_contraction(
|
|
StructuredIndexed A, StructuredIndexed B, StructuredIndexed C,
|
|
ArrayRef<IteratorType> iteratorTypes) {
|
|
using IndexingExprs = ArrayRef<ArrayRef<AffineExpr>>;
|
|
return vector_contract(
|
|
A.getValue(), B.getValue(), C.getValue(),
|
|
IndexingExprs{A.getExprs(), B.getExprs(), C.getExprs()},
|
|
ArrayRef<StringRef>{
|
|
llvm::to_vector<8>(llvm::map_range(iteratorTypes, toString))});
|
|
}
|
|
|
|
Value mlir::edsc::ops::vector_contraction_matmul(Value A, Value B, Value C) {
|
|
AffineExpr m, n, k;
|
|
bindDims(ScopedContext::getContext(), m, n, k);
|
|
return vector_contraction(StructuredIndexed(A, {m, k}),
|
|
StructuredIndexed(B, {k, n}),
|
|
StructuredIndexed(C, {m, n}),
|
|
{IteratorType::Parallel, IteratorType::Parallel,
|
|
IteratorType::Reduction});
|
|
}
|