blob: 5d86195305a95739337943637175e4bb757d0f7d [file] [log] [blame]
//===- Traits.h - Trait Declaration for MLIR DLTI dialect -------*- C++ -*-===//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#include "mlir/IR/OpDefinition.h"
#include "mlir/Interfaces/DataLayoutInterfaces.h"
namespace mlir {
class DataLayoutSpecAttr;
namespace impl {
LogicalResult verifyHasDefaultDLTIDataLayoutTrait(Operation *op);
DataLayoutSpecInterface getDataLayoutSpec(Operation *op);
} // namespace impl
/// Trait to be used by operations willing to use the implementation of the
/// data layout interfaces provided by the Target dialect.
template <typename ConcreteOp>
class HasDefaultDLTIDataLayout
: public OpTrait::TraitBase<ConcreteOp, HasDefaultDLTIDataLayout> {
/// Verifies that the operation to which this trait is attached is valid for
/// the trait, i.e., that it implements the data layout operation interface.
static LogicalResult verifyTrait(Operation *op) {
return impl::verifyHasDefaultDLTIDataLayoutTrait(op);
/// Returns the data layout specification as provided by the Target dialect
/// specification attribute.
DataLayoutSpecInterface getDataLayoutSpec() {
return impl::getDataLayoutSpec(this->getOperation());
} // namespace mlir