6cbcb79350
Introduce an operation to specialize linalg.generics, for example, detecting a linalg.generic that is semantically equivalent to a linalg.copy and replacing the former with the latter. After code generation, it is helpful to lower named operations to vendor-optimized libraries.
33 lines
1.2 KiB
C++
33 lines
1.2 KiB
C++
//===- Specialize.cpp - linalg generic ops to named 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements a method to specialize generic operations to named
|
|
// operations. Conceptually it is the opposite of generalize.cpp.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/Linalg/IR/Linalg.h"
|
|
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"
|
|
#include "mlir/Dialect/Linalg/Transforms/Transforms.h"
|
|
#include "llvm/Support/Debug.h"
|
|
|
|
#define DEBUG_TYPE "linalg-specialization"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::linalg;
|
|
|
|
FailureOr<LinalgOp> mlir::linalg::specializeGenericOp(RewriterBase &rewriter,
|
|
GenericOp genericOp) {
|
|
if (isaCopyOpInterface(genericOp)) {
|
|
LinalgOp namedOp = rewriter.replaceOpWithNewOp<CopyOp>(
|
|
genericOp, genericOp.getDpsInputs()[0], genericOp.getDpsInits()[0]);
|
|
return namedOp;
|
|
}
|
|
return failure();
|
|
}
|