fc8b2bf2f8
Add support for importing `dereferenceable` and `dereferenceable_or_null` metadata into LLVM dialect. Add a new attribute which models these two metadata nodes and a new OpInterface.
125 lines
3.8 KiB
C++
125 lines
3.8 KiB
C++
//===- LLVMInterfaces.cpp - LLVM Interfaces ---------------------*- C++ -*-===//
|
|
//
|
|
// 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 defines op interfaces for the LLVM dialect in MLIR.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "mlir/Dialect/LLVMIR/LLVMInterfaces.h"
|
|
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
|
|
|
|
using namespace mlir;
|
|
using namespace mlir::LLVM;
|
|
|
|
/// Verifies that all elements of `array` are instances of `Attr`.
|
|
template <class AttrT>
|
|
static LogicalResult isArrayOf(Operation *op, ArrayAttr array) {
|
|
for (Attribute iter : array)
|
|
if (!isa<AttrT>(iter))
|
|
return op->emitOpError("expected op to return array of ")
|
|
<< AttrT::getMnemonic() << " attributes";
|
|
return success();
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// AccessGroupOpInterface
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
LogicalResult mlir::LLVM::detail::verifyAccessGroupOpInterface(Operation *op) {
|
|
auto iface = cast<AccessGroupOpInterface>(op);
|
|
ArrayAttr accessGroups = iface.getAccessGroupsOrNull();
|
|
if (!accessGroups)
|
|
return success();
|
|
|
|
return isArrayOf<AccessGroupAttr>(op, accessGroups);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// AliasAnalysisOpInterface
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
LogicalResult
|
|
mlir::LLVM::detail::verifyAliasAnalysisOpInterface(Operation *op) {
|
|
auto iface = cast<AliasAnalysisOpInterface>(op);
|
|
|
|
if (auto aliasScopes = iface.getAliasScopesOrNull())
|
|
if (failed(isArrayOf<AliasScopeAttr>(op, aliasScopes)))
|
|
return failure();
|
|
|
|
if (auto noAliasScopes = iface.getNoAliasScopesOrNull())
|
|
if (failed(isArrayOf<AliasScopeAttr>(op, noAliasScopes)))
|
|
return failure();
|
|
|
|
ArrayAttr tags = iface.getTBAATagsOrNull();
|
|
if (!tags)
|
|
return success();
|
|
|
|
return isArrayOf<TBAATagAttr>(op, tags);
|
|
}
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// DereferenceableOpInterface
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
LogicalResult
|
|
mlir::LLVM::detail::verifyDereferenceableOpInterface(Operation *op) {
|
|
auto iface = cast<DereferenceableOpInterface>(op);
|
|
|
|
if (auto derefAttr = iface.getDereferenceableOrNull())
|
|
if (op->getNumResults() != 1 ||
|
|
!mlir::isa<LLVMPointerType>(op->getResult(0).getType()))
|
|
return op->emitOpError(
|
|
"expected op to return a single LLVM pointer type");
|
|
|
|
return success();
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::AtomicCmpXchgOp::getAccessedOperands() {
|
|
return {getPtr()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::AtomicRMWOp::getAccessedOperands() {
|
|
return {getPtr()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::LoadOp::getAccessedOperands() {
|
|
return {getAddr()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::StoreOp::getAccessedOperands() {
|
|
return {getAddr()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::MemcpyOp::getAccessedOperands() {
|
|
return {getDst(), getSrc()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::MemcpyInlineOp::getAccessedOperands() {
|
|
return {getDst(), getSrc()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::MemmoveOp::getAccessedOperands() {
|
|
return {getDst(), getSrc()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::MemsetOp::getAccessedOperands() {
|
|
return {getDst()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::MemsetInlineOp::getAccessedOperands() {
|
|
return {getDst()};
|
|
}
|
|
|
|
SmallVector<Value> mlir::LLVM::CallOp::getAccessedOperands() {
|
|
return llvm::filter_to_vector(getArgOperands(), [](Value arg) {
|
|
return isa<LLVMPointerType>(arg.getType());
|
|
});
|
|
}
|
|
|
|
#include "mlir/Dialect/LLVMIR/LLVMInterfaces.cpp.inc"
|